From 627aae79da6e4eaede2e38a887477ebef3e05729 Mon Sep 17 00:00:00 2001 From: Yonghye Kwon Date: Sun, 28 Feb 2021 17:54:47 +0900 Subject: [PATCH] change package import method for coherence https://github.com/Tianxiaomo/pytorch-YOLOv4/blob/4ccef0ec8fe984e059378813e33b3740929e0c19/tool/darknet2pytorch.py#L4-L7 --- tool/torch_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tool/torch_utils.py b/tool/torch_utils.py index 68cbfb27..56e1a5db 100644 --- a/tool/torch_utils.py +++ b/tool/torch_utils.py @@ -10,7 +10,7 @@ import struct # get_image_size import imghdr # get_image_size -from tool import utils +from tool.utils import post_processing def bbox_ious(boxes1, boxes2, x1y1x2y2=True): @@ -100,5 +100,5 @@ def do_detect(model, img, conf_thresh, nms_thresh, use_cuda=1): print(' Model Inference : %f' % (t2 - t1)) print('-----------------------------------') - return utils.post_processing(img, conf_thresh, nms_thresh, output) + return post_processing(img, conf_thresh, nms_thresh, output)