浏览代码

add coco error analysis

will-jl944 4 年之前
父节点
当前提交
65606768e2
共有 1 个文件被更改,包括 2 次插入0 次删除
  1. 2 0
      paddlex/det.py

+ 2 - 0
paddlex/det.py

@@ -15,6 +15,7 @@
 import sys
 from . import cv
 from .cv.models.utils.visualize import visualize_detection, draw_pr_curve
+from .cv.models.utils.det_metrics.coco_utils import coco_error_analysis
 
 message = 'Your running script needs PaddleX<2.0.0, please refer to {} to solve this issue.'.format(
     'https://github.com/PaddlePaddle/PaddleX/tree/release/2.0-rc/tutorials/train#%E7%89%88%E6%9C%AC%E5%8D%87%E7%BA%A7'
@@ -31,6 +32,7 @@ def __getattr__(attr):
 
 visualize = visualize_detection
 draw_pr_curve = draw_pr_curve
+coco_error_analysis = coco_error_analysis
 
 # detection
 YOLOv3 = cv.models.YOLOv3