megvii-model/HINet

ImportError

Yuri6037 opened this issue · 4 comments

Yuri6037 commented

Your network does not run on my research machine:

Traceback (most recent call last):
  File "basicsr/demo.py", line 10, in <module>
    from basicsr.models import create_model
ImportError: cannot import name 'create_model'

EDIT: Just tried on my local computer (RTX 3060) different error:

Traceback (most recent call last):
  File "/home/X/HINet/basicsr/demo.py", line 10, in <module>
    from basicsr.models import create_model
ModuleNotFoundError: No module named 'basicsr'
mexthecat commented

maybe you have forgotten: "python setup.py develop --no_cuda_ext"

Yuri6037 commented

I've got the software to run on my RTX 3060 by moving main.py outside of the basicsr directory and by implementing image decomposition to avoid VRAM overload (my RTX only has 6Gb of VRAM).

On the remote machine according to some Python Matrix channel it's unfixable: the problem is because you've named the directory basicsr which conflicts with the module I have installed (also named basicsr).

mayorx commented

Hi, Yuri6037,

Thanks, and you could try export PYTHONPATH=path_to_the_HINet_folder in your bash before run basicsr/demo.py. It might be helpful.

dante1024 commented

maybe you have forgotten: "python setup.py develop --no_cuda_ext"

What is the operation of "python setup.py develop --no_cuda_ext" used for?