Hugging Face's logo

Spaces:
mosidi
/
fi-ber-detec-api
Sleeping

App Files Community
1
mosidi commited on
Commit
715faf9
·
1 Parent(s): 8df9be6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
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")
77
  cfg.MODEL.WEIGHTS = model_path #os.path.join(cfg.OUTPUT_DIR, "model_final.pth")
 
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")
77
  cfg.MODEL.WEIGHTS = model_path #os.path.join(cfg.OUTPUT_DIR, "model_final.pth")