Spaces:

asalhi85
/

DemoSmartathon

Sleeping

asalhi85 commited on Feb 1, 2023

Commit

64b01d4

•

1 Parent(s): bbda759

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ from detectron2.data import MetadataCatalog
 model_path = "https://huggingface.co/dbmdz/detectron2-model/resolve/main/model_final.pth"
 cfg = get_cfg()
-cfg.merge_from_file("./faster_rcnn_R_50_FPN_3x.yaml")
 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
 cfg.MODEL.WEIGHTS = model_path

 model_path = "https://huggingface.co/dbmdz/detectron2-model/resolve/main/model_final.pth"
 cfg = get_cfg()
+cfg.merge_from_file("./faster_rcnn_X_101_32x8d_FPN_3x.yaml")
 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
 cfg.MODEL.WEIGHTS = model_path