pytorch-hed 0.1

pytorch-hed - Holistically-Nested Edge Detection based on Pytorch

Pytorch Holistically-Nested Edge Detection (HED)

This is a reimplementation in the form of a python package of Holistically-Nested Edge Detection [1] using PyTorch based on the previous pytorch implementation by sniklaus [2]. If you would like to use of this work, please cite the paper accordingly. Also, make sure to adhere to the licensing terms of the authors. Moreover, if you will be making use of this particular implementation, please acknowledge the present [3] implementation.

	GitHub	Ref
Original version based on Caffe	https://github.com/s9xie/hed	[1]
Another reimplementation based on Caffe	https://github.com/zeakey/hed
Original reimplementation based on PyTorch	https://github.com/sniklaus/pytorch-hed	[2]

Usage

First, you have to install the package with

pip install pytorch-hed

or

git clone https://github.com/Davidelanz/pytorch-hed.git
pip install ./pytorch-hed

To run it on your own image, use the following function:

import torchHED
torchHED.process_img("path/to/input/image.png", "path/to/output/image.png")

Results

Input	Original Caffe Implementation [1]	pytorch-hed [3]

References

[1]  @inproceedings{Xie_ICCV_2015,
         author = {Saining Xie and Zhuowen Tu},
         title = {Holistically-Nested Edge Detection},
         booktitle = {IEEE International Conference on Computer Vision},
         year = {2015}
     }

[2]  @misc{pytorch-hed,
         author = {Simon Niklaus},
         title = {A Reimplementation of {HED} Using {PyTorch}},
         year = {2018},
         howpublished = {\url{https://github.com/sniklaus/pytorch-hed}}
    }

[3]  @misc{pytorch-hed-2,
         author = {Davide Lanza},
         title = {The {pytorch-hed} Python Package},
         year = {2020},
         howpublished = {\url{https://github.com/Davidelanz/pytorch-hed}}
    }