Home

Awesome

hed_pytorch

Pytorch implementation of Holistically-nested Edge Detection (HED). This repo can be used to train a HED model and perform test inference. The implementation was evaluated on the BSDS dataset.

Prerequisites

I use docker to avoid dependency problems. Installation instructions for my setup are available here.

Usage

Performance

Files

Acknowledgement

The source code is derived from three different implementations available online. Thanks to @s9xie for original caffe implementation. Thanks to @EliasVansteenkiste, @xlliu, @BinWang-shu for the pytorch implementations.