Remove one data validation check from Polaris (#66)

vanvalenlab · Jun 7, 2023 · 58c7447 · 58c7447
1 parent 681dcec
commit 58c7447
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/deepcell_spots/applications/polaris.py b/deepcell_spots/applications/polaris.py
@@ -244,11 +244,6 @@ def _validate_prediction_input(self,
                                  'has shape {} and segmentation_image has shape {}'
                                  ''.format(spots_image.shape, segmentation_image.shape))
 
-            if segmentation_image.shape[-1] != 1:
-                raise ValueError('Shape of channel dimension of segmentation_image should equal '
-                                 'to zero, but input segmentation_image had shape {} (b,x,y,c).'
-                                 ''.format(segmentation_image.shape))
-
         if background_image is not None:
             if spots_image.shape[:-1] != background_image.shape[:-1]:
                 raise ValueError('Batch, x, and y dimensions of spots_image '