https://dev.to/masonandrewharrison/i-build-a-pytorch-edge-detection-package-24dm