Update app.py

app.py

@@ -66,12 +66,12 @@ model_path = "./model_final.pth"
 
 # cfg = get_cfg()
 # cfg.merge_from_file("./configs/detectron2/faster_rcnn_R_50_FPN_3x.yaml")
-# cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
+cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
 # cfg.MODEL.WEIGHTS = model_path
 
-# my_metadata = MetadataCatalog.get("dbmdz_coco_all")
+my_metadata = MetadataCatalog.get("dbmdz_coco_all")
 # Fiber_metadata.thing_classes = ["Fiber", "Fiber","Fiber"]
-# my_metadata.thing_classes = ["Fiber", "Fiber";]
+my_metadata.thing_classes = ["Fiber", "Fiber";]
 cfg = get_cfg()
 cfg.merge_from_file("./configs/detectron2/mask_rcnn_R_50_FPN_3x.yaml")
 cfg.MODEL.WEIGHTS = model_path #os.path.join(cfg.OUTPUT_DIR, "model_final.pth")
@@ -121,7 +121,7 @@ def inference(image_url, image, min_score):
     # im = cv2.imread(d["file_name"])
     # outputs = predictor(im)
     v = Visualizer(im[:, :, ::-1],
-                    metadata=Fiber_metadata, 
+                    metadata=my_metadata , #Fiber_metadata, 
                     scale=1, 
                     instance_mode=ColorMode.IMAGE_BW   # remove the colors of unsegmented pixels
     )