From 4a3b51c718e7a7de0efc7a68a4cfa6cf80f0e104 Mon Sep 17 00:00:00 2001 From: zhiqiang Date: Sat, 21 Aug 2021 01:28:52 +0800 Subject: [PATCH] Fixing lint --- yolort/utils/image_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/yolort/utils/image_utils.py b/yolort/utils/image_utils.py index 838416045..ea514d7fe 100644 --- a/yolort/utils/image_utils.py +++ b/yolort/utils/image_utils.py @@ -161,7 +161,7 @@ def non_max_suppression(prediction, conf_thres=0.25, iou_thres=0.45, classes=Non assert 0 <= iou_thres <= 1, f'Invalid IoU {iou_thres}, valid values are between 0.0 and 1.0' # Settings - min_wh, max_wh = 2, 4096 # (pixels) minimum and maximum box width and height + max_wh = 4096 # (pixels) minimum and maximum box width and height max_nms = 30000 # maximum number of boxes into torchvision.ops.nms() time_limit = 10.0 # seconds to quit after redundant = True # require redundant detections @@ -238,6 +238,7 @@ def non_max_suppression(prediction, conf_thres=0.25, iou_thres=0.45, classes=Non return output + def get_image_from_url( url: str, flags: int = cv2.IMREAD_COLOR,