Update segment.py

segment.py

@@ -54,7 +54,6 @@ def draw_panoptic_segmentation(segmentation, segments_info,save_folder=None, nos
             mask = segmentation==0
             mask = mask.cpu().detach().numpy()   # [512,512]   bool
             print(mask.shape)
-            mask = cv2.resize(mask,(512,512))
             segment_label = "rest"
             color = viridis(0)
             label = f"{segment_label}-{0}"
@@ -69,7 +68,6 @@ def draw_panoptic_segmentation(segmentation, segments_info,save_folder=None, nos
                 segment_id -= 1
             mask = mask.cpu().detach().numpy()   # [512,512] bool
             print(mask.shape)
-            mask = cv2.resize(mask,(512,512))
             mask_np_list.append(mask) 
             segment_label = model.config.id2label[segment['label_id']]
             instances_counter[segment['label_id']] += 1
@@ -82,7 +80,6 @@ def draw_panoptic_segmentation(segmentation, segments_info,save_folder=None, nos
     else:
         mask = np.full(segmentation.shape, True)
         print(mask.shape)
-        mask = cv2.resize(mask,(512,512))
             
         segment_label = "all"
         mask_np_list.append(mask)