这是用 caffe
写的边缘检测项目,配置和运行环境就不说了, caffe
永远的噩梦。
但是,我使用它来对 pix2pix
系列,来进行边缘检测。
相关
在这里注意的是,上面的脚本是利用 hed 进行批处理写的边缘检测脚本。其使用的 model
来自于。
model
我已经上传到百度云盘。在源码中可以看到,这个 model
的名字叫做 hed_pretrained_bsds.caffemodel
根据项目说明
Extracting Edges
We provide python and Matlab scripts to extract coarse edges from photos. Run scripts/edges/batch_hed.py to compute HED edges. Run scripts/edges/PostprocessHED.m to simplify edges with additional post-processing steps. Check the code documentation for more details.
batch_hed.py
对应于python
PostprocessHED.m
对应于matlab
观看 batch_hed.py
发现有这么一句话
download the models and protoxt, and put them under {caffe_root}/examples/hed/
也就是,要使这个脚本运行起来,需要两个东西,一个是上面的 model
一个是名为 deploy.prototxt
「caffe 的网络结构」文件。
这个文件可以在下面的链接中找到。