소스 검색

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):
     def log(self):
         map_stat = 100. * self.detection_map.get_map()
         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):
     def get_results(self):
         return {'bbox': [self.detection_map.get_map()]}
         return {'bbox': [self.detection_map.get_map()]}
 
 
     def get(self):
     def get(self):
         map_stat = 100. * self.detection_map.get_map()
         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
         return stats