[feat] NNpart2 hyperparameters and mse calculation

Author: jianwensong

README.md

@@ -157,4 +157,4 @@ Fabien Racapé, Hyomin Choi, Eimran Eimon, Sampsa Riikonen, Jacky Yat-Hong Lam
  * [JDE](https://github.com/Zhongdao/Towards-Realtime-MOT.git)
  * [YOLOX](https://github.com/Megvii-BaseDetection/YOLOX)
  * [MMPOSE RTMO](https://github.com/open-mmlab/mmpose/tree/main/projects/rtmo)
- * [Segment Anything](https://github.com/facebookresearch/segment-anything/tree/main)
+ * [Segment Anything](https://github.com/facebookresearch/segment-anything/tree/main)

cfgs/evaluator/default.yaml

@@ -1,4 +1,5 @@
 type: "COCO-EVAL"
 output_dir: "${pipeline.evaluation.evaluation_dir}"
 overwrite_results: False
+calculate_feature_mse: False
 eval_criteria: ""

compressai_vision/evaluators/base_evaluator.py

@@ -81,7 +81,7 @@ def get_miou_eval_info_name(name):
     def reset(self):
         raise NotImplementedError
 
-    def digest(self, gt, pred):
+    def digest(self, gt, pred, mse_results=None):
         raise NotImplementedError
 
     def results(self, save_path: str = None):