1、准备自己训练的原图和边缘图,边缘图制作参考https://blog.csdn.net/weixin_38517705/article/details/84670150
2、将制作好的数据集放在.../rcf-master/data下,我是分为两个文件夹,一个存放原图(hed)、一个存放边缘图(hedshu)
3、my_dataset.lst存放的是数据集的列表,方便读取hed和hedshu的数据。原始的列表长这样
模仿着做自己的my_dataset.lst
先将hed文件夹中的图片信息读取到一个Excel中,方法可参考https://blog.csdn.net/Broccoli_Lian/article/details/84527539
同样的hedshu中的也做一个excel,然后将数据合并到一个excel中。
现在还差一个前缀路径,在每一列添加同样的前缀路径,可以参考https://jingyan.baidu.com/article/8ebacdf066e7f249f75cd577.html
接着将excel另存为txt文件,进而修改后缀为xls,至此就完成了列表读取文件
4、数据集制作完成,修改源文件的几个地方
solver.py中
sys.path.append加载的是caffe中python的路径,改成自己的路径
train_val.prototxt中
修改source为自己的lst
至此就可以训练啦~