sobel-operator-pytorch icon indicating copy to clipboard operation
sobel-operator-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon chaddy1004
→

Metadata

Sobel edge detection implemented on PyTorch

Results 2 sobel-operator-pytorch issues
Sort by recently updated
recently updated
newest added

Add epsilon

Hi, thanks for the code. you need to add to the sqrt, small number epsilon. Otherwise you will get in the backpropogation nan values. https://github.com/chaddy1004/sobel-operator-pytorch/blob/eca741aafbcef9d2639acedca10b09bbfa100e09/model.py#L20

MordehayM avatar MordehayM

padding=0 result in a reduced image size

Hi, thanks for the work. I just noticed that the padding should be changed to 1, otherwise the image size will be reduced https://github.com/chaddy1004/sobel-operator-pytorch/blob/eca741aafbcef9d2639acedca10b09bbfa100e09/model.py#L8C3-L8C3

chengengliu avatar chengengliu

← Metadata

32
Stars
7
Forks
Watchers

Owner

verified publisher icon chaddy1004

Metadata

Sobel edge detection implemented on PyTorch

Back