Spaces:

asalhi85
/

DemoSmartathon

Runtime error

asalhi85 commited on Feb 1, 2023

Commit

2481812

•

1 Parent(s): 3d4b1ac

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ cfg.merge_from_file("./faster_rcnn_X_101_32x8d_FPN_3x.yaml")
 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 11
 cfg.MODEL.WEIGHTS = model_path
-my_metadata = MetadataCatalog.get("")
 my_metadata.thing_classes = ["GRAFFITI", "FADED_SIGNAGE","POTHOLES","GARBAGE","CONSTRUCTION_ROAD","BROKEN_SIGNAGE","BAD_STREETLIGHT","BAD_BILLBOARD","SAND_ON_ROAD","CLUTTER_SIDEWALK","UNKEPT_FACADE"]
 # #smart_dict={'GRAFFITI' : 0.0 , 'FADED_SIGNAGE': 1.0 , 'POTHOLES': 2.0,
 #             'GARBAGE' : 3.0 , 'CONSTRUCTION_ROAD': 4.0 , 'BROKEN_SIGNAGE': 5.0,

 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 11
 cfg.MODEL.WEIGHTS = model_path
+my_metadata = MetadataCatalog.get("dbmdz_coco_all")
 my_metadata.thing_classes = ["GRAFFITI", "FADED_SIGNAGE","POTHOLES","GARBAGE","CONSTRUCTION_ROAD","BROKEN_SIGNAGE","BAD_STREETLIGHT","BAD_BILLBOARD","SAND_ON_ROAD","CLUTTER_SIDEWALK","UNKEPT_FACADE"]
 # #smart_dict={'GRAFFITI' : 0.0 , 'FADED_SIGNAGE': 1.0 , 'POTHOLES': 2.0,
 #             'GARBAGE' : 3.0 , 'CONSTRUCTION_ROAD': 4.0 , 'BROKEN_SIGNAGE': 5.0,