Edge detection framework

October 2, 2017 ยท View on GitHub

Pytorch implementation of different edge detection networks, such as

  • Holistically-Nested Edge Detection (link to caffe implementation)
  • PixelNet (link to caffe implementation)

Training models

python train.py <config name>

Testing trained models

python test.py <valid/test> <config name>

Warning

This is a work in progress, so not every feature is working completely

Contents

  1. 1Training models
  2. 2Testing trained models
  3. 3Warning