Merge pull request IDEA-Research#42 from Greywan/dev

fix:fixed interruptions when there is no mask result for the current …

Author: rentainhe

utils/common_utils.py

@@ -48,6 +48,10 @@ def draw_masks_and_box_with_supervision(raw_image_path, mask_path, json_path, ou
                     object_mask = (mask == uid)
                     all_object_masks.append(object_mask[None])
 
+            if len(all_object_masks) == 0:
+                output_image_path = os.path.join(output_path, raw_image_name)
+                cv2.imwrite(output_image_path, image)
+                continue
             # get n masks: (n, h, w)
             all_object_masks = np.concatenate(all_object_masks, axis=0)