
MATLAB的Edge源代码-HED:使用PyTorch的整体嵌套边缘检测重新实现
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目为基于PyTorch框架对HED(Hierarchy Edge Detection)算法的复现工作。该算法通过深层网络学习,实现了高质量的整体边缘检测效果。
本段落介绍了MATLAB的edge源代码HED在PyTorch中的重新实现。该代码已在PyTorch1.0(CUDA9,CUDNN7)与MATLAB R2018b环境下使用Python3.6进行了评估。
准备步骤:
- 克隆存储库:git clone https://github.com/xwjabc/hed.git
- 下载并解压数据:wget https://cseweb.ucsd.edu/~weijian/static/datasets/hed/hed-data.tar;tar xvf ./hed-data.tar
操作指南:
- 训练和评估:cd hed,python hed.py --vgg16_caffe ./data/5stage_vgg_py36.pickle
- 结果存储在output文件夹中。
- 默认设置下,HED模型训练40个周期,在NVIDIA GeForce GTX Titan X(Maxwell)上大约需要27小时完成。
评估指令:
cd eval;echo data_dir=../output/epoch-39-test | cat eval
全部评论 (0)
还没有任何评论哟~


