apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-09 02:20:17 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix evaluation import problem (#46)

Browse Source 
* Detection evaluation function

* Add license

* Fix python import problem

* Modify requirement.txt

* Add requirements.txt

Co-authored-by: Jason <jiangjiajun@baidu.com>
This commit is contained in:
huangjianhui
2022-07-27 16:42:45 +08:00
committed by GitHub
parent 01b04363e4
commit 83b8350b96
4 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
fastdeploy/vision/evaluation/utils/__init__.py
View File

@@ -19,4 +19,3 @@ from .json_results import *
from .map_utils import * from .map_utils import *
from .coco_utils import * from .coco_utils import *
from .coco import * from .coco import *
from .cityscapes import Cityscapes

1
fastdeploy/vision/evaluation/utils/coco.py
View File

@@ -15,7 +15,6 @@
from __future__ import absolute_import from __future__ import absolute_import
import copy import copy
import os.path as osp import os.path as osp
import six
import sys import sys
import numpy as np import numpy as np
from . import fd_logging as logging from . import fd_logging as logging

5
fastdeploy/vision/evaluation/utils/coco_utils.py
View File

@@ -18,10 +18,9 @@ from __future__ import print_function
import sys import sys
import numpy as np import numpy as np
import itertools
from .map_utils import draw_pr_curve from .map_utils import draw_pr_curve
from .json_results import get_det_res, get_det_poly_res, get_seg_res, get_solov2_segm_res from .json_results import get_det_res, get_det_poly_res, get_seg_res, get_solov2_segm_res
import logging as logging from . import fd_logging as logging
import copy import copy
@@ -202,6 +201,8 @@ def cocoapi_eval(anns,
file_name='{}_precision_recall_curve.jpg'.format(nm["name"])) file_name='{}_precision_recall_curve.jpg'.format(nm["name"]))
num_columns = min(6, len(results_per_category) * 2) num_columns = min(6, len(results_per_category) * 2)
import itertools
results_flatten = list(itertools.chain(*results_per_category)) results_flatten = list(itertools.chain(*results_per_category))
headers = ['category', 'AP'] * (num_columns // 2) headers = ['category', 'AP'] * (num_columns // 2)
results_2d = itertools.zip_longest( results_2d = itertools.zip_longest(

3
requirements.txt
View File

@@ -1,2 +1,5 @@
opencv-python opencv-python
tqdm tqdm
numpy
pycocotools
colorama