• Stars
    star
    451
  • Rank 96,968 (Top 2 %)
  • Language
    Python
  • License
    GNU General Publi...
  • Created about 6 years ago
  • Updated 12 months ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

a reimplementation of Holistically-Nested Edge Detection in PyTorch

pytorch-hed

This is a personal reimplementation of Holistically-Nested Edge Detection [1] using PyTorch. Should you be making use of this work, please cite the paper accordingly. Also, make sure to adhere to the licensing terms of the authors. Should you be making use of this particular implementation, please acknowledge it appropriately [2].

Paper

For the original version of this work, please see: https://github.com/s9xie/hed
For another reimplementation based on Caffe, please see: https://github.com/zeakey/hed

usage

To run it on your own image, use the following command. Please make sure to see their paper / the code for more details.

python run.py --model bsds500 --in ./images/sample.png --out ./out.png

I am afraid that this reimplementation is not entirely true to the original Caffe version, even though it utilizes the official weights. It achieves an ODS=0.774 versus the official ODS=0.780 on the BSDS500 dataset, evaluated using this code. Please feel free to contribute to this repository by submitting issues and pull requests.

comparison

Comparison

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}}
    }

More Repositories

1

3d-ken-burns

an implementation of 3D Ken Burns Effect from a Single Image using PyTorch
Python
1,497
star
2

sepconv-slomo

an implementation of Video Frame Interpolation via Adaptive Separable Convolution using PyTorch
Python
1,008
star
3

pytorch-pwc

a reimplementation of PWC-Net in PyTorch that matches the official Caffe version
Python
602
star
4

softmax-splatting

an implementation of softmax splatting for differentiable forward warping using PyTorch
Python
436
star
5

pytorch-liteflownet

a reimplementation of LiteFlowNet in PyTorch that matches the official Caffe version
Python
400
star
6

pytorch-spynet

a reimplementation of Optical Flow Estimation using a Spatial Pyramid Network in PyTorch
Python
298
star
7

wasm-raytracer

a performance comparison of a simple raytracer in JavaScript, asm.js, WebAssembly, and GLSL
HTML
168
star
8

pytorch-unflow

a reimplementation of UnFlow in PyTorch that matches the official TensorFlow version
Python
143
star
9

youtube-watchmarker

a browser extension that keeps track of your YouTube watch history and marks videos that you have already watched
JavaScript
141
star
10

pytorch-extension

an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors
Python
118
star
11

revisiting-sepconv

an implementation of Revisiting Adaptive Convolutions for Video Frame Interpolation using PyTorch
Python
77
star
12

arxiv-doom

a parody of the ever-increasing amount of papers that appear on arXiv
HTML
32
star
13

teaching-vision

the framework for my computer vision class, in which the students are ought to solve various exercises
Python
24
star
14

teaching-webdev

the framework for my full stack web development class, in which the students are ought to solve various exercises
HTML
22
star
15

bookmark-tab

JavaScript
14
star
16

teaching-minichess

the framework for my advanced artificial intelligence class, in which an artificial chess player is ought to be implemented
C
14
star
17

nes-memoryview

visualizing the value of each individual byte in an emulated NES as a time series
HTML
9
star
18

teaching-confour

the framework for my advanced artificial intelligence class, in which an connect-four player is ought to be implemented
C
3
star
19

resume

my personal resume written in LaTeX for others to use
HTML
1
star