zllrunning / video-object-removal

Just draw a bounding box and you can remove the object you want to remove.
MIT License
2.66k stars 474 forks source link

I have a problem with importing torch #5

Open bleachjade opened 5 years ago

bleachjade commented 5 years ago

sorry I'm new here, can anybody help?

Traceback (most recent call last): File "demo.py", line 2, in from mask import mask File "C:\Users\Jade\video-object-removal\mask.py", line 8, in from get_mask.test import * File "C:\Users\Jade\video-object-removal\get_mask\test.py", line 16, in from get_mask.utils.load_helper import load_pretrain File "C:\Users\Jade\video-object-removal\get_mask\utils\load_helper.py", line 1, in import torch ModuleNotFoundError: No module named 'torch'

zllrunning commented 5 years ago

You do not install PyTorch. Furthermore, this repo needs exactly Pytorch 0.4.0.