OCGAN - UnOfficial PyTorch Implementation
This is the unofficial PyTorch implementation of OCGAN. The code accompanies the paper "OCGAN: One-class Novelty Detection Using GANs with Constrained Latent Representations".
The author's implementation of OCGAN in MXNet is at here.
- Unified interface for different network architectures
- Multi-GPU support
- Training progress bar with rich info
- Training log and training curve visualization code (see
./utils/logger.py)
This code is written in Python 3.7 and tested with Pytorch 1.2.0.
- Install PyTorch
- Clone recursively
git clone --recursive https://github.com/xiehousen/OCGAN-Pytorch.git
python train.py
During training, every five epochs will store the training input and output pictures in the result folder.
The picture on the left is the generated image of the autoencoder, and the right is the real training image
python test.py
The test results are also saved in the result folder.
The picture on the left is the generated image of the test data set after passing through the automatic encoder, and the right is the real test image.
During the test, the AUC was also calculated, and the calculation results were stored in the log file of the result folder.
When training 100 epochs, the auc did not achieve the results in the paper. We continue to improve the code to achieve the results in the paper.
This code borrows part from