Added requirements, and updated readme file

RForestLiu · Jun 18, 2018 · fd6785f · fd6785f
1 parent 74ab164
commit fd6785f
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 5 deletions.
diff --git 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
+   ```
+   [email protected]: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)
+[[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
@@ -0,0 +1,4 @@
+torch>=0.4.0
+torchvision>=0.2.1
+tqdm>=4.23.4
+visdom>=0.1.8.3