EGNet使用说明-训练测试-vgg还是res-等指令

1 训练

python run.py --mode train

 训练集的路径：dataset.py

2 测试

python run.py --mode test --sal_mode e

其中选择数据集只需要换代表的数据集简写字母即可

e ecssd
p pascal
d dut-omron
t duts
h hku

测试集路径设置：dataset.py

2 选择使用哪个模型进行测试

run.py 文件中

./model_test/xxx.pth

一定记得在这立选择模型进行测试

 

2 测试结果的输出位置

Result/saliency 文件中

2 读入图片后缀设置（索引.lst文件里面只有文件名没有后缀）

dataset.py

dataset.py  文件中读入结果是 ./data/ECSSD/image/0123
注意这里没有后缀，后缀是在load_image_test文件中添加的

./data/ECSSD/image/0123.jpg

 

3 选择不同的干网backbone

solver.py文件中选

4 训练得到的模型&跑测试的模型存放地

5 特别注意在dataset.py文件中可以改变测试集的输入图片的后缀jpg还是png这个一定要写对