Skip to content

saksham-koul/Self-Attention-GAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

Self Attention Generative Adversarial Networks

Procedure to run the project:

  • Install the Celeb-A dataset

  • Celeb-A Faces dataset can be downloaded at the linked site, or in Google Drive. The dataset will download as a file named img_align_celeba.zip

  • Extract the dataset into data/celeba

  • Install Pytorch, NumPy, Matplotlib

  • Install Pytorch Ignite using

    pip:

      pip install pytorch-ignite
    

    conda:

      conda install ignite -c pytorch
    
  • CUDA Device(GPU) must be present to run the IPython notebook.

  • Run all cells of the IPython Notebook.

About

Implementation of the Self-Attention GAN research paper in PyTorch.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published