Update app.py

app.py

@@ -24,8 +24,8 @@ model_path = 'model_final.pth'
 
 cfg = get_cfg()
 cfg.merge_from_file(model_zoo.get_config_file("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml"))
-cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.75
-cfg.MODEL.ROI_HEADS.NUM_CLASSES = 19
+cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.6
+cfg.MODEL.ROI_HEADS.NUM_CLASSES = 13
 cfg.MODEL.WEIGHTS = model_path
 
 
@@ -33,23 +33,17 @@ if not torch.cuda.is_available():
     cfg.MODEL.DEVICE='cpu'
     
 predictor = DefaultPredictor(cfg)
-my_metadata = MetadataCatalog.get("car_parts_dataset_val")
-my_metadata.thing_classes = ['_background_',
- 'back_bumper',
+my_metadata = MetadataCatalog.get("car_part_merged_dataset_val")
+my_metadata.thing_classes = ['back_bumper',
  'back_glass',
- 'back_left_door',
- 'back_left_light',
- 'back_right_door',
- 'back_right_light',
+ 'back__door',
+ 'back__light',
  'front_bumper',
  'front_glass',
- 'front_left_door',
- 'front_left_light',
- 'front_right_door',
- 'front_right_light',
+ 'front__door',
+ 'front__light',
  'hood',
- 'left_mirror',
- 'right_mirror',
+ '_mirror',
  'tailgate',
  'trunk',
  'wheel']