Efficient way to relabel masks?

[ptrampert (Patrick Trampert)]

Just use this:

Unet Segmentation Mask Converter to help against common errors / problems

You do not need a loss function, it is just a mask values conversion tool.

Hope it helps