rcf

variety of Richer Convolutional Features for Edge Detection (resnet101-based)

results

ODS: 0.796 , OIS: 0.814 on BSDS500 dataset pretrained model

requirements

• pytorch 0.4.1
• python 3.6.6
• dataset(provide by original rcf repo)
  • http://mftp.mmcheng.net/liuyun/rcf/data/bsds_pascal_train_pair.lst
  • http://mftp.mmcheng.net/liuyun/rcf/data/HED-BSDS.tar.gz
  • http://mftp.mmcheng.net/liuyun/rcf/data/PASCAL.tar.gz
• and other requiremnts... (cv2, numpy , etc.)

usage

train:

• put your data in 'data/HED-BSDS_PASCAL' (or make a soft link)
• python train.py

test:

• python test.py

evaluate:

it may take several hours...

• pretrained model
• requirements:
  • matlab
  • hed link
• you should modify path to your predicts and path to ground truth (.mat)
  • path to your predicts should contain two folders: png and mat
• sh eval.sh

related projects

rcf pytorch implementation rcf hed pytorch implementation