-
Notifications
You must be signed in to change notification settings - Fork 300
Description
I love the idea of this project, but following the readme.md does not produce a working test. Is there something I'm missing? For instance, after creating the environment
conda create -n vsr python=3.7
conda activate vsr
conda install tensorflow-gpu==1.15.0
pip install -e .
python prepare_data.py --filter vespcn vid4
you will get an error that torch is needed (which the readme says is optional). That's fine, so the user may then install torch that's compatible with this version of python and tensorflow from conda
conda install pytorch torchvision
Now you can run the prepare data example, but when you go to evaluate:
python eval.py srcnn -t vid4 --pretrain=C:\repos\VideoSuperResolution\models\srcnn_ep0200.pth
You will get "RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU."
A user's next inclination would be to find out how to find out how to install torch with gpu support, so they go over to https://pytorch.org/get-started/locally/ and follow the directions. Making a new environment.
conda create -n vsr python=3.9
activate vsr
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
pip install -e .
conda install tensorflow-gpu keras
python prepare_data.py --filter vespcn vid4
This now fails with
2023-04-14 22:02:00,744 WARNING: [!] PyTorch version too low: 2.0.0, recommended 1.2.0
C:\repos\VideoSuperResolution\VSR\DataLoader\VirtualFile.py:423: SyntaxWarning: "is" with a literal. Did you mean "=="?
assert len(padding) is 2, f"Invalid padding, {padding}"
C:\repos\VideoSuperResolution\VSR\Util\ImageProcess.py:31: SyntaxWarning: "is" with a literal. Did you mean "=="?
return ret.pop() if len(ret) is 1 else ret
C:\repos\VideoSuperResolution\VSR\DataLoader\Loader.py:335: SyntaxWarning: "is" with a literal. Did you mean "=="?
assert len(shape) is 5, f"Shape is not 5D, which is {len(shape)}"
I would be happy to help with development if that's what's needed.