Just an experimental version to test masked loss. The result seems to be better than the original, it is less overcooked and the masked loss removed a bit of the overwhelming background.
Just an experimental version to test masked loss. The result seems to be better than the original, it is less overcooked and the masked loss removed a bit of the overwhelming background.