TF_SWAG

Unofficial Implementation of SWAG in TensorFlow

This repository cantains the implemetation of the paper Rethinking and Improving the Robustness of Image Style Transfer in TensorFlow.

Usage

Install the dependencies

pip install -r requirements.txt

Run the following command

python resnet_swag.py

Choose any one of the architecture between resnet50_swag and resnet50 by using the --architecture flag. The default being resnet50_swag

python resnet_swag.py --architecture='resnet50'

Change the number of iterations by using the --iter flag

python resnet_swag.py --iter=500

The code is heavily borrowed from the original implementation of the paper

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
content_images		content_images
style_images		style_images
README.md		README.md
requirements.txt		requirements.txt
resnet_swag.py		resnet_swag.py