ashukid / hed-edge-detector

Using pretrained hed edge detector model from caffe with opencv

Code for edge detection using pretrained hed model(caffe) using OpenCV

Command to run the edge detection model on video

python edge.py --input video.mp4 --prototxt deploy.prototxt --caffemodel hed_pretrained_bsds.caffemodel 
--width 300 --height 300

Command to run the edge detection model on image

python edge_detector.py --input image.png --prototxt deploy.prototxt --caffemodel hed_pretrained_bsds.caffemodel
--width 300 --height 300