[跑代码] MVGNet论文解析(一)

github

https://github.com/1900zpf/MVGNet

仅提供了
1 预测结果
2 训练数据
3 测试数据
4 验证集(根据源代码显示是COD10K)
5 训练集是TrainDataset_aug(不知道是不是标准训练集)

一 、 MyTrain.py

0 训练参数 code-paper不一致，以论文为准
batchsize=2 （论文没说）
trainsize=384 (论文是352)
decay_rate=0.1(论文是0.0005=5e-4)
优化器SGD代码没找到
momentum代码没有 (论文是0.9)

1 默认周期
--epoch=50
--batchsize=2

2 默认gpu id 0  （改成9）

3 backbone

默认是P2Tlarge

 

4 读取数据子文件名
默认 Imgs GT

改为image mask(验证集也要改)

5 训练集 验证集路径需要修改（验证集原代码：COD10K）

6 模型checkpoint保存路径（换路径）
--save_path

7 导入主要网络(需要修改)
from Net.MyNet import MyNet as Network

8 安装库
1) 缺少mmseg, pip install mmsegmentation
2) 缺少mmcv,
pip install mmcv
报错 error: legacy-install-failure
pip install --upgrade setuptools
pip install mmcv-full

pip install -U openmim
mim install mmcv-full

总之，上述解决办法均不行

解决办法：

mim install mmcv==2.0.0

3) 缺少 ftfy
ModuleNotFoundError: No module named 'ftfy'

8 下载和放置 PreTrained model
'/home/server/zpf/best_code_cod/Pred_model/p2t_large.pth'

p2t.py 414行
改为
/cluster/data7a/xxx/Code/Backbone/p2t_large.pth
去github找p2t tpami 下载

二、Net/MyNet文件修改

删除 Huang123

三、Test.py测试文件修改

1 输入MyNet

2 模型路径pth
3 gpu
4 显著图路径SM

5 测试集根目录

6 修改为2个输出
因为MyNet输出是2个（多监督）
因此MyTest需要改成2输出
output_sum, _