Update python_utils/get_model.py

python_utils/get_model.py

@@ -79,7 +79,7 @@ def apply_nms(prediction, cls_agnostic_nms=0.85, mask=False):
     pred = {"instances": Instances(image_size=prediction["instances"].image_size,
                                    pred_boxes=prediction["instances"].pred_boxes[nms_indices],
                                    scores=prediction["instances"].scores[nms_indices],
-                                   pred_classes=prediction["instances"].pred_classes[nms_indices]+1,
+                                   pred_classes=prediction["instances"].pred_classes[nms_indices],
                                    pred_masks=prediction["instances"].pred_masks[nms_indices])}
 
     return pred