diff --git a/README.md b/README.md index 75b712e..4a78363 100644 --- a/README.md +++ b/README.md @@ -16,10 +16,24 @@ This repository contains PyTorch implementation of the following paper: GANomaly ## Prerequisites -1. OS: Linux or MacOS -2. PyTorch v0.3 - For now v0.4 is not supported -3. GPU - Highly recommended - +1. Linux or MacOS +2. Python 3 +3. CPU or GPU + CUDA & CUDNN +2. To install the required modules, run the following: + ``` + pip install -R requirements + ``` + +## Installation +1. First clone the repository + ``` + git@github.com:samet-akcay/ganomaly.git + ``` +2. Install PyTorch and torchvision from [https://pytorch.org](https://pytorch.org/) +3. Install the dependencies. + ``` + pip install -r requirements.txt + ``` ## Experiment @@ -155,4 +169,4 @@ If you use this repository or would like to refer the paper, please use the foll ``` ## Reference -[[1] S. Akcay, A. Atapour-Abarghouei, and T. P. Breckon. GANomaly: Semi-SupervisedAnomaly Detection via Adversarial Training. ArXiv e-prints, May 2018.](https://arxiv.org/abs/1805.06725) \ No newline at end of file +[[1] S. Akcay, A. Atapour-Abarghouei, and T. P. Breckon. GANomaly: Semi-SupervisedAnomaly Detection via Adversarial Training. ArXiv e-prints, May 2018.](https://arxiv.org/abs/1805.06725) diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..784cc71 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,4 @@ +torch>=0.4.0 +torchvision>=0.2.1 +tqdm>=4.23.4 +visdom>=0.1.8.3