SINetV2复现代码问题及解决方法

需要安装的依赖包

pip install thop
pip install tensorboardX==2.1
pip install protobuf==3.19.4

安装指定的cuda版本pytorch命令

conda install pytorch torchvision cudatoolkit=11.3 -c pytorch

如何利用下载的Res2net50预训练权重文件

假设你决定将权重文件放在 lib 目录下的 models 子目录中，你可以这样组织：

然后，你需要在 Res2Net_v1b.py 中修改加载模型权重的代码部分，以确保它正确地指向新的路径。例如，如果你将权重文件放在了 lib/models/ 下，代码中加载模型权重的部分应该修改为：

model_state = torch.load('lib/models/res2net50_v1b_26w_4s-3cf99910.pth')

安装 opencv-python 报错

首先安装 scikit-build

pip install scikit-build

然后再安装 opencv-python ，python3.6对应的 opencv-python版本为

pip install opencv-python==4.5.4.60

安装 opencv-python 卡住在 running setup.py bdist_wheel for opencv-python

解决方法: 更新 pip 版本

pip install --upgrade pip

执行 MyTesting.py 报错

Traceback (most recent call last):
  File "MyTesting.py", line 40, in <module>
    misc.imsave(save_path+name, res)
AttributeError: module 'scipy.misc' has no attribute 'imsave'

解决方法:

pip install imageio

然后修改这部分代码:

from scipy import misc
misc.imsave(save_path+name, res)

为:

import imageio
imageio.imwrite(save_path+name, res)