EliasVansteenkiste/edge_detection_framework

Pytorch implementation of different edge detection networks, such as Holistically-Nested Edge Detection and PixelNet

Edge detection framework

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