[feat] dumping coco instance json for YOLO-Eval such that class-wise mAP computation is available

Author: chyomin06

compressai_vision/evaluators/base_evaluator.py

@@ -69,7 +69,8 @@ def get_jde_eval_info_name(name):
         return f"{name}_info_to_eval.h5"
 
     @staticmethod
-    def get_coco_eval_info_name(name):
+    def get_coco_eval_info_name(name=None):
+        # the not used input is intended to interface with the function computing class-wise peformance
         return "coco_instances_results.json"
 
     def reset(self):

compressai_vision/evaluators/evaluators.py

@@ -690,6 +690,18 @@ def results(self, save_path: str = None):
             self.data_list, dummy_statistics
         )
 
+        assert self.output_dir is not None
+
+        file_path = Path(f"{self.output_dir}")
+        if not file_path.is_dir():
+            self._logger.info(f"creating output folder: {file_path}")
+            file_path.mkdir(parents=True, exist_ok=True)
+            self._logger.info("Saving results to {}".format(file_path))
+
+        with open(f"{file_path}/{self.get_coco_eval_info_name()}", "w") as f:
+            f.write(json.dumps(self.data_list))
+            f.flush()
+
         if save_path:
             self.write_results(eval_results, save_path)