EliasVansteenkiste / edge_detection_framework

Pytorch implementation of different edge detection networks, such as Holistically-Nested Edge Detection and PixelNet
MIT License
33 stars 9 forks source link

Edge detection framework

Pytorch implementation of different edge detection networks, such as

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