Add Exception for image size not matching configured size, and Warning for cell boxes outside the image

Author: thouis

deepprofiler/imaging/cropping.py

@@ -332,6 +332,15 @@ def prepare_image(self, session, image_array, meta, sample_first_crops=False):
             self.input_variables["mask_ind_ph"]: mask_ind
         }
 
+        # check that all boxes overlap the image
+        ymins = boxes[:, [0, 2]].min(axis=1)
+        ymaxs = boxes[:, [0, 2]].max(axis=1)
+        xmins = boxes[:, [1, 3]].min(axis=1)
+        xmaxs = boxes[:, [1, 3]].max(axis=1)
+        if (np.any(ymins > 1) or np.any(xmins > 1) or
+            np.any(ymaxs < 0) or np.any(ymaxs < 0)):
+            print("WARNING: Some cell boxes are entirely outside the image")
+
         for i in range(num_targets):
             tname = "target_" + str(i)
             feed_dict[self.input_variables["targets_phs"][tname]] = targets[i]

deepprofiler/learning/profiling.py

@@ -82,6 +82,16 @@ def extract_features(self, key, image_array, meta):  # key is a placeholder
         if total_crops == 0:
             print("No cells to profile:", output_file)
             return
+
+        # check image size matches config
+        if (self.config["dataset"]["images"]["width"] != image_array.shape[1] or
+            self.config["dataset"]["images"]["height"] != image_array.shape[0]):
+            config_shape = (self.config["dataset"]["images"]["width"],
+                            self.config["dataset"]["images"]["height"])
+            im_shape = (image_array.shape[1], image_array.shape[0])
+            raise ValueError("Loaded image shape WxH " + str(im_shape) +
+                             " != configured image shape WxH " + str(config_shape))
+
         repeats = self.config["train"]["model"]["crop_generator"] in ["repeat_channel_crop_generator", "individual_channel_cropgen"]
 
         # Extract features