https://github.com/voletiv/self-attention-gan-pytorch

This is an almost exact replica in PyTorch of the Tensorflow version of Self-Attention GAN released by Google Brain in August 2018.
https://github.com/voletiv/self-attention-gan-pytorch

Last synced: 9 days ago
JSON representation

This is an almost exact replica in PyTorch of the Tensorflow version of Self-Attention GAN released by Google Brain in August 2018.

• Host: GitHub
• URL: https://github.com/voletiv/self-attention-gan-pytorch
• Owner: voletiv
• License: mit
• Created: 2019-01-16T20:08:05.000Z (almost 6 years ago)
• Default Branch: master
• Last Pushed: 2024-07-25T13:12:29.000Z (3 months ago)
• Last Synced: 2024-07-25T15:11:40.216Z (3 months ago)
• Language: Python
• Size: 37.1 KB
• Stars: 170
• Watchers: 7
• Forks: 27
• Open Issues: 1
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# self-attention-GAN-pytorch

This is an almost exact replica in PyTorch of the Tensorflow version of [SAGAN](https://arxiv.org/abs/1805.08318) released by Google Brain [[repo](https://github.com/brain-research/self-attention-gan)] in August 2018.

Code structure is inspired from [this repo](https://github.com/heykeetae/Self-Attention-GAN), but follows the details of [Google Brain's repo](https://github.com/brain-research/self-attention-gan).

## Prerequisites

Check `requirements.txt`.

* [Python 3.5+](https://www.continuum.io/downloads)

* [PyTorch 0.4.1](http://pytorch.org/)

## Training

#### 1. Check `parameters.py` for all arguments and their default values

#### 2. Train on custom images in folder a/b/c:

```bash

$ python train.py --data_path 'a/b/c' --save_path 'o/p/q' --batch_size 64 --name sagan

```

(Warning: Works only on *128x128* images, input images are resized to that. Tweak the Generator & Discriminator first if you would like to use some other image size. And then use the `imsize` option:

```bash

$ python train.py --data_path 'a/b/c' --save_path 'o/p/q' --batch_size 64 --imsize 32 --name sagan

```

)

Model training will be recorded in a new folder inside `--save_path` with the name `__`.

By default, model weights are saved in a subfolder called `weights`, and train & validation samples during training in a subfolder called `samples` (can be changed in `parameters.py`).

## Testing/Evaluating

Check `test.py`.

## Self-Attention GAN

**[Han Zhang, Ian Goodfellow, Dimitris Metaxas and Augustus Odena, "Self-Attention Generative Adversarial Networks." arXiv preprint arXiv:1805.08318 (2018)](https://arxiv.org/abs/1805.08318).**

```

@article{Zhang2018SelfAttentionGA,

    title={Self-Attention Generative Adversarial Networks},

    author={Han Zhang and Ian J. Goodfellow and Dimitris N. Metaxas and Augustus Odena},

    journal={CoRR},

    year={2018},

    volume={abs/1805.08318}

}

```