Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

macOS/Linux install instructions #170

Open
wants to merge 3 commits into
base: master
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
25 changes: 19 additions & 6 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -33,16 +33,29 @@ If you use this code for your research, please cite our paper:

### Install dependencies

```
python -m pip install -r requirements.txt
```
This code was tested with python 3.6, torch 1.4. Using newer versions is likely to produce runtime errors as well as different performance.

This code was tested with python 3.6, torch 1.4
For later torch versions, you may try this repository: https://github.com/kligvasser/SinGAN (results won't necessarily be identical to the official implementation and cpu is not an option).
#### Install python 3.6, torch 1.4 - MacOS/Linux

Please note: the code currently only supports torch 1.4 or earlier because of the optimization scheme.
Install Anaconda package manager (needed, since they have correct older binaries in their repos).
On macOS (assuming you have homebrew) it can be installed with:
```
brew install --cask anaconda
```

For later torch versions, you may try this repository: https://github.com/kligvasser/SinGAN (results won't necessarily be identical to the official implementation).
Then:
```
conda create --name singanEnvironment
conda activate singanEnvironment
```
'activate' might produce a warning regarding shell integration, follow instructions of conda

```
conda install python==3.6.13
conda install pytorch==1.4.0 torchvision==0.5.0 -c pytorch
conda install --file requirements.txt
```

### Train
To train SinGAN model on your own image, put the desired training image under Input/Images, and run
Expand Down
12 changes: 6 additions & 6 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
matplotlib
scikit-image
scikit-learn
scipy
matplotlib==3.3.4
scikit-image==0.17.2
scikit-learn==0.24.2
scipy==1.5.2
numpy
torch
torchvision
pytorch==1.4.0
torchvision==0.5.0