0%

开源项目 | hed

这是用 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 的网络结构」文件。

这个文件可以在下面的链接中找到。

请我喝杯咖啡吧~