Thank you for response.
I wanted to perform neural style transfer, however instead of content and style loss functions, i’ve intended to use a loss that has distance transform in it. Numpy function is here: Distance transform 36, but it seems PyTorch doesn’t have it yet.
I’ll try the tutorial though.