Skip to content

MB-29/GAN-anomaly-detection

 
 

Repository files navigation

Anomaly detection

Anomaly detection using cycle-GAN.

Originial Repository

Image-to-Image Translation in PyTorch

Setup

Please install CycleGAN network and download the pix2pix maps dataset following the instructions provided in the original repository. The checkpoint files and weights we obtained after a training phase of 11 epochs and that we used for our project are available here.

Compute and save embeddings

python anomaly_detection/features.py --dataroot ./datasets/[Data folder] --name maps_cyclegan --model cycle_gan --gpu_ids -1

Naive tests

Run

python anomaly_detection/data_analysis.py

Anomaly detection

Please find our work in the Jupyter notebooks here.

About

Image-to-Image Translation in PyTorch

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 87.3%
  • Jupyter Notebook 8.2%
  • Shell 2.4%
  • MATLAB 1.1%
  • TeX 1.0%