YuleZhang / GTNC

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

pip install -r requirements.txt

You can run start_train_gtnc.py in python3 and the parameters can be changed in Parameters.py.

By default, cpu is used for calculation. GPU can also be used in start_train_gtnc.py. (see it for details)

Here are the version informations:

Python 3.6.8

numpy 1.16.0

spicy 0.16.0

torch 1.1.0

torchvision 0.3.0

scikit-image 0.15.0

opencv-python 4.1.0.25

nvidia-ml-py3 7.352.0

About


Languages

Language:Python 100.0%