satoshiiizuka / siggraphasia2019_remastering

Code for the paper "DeepRemaster: Temporal Source-Reference Attention Networks for Comprehensive Video Enhancement". http://iizuka.cs.tsukuba.ac.jp/projects/remastering/
Other
482 stars 102 forks source link

ImportError #2

Open idanka opened 4 years ago

idanka commented 4 years ago

Hi! What could be the cause of the error? install "conda install pytorch torchvision cudatoolkit=10.2 -c pytorch"

python remaster.py --input elefant.mp4 --gpu --disable_colorization

Traceback (most recent call last): File "remaster.py", line 17, in from torchvision import transforms File "/home/dankahazi/anaconda3/lib/python3.7/site-packages/torchvision/init.py", line 2, in from torchvision import datasets File "/home/dankahazi/anaconda3/lib/python3.7/site-packages/torchvision/datasets/init.py", line 9, in from .fakedata import FakeData File "/home/dankahazi/anaconda3/lib/python3.7/site-packages/torchvision/datasets/fakedata.py", line 3, in from .. import transforms File "/home/dankahazi/anaconda3/lib/python3.7/site-packages/torchvision/transforms/init.py", line 1, in from .transforms import * File "/home/dankahazi/anaconda3/lib/python3.7/site-packages/torchvision/transforms/transforms.py", line 17, in from . import functional as F File "/home/dankahazi/anaconda3/lib/python3.7/site-packages/torchvision/transforms/functional.py", line 5, in from PIL import Image, ImageOps, ImageEnhance, PILLOW_VERSION ImportError: cannot import name 'PILLOW_VERSION' from 'PIL' (/home/dankahazi/anaconda3/lib/python3.7/site-packages/PIL/init.py)

idanka commented 4 years ago

or use "conda install pytorch torchvision cudatoolkit=10.1 -c pytorch"

python remaster.py --input elefant.mp4 --gpu --disable_colorization

Traceback (most recent call last): File "remaster.py", line 19, in import cv2 ModuleNotFoundError: No module named 'cv2'