fi-ber-detec-api

Sleeping

mosidi commited on Feb 23, 2023

Commit

cf963dd

1 Parent(s): 8e5935d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -51,7 +51,7 @@ model_path = "./model_final.pth"
 # my_metadata = MetadataCatalog.get("dbmdz_coco_all")
 # my_metadata.thing_classes = ["Fiber", "Fiber"]
 cfg = get_cfg()
-cfg.merge_from_file("./configs/detectron2/faster_rcnn_R_50_FPN_3x.yaml")
 cfg.MODEL.WEIGHTS = model_path #os.path.join(cfg.OUTPUT_DIR, "model_final.pth")
 cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.05  # set the testing threshold for this model
 cfg.DATASETS.TEST = ("fiber", )

 # my_metadata = MetadataCatalog.get("dbmdz_coco_all")
 # my_metadata.thing_classes = ["Fiber", "Fiber"]
 cfg = get_cfg()
+cfg.merge_from_file("./configs/detectron2/mask_rcnn_R_50_FPN_3x")
 cfg.MODEL.WEIGHTS = model_path #os.path.join(cfg.OUTPUT_DIR, "model_final.pth")
 cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.05  # set the testing threshold for this model
 cfg.DATASETS.TEST = ("fiber", )