My target image is like this.
I want to calculate L1 loss, but do not want to include black space.
What should I do?
1 Like
You could create a mask using all black pixels to zero out the losses corresponding to these pixels.
After you’ve created the mask, you could calculate the unreduced loss, multiply it with the mask, average and finally call backward
.