Fail with Transfer Learning with Unet Multiclass, Color Images, Semantic Segmentation

May I know if they are two completely different dataset , right? One is grayscale dataset, another is color images dataset?

Yes, its two datasets. The grayscale works well with masks with all pixels in {0, 255} (but does not work on masks with all pixels in {0, 1}) Now I’m trying to go to multiclass segmentation with color images. But I’m trying to build up in baby steps. So before doing the multiclass, I’m trying to do a binary color segmentation.

Making that change results in a bad evaluation with

/usr/local/lib/python3.6/dist-packages/iva/unet/scripts/evaluate.py:80: RuntimeWarning: invalid value encountered in true_divide
/usr/local/lib/python3.6/dist-packages/iva/unet/scripts/evaluate.py:81: RuntimeWarning: invalid value encountered in true_divide
/usr/local/lib/python3.6/dist-packages/iva/unet/scripts/evaluate.py:82: RuntimeWarning: invalid value encountered in true_divide

Here is the spec file:
unet_train_resnet_unet_isbi.txt (1.4 KB)