论文:holistically-nested edge detection 源码:https://github.com/lc82111/Keras_HED
网络结构图
- 下载训练数据:http://vcl.ucsd.edu/hed/HED-BSDS.tar 并解压到工程根目录下
- 下载预训练模型:https://github.com/fchollet/deep-learning-models/releases 中搜索文件’vgg16_weights_tf_dim_ordering_tf_kernels_notop.h5‘,下载并拷贝到./models目录下
- 执行
python main_segmentation.py
进行训练
loss 收敛曲线图:
- 下载模型文件:链接: https://pan.baidu.com/s/17VjM8xghFHQbSAlymfsf0w 密码: rcxi
- 将文件拷贝到 ./checkpoints/HEDSeg/目录下
- 修改test.py中的模型文件目录
- 执行
python test.py