diff --git a/predict_mmdetection.py b/predict_mmdetection.py index 2c82ed595b45a1789b66f43eb05688df46de1432..f70e4ddb3c66617ce63b0c1617c883442953ab6e 100644 --- a/predict_mmdetection.py +++ b/predict_mmdetection.py @@ -164,8 +164,8 @@ class MMDetectionPredictor(FoodChallengePredictor): # with v2.1 dataset. # Please disable if you are submitting model trained on v2.1 dataset #data["category_id"] = self.v2_0_to_v2_1_mapping(data["category_id"]) - if data["category_id"] is not None: - segm_json_results.append(data) + #if data["category_id"] is not None: + segm_json_results.append(data) return segm_json_results def create_test_predictions(self, images_path):