pytorch-hed - Holistically-Nested Edge Detection based on Pytorch
Project description
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}}
}
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
pytorch-hed-0.4.tar.gz
(5.3 kB
view hashes)