search

richzhang / colorization

Automatic colorization using deep neural networks. "Colorful Image Colorization." In ECCV, 2016.
http://richzhang.github.io/colorization/
BSD 2-Clause "Simplified" License
3.34k stars 921 forks source link

requirements.txt #105

Open AnUnassumingUsername opened 1 year ago

AnUnassumingUsername commented 1 year ago

Errors on installing dependencies, Pil apparently discontinued and scikit-image worked rather than skimage - running from windows cmd line in cloned directory. Using Python 3.11.3 - so may just be some quirks of my configuration but these changes to the requirements.txt worked for this configuration

torch scikit-image numpy matplotlib argparse Pillow

Keshawn commented 1 year ago

add IPython

michaelnew commented 1 month ago

FWIW, this is my working requirements.txt file:

scikit-image
numpy
matplotlib
argparse
Pillow
IPython
torch

And to run on AMD, add the line: --extra-index-url https://download.pytorch.org/whl/rocm5.7 above torch