Fix evaluation import problem (#46)

* Detection evaluation function

* Add license

* Fix python import problem

* Modify requirement.txt

* Add requirements.txt

Co-authored-by: Jason <jiangjiajun@baidu.com>

fastdeploy/vision/evaluation/utils/__init__.py

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

fastdeploy/vision/evaluation/utils/coco.py

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

fastdeploy/vision/evaluation/utils/coco_utils.py

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

requirements.txt

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