运行MINet

1 github下载code

2 调整数据集路径
打开文件config.py
先更改下根目录 datasets_root为MINet文件夹所在路径
然后放那个进去我的data
datasets_root &＃61; "/home/nk/zjc/PycharmProjects/3ClassicAlgo
rithm/MINet/MINet-master/code"
dutstr_path &＃61; os.path.join(datasets_root, "data", "DUTS-TRAIN")

3 设置模型名字为已经被 import 进 network里面的_init__.py

4 pytorch 要求1.1.0

5要求装tensorboar

6 Image Mask 改为image mask

7 左边config.py 改程对应的类

在测试的时候&＃xff0c;pytorch 小于1.2.0就会出现问题&＃xff0c;很不幸我就是1.1.0会在
solver文件的270附近无法进行for循环&＃xff0c;

解决办法&＃xff1a;注释掉dataloader.py文件中的

collate_fn&＃61;partial(_collate_fn, size_list&＃61;size_list) if size_list else None,

因为版本不同默认值就变了1.2以上的collate_fn被设置为None的时候会调用默认值&＃xff0c;低版本不会&＃xff0c;
注释掉之后不对collate_fn传递值