瀏覽代碼

Merge pull request #903 from will-jl944/develop_jf

change det eval metric keys
FlyingQianMM 4 年之前
父節點
當前提交
2cffea651a
共有 1 個文件被更改,包括 2 次插入6 次删除
  1. 2 6
      dygraph/paddlex/cv/models/utils/det_metrics/metrics.py

+ 2 - 6
dygraph/paddlex/cv/models/utils/det_metrics/metrics.py

@@ -139,18 +139,14 @@ class VOCMetric(Metric):
 
     def log(self):
         map_stat = 100. * self.detection_map.get_map()
-        logging.info("mAP({:.2f}, {}) = {:.2f}%".format(
-            self.overlap_thresh, self.map_type, map_stat))
+        logging.info("bbox_map = {:.2f}%".format(map_stat))
 
     def get_results(self):
         return {'bbox': [self.detection_map.get_map()]}
 
     def get(self):
         map_stat = 100. * self.detection_map.get_map()
-        stats = {
-            "mAP({:.2f}, {})".format(self.overlap_thresh, self.map_type):
-            map_stat
-        }
+        stats = {"bbox_map": map_stat}
         return stats