matlab的egde源代码-hed:整体嵌套边缘检测的PyTorch重新实现

matlab的egde源代码HED的PyTorch重新实现 介绍 这是的PyTorch重新实现。 该代码在PyTorch 1.0（CUDA9，CUDNN7）和MATLAB R2018b的Python 3.6上进行了评估。 指示 准备 克隆存储库： git clone https://github.com/xwjabc/hed.git 下载并提取数据： cd hed wget https://cseweb.ucsd.edu/~weijian/static/datasets/hed/hed-data.tar tar xvf ./hed-data.tar 培训和评估 火车： python hed.py --vgg16_caffe ./data/5stage-vgg.py36pickle 结果在output文件夹中。 在默认设置下，HED模型训练了40个时期，使用一台NVIDIA Geforce GTX Titan X（Maxwell）大约需要27个小时。 评估： cd eval (echo " data_dir = '../output/epoch-39-test' " ; cat eval