diff --git a/fastdeploy/vision/evaluation/classify.py b/fastdeploy/vision/evaluation/classify.py index 66ab47bbd..2d591f744 100644 --- a/fastdeploy/vision/evaluation/classify.py +++ b/fastdeploy/vision/evaluation/classify.py @@ -12,10 +12,8 @@ # See the License for the specific language governing permissions and # limitations under the License. import numpy as np -import cv2 import os import re -from tqdm import trange def topk_accuracy(topk_list, label_list): @@ -25,6 +23,9 @@ def topk_accuracy(topk_list, label_list): def eval_classify(model, image_file_path, label_file_path, topk=5): + from tqdm import trange + import cv2 + result_list = [] label_list = [] image_label_dict = {} diff --git a/fastdeploy/vision/evaluation/detection.py b/fastdeploy/vision/evaluation/detection.py index cd09046f7..4750756b7 100644 --- a/fastdeploy/vision/evaluation/detection.py +++ b/fastdeploy/vision/evaluation/detection.py @@ -12,11 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -from tqdm import trange -import cv2 import numpy as np -from .utils import CocoDetection -from .utils import COCOMetric import copy import collections @@ -27,6 +23,11 @@ def eval_detection(model, conf_threshold=None, nms_iou_threshold=None, plot=False): + from .utils import CocoDetection + from .utils import COCOMetric + import cv2 + from tqdm import trange + if conf_threshold is not None or nms_iou_threshold is not None: assert conf_threshold is not None and nms_iou_threshold is not None, "The conf_threshold and nms_iou_threshold should be setted at the same time" assert isinstance(conf_threshold, ( diff --git a/requirements.txt b/requirements.txt index 7e18ca034..f2293605c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1 @@ -opencv-python -tqdm -numpy -pycocotools -colorama +requests