mosidi commited on
Commit
ded1019
·
1 Parent(s): d6495a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ model_path = "./model_final.pth"
70
  # cfg.MODEL.WEIGHTS = model_path
71
 
72
  # my_metadata = MetadataCatalog.get("dbmdz_coco_all")
73
- Fiber_metadata.thing_classes = ["Fiber", "Fiber","Fiber"]
74
  # my_metadata.thing_classes = ["Fiber", "Fiber";]
75
  cfg = get_cfg()
76
  cfg.merge_from_file("./configs/detectron2/mask_rcnn_R_50_FPN_3x.yaml")
 
70
  # cfg.MODEL.WEIGHTS = model_path
71
 
72
  # my_metadata = MetadataCatalog.get("dbmdz_coco_all")
73
+ # Fiber_metadata.thing_classes = ["Fiber", "Fiber","Fiber"]
74
  # my_metadata.thing_classes = ["Fiber", "Fiber";]
75
  cfg = get_cfg()
76
  cfg.merge_from_file("./configs/detectron2/mask_rcnn_R_50_FPN_3x.yaml")