Improve normalization when generating instance segmentation model

model_server/extensions/ilastik/models.py

@@ -252,7 +252,12 @@ class IlastikObjectClassifierFromPixelPredictionsModel(IlastikModel, ImageToImag
             def label_instance_class(
                     self, img: GenericImageDataAccessor, mask: GenericImageDataAccessor, **kwargs
             ) -> GenericImageDataAccessor:
-                return super().label_instance_class(img, mask, pixel_classification_channel=px_ch)
+                if mask.dtype == 'bool':
+                    norm_mask = 1.0 * mask.data
+                else:
+                    norm_mask = mask.data / np.iinfo(mask.dtype).max
+                norm_mask_acc = InMemoryDataAccessor(norm_mask.astype('float32'))
+                return super().label_instance_class(img, norm_mask_acc, pixel_classification_channel=px_ch)
         return _Mod(params={'project_file': self.project_file})