Skip to content
Ready to train Pytorch implementation of the CVPR'19 paper "Self-Supervised GANs via Auxiliary Rotation Loss"
Python
Branch: master
Clone or download
Latest commit 01408fc Nov 11, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
LICENSE Initial commit Nov 10, 2019
README.md Update README.md Nov 12, 2019
dataloaders.py Adds initial code for SSGAN Nov 10, 2019
main.py Adds initial code for SSGAN Nov 10, 2019
model.py Adds initial code for SSGAN Nov 10, 2019
ops.py Adds initial code for SSGAN Nov 10, 2019
ssgan.png adds image for readme Nov 10, 2019
training.py Adds initial code for SSGAN Nov 10, 2019

README.md

Self-Supervised-Gans-Pytorch

Pytorch implementation of the CVPR'19 paper "Self-Supervised GANs via Auxiliary Rotation Loss"

Ting Chen, Xiaohua Zhai(Google Brain), Marvin Ritter(Google Brain), Mario Lucic(Google Brain), Neil Houlsby(Google Brain)

Dependencies

  • Python (>=3.6)
  • Pytorch (>=1.2.0)

Review article of the paper

Medium Article

Training

python main.py

How it works

The paper presents a method to combine adverserial training with self-supervised learning. It uses the concept of Auxilliary Rotation Loss. The main idea behind self-supervision is to train a model on a pretext task like predicting rotation angle and then extracting representations from the resulting networks. The discriminator also tries to predict the rotatory angle(0, 90, 180, 270) along with the normal prediction of fake vs real.

alt-text

References

You can’t perform that action at this time.