CannyEdgePytorch

Uses PyTorch 0.4.1 and Python 3.7 (but probably works with 2.7 also).

A simple implementation of the Canny Edge Detection Algorithm (currently without hysteresis).

This project was implemented with PyTorch to take advantage of the parallelization of convolutions.

The original image:

<img src="https://github.com/DCurro/CannyEdgePytorch/blob/master/fb_profile.jpg" width="400">

Finding the gradient magnitude:

<img src="https://github.com/DCurro/CannyEdgePytorch/blob/master/gradient_magnitude.png" width="400">

Early thresholding (to show that edge thingging matters):

<img src="https://github.com/DCurro/CannyEdgePytorch/blob/master/thresholded.png" width="400">

And finally, the image after non-maximum supressions:

<img src="https://github.com/DCurro/CannyEdgePytorch/blob/master/final.png" width="400">