这是在caffe上基于Deep Residual Network实现显著性目标检测的模型,算法是fine tune训练好的Resnet网络实现的
该网络实现是基于Pyramid Scene Parsing Network新建的caffe版本实现,其中主要增加了Interp层的功能。
caffe版本下载地址:https://github.com/hszhao/PSPNet; 按流程安装即可。
本人实现是在10000张图片库THUS上训练的,请自行构建train.txt和train_gt01.txt文件,图片和真值图索引,其中所有图片的分类标签都设置为0。 因为我们没有涉及分类,所以分类标签不影响结果。但是请将所有真值图中0/1标识前背景从double类型转为uint8类型存储。
原网络是kaiming he的深度残差网络Resnet。
train.ptototxt描述整体训练网络框架,其中需要将训练数据txt文件路径两个修改成相对应你的路径;
solver.prototxt是网络参数配置文件;
deploy.prototxt是测试网络框架;
trainModel.sh是fine tune训练网络的shell指令,请将对应Resnet的caffemodel修改成自己的路径;
test.m是调用caffe的matlab接口实现测试。其中网络参数是调用训练新生成的caffemodel文件;请修改文件中对应对方的路径。