Update image_segmenter.py

image_segmenter.py

@@ -11,6 +11,10 @@ class ImageSegmenter:
         self.model = YOLO('models/'+ model_type +'.pt')
         self.model.to(self.device)
 
+        # Print all categories
+        print("YOLOv8 Categories:")
+        print(self.model.names)
+
         self.is_show_bounding_boxes = True
         self.is_show_segmentation_boundary = False
         self.is_show_segmentation = False