Spaces:

ruiite
/

car_damage_detection

Runtime error

ruiite commited on Oct 5, 2022

Commit

3f676f8

•

1 Parent(s): be13df8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ from detectron2.data import MetadataCatalog
 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.SCORE_THRESH_TEST = 0.5
 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 1
 cfg.MODEL.WEIGHTS = model_path

 model_path = 'model_final.pth'
 cfg = get_cfg()
+cfg.merge_from_file("./configs/faster_rcnn_R_50_FPN_3x.yaml")
 cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5
 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 1
 cfg.MODEL.WEIGHTS = model_path