在运行MambaIR项目脚本的时候,尽管按照作者要求配置环境如下:
conda env update --name mambair --file environment.yaml
还是会报错找不到’basicsr’?
(mambair) root@autodl-container-18b640a1a0-4f72471d:~/autodl-tmp/MambaIR# python basicsr/test.py -opt options/test/test_MambaIR_SR_x4.yml
Traceback (most recent call last):
File "basicsr/test.py", line 7, in <module>
from basicsr.data import build_dataloader, build_dataset
ModuleNotFoundError: No module named 'basicsr'
这是basicsr
作为本项目的一部分没有包含在 requirements.txt
或 environment.yaml
中。但本地模块(包括 basicsr
)并未被正确地注册到 Python 环境中,导致环境配置的过程中并没有导入这个模块导致找不到。也就是说,这里的项目工作以开发者模式引用了另一个名为basicsr的工作,因此需要先添加至python环境中,而不能简单的pip install解决。
解决办法:在test.py文件的第五行加入下面的代码,将路径改成项目文件夹的绝对路径
sys.path.append('/root/autodl-tmp/MambaIR')
即可解决,不需要手动安装。
没有回复内容