From 582de735ad7f27463572774747c390db0593dfcb Mon Sep 17 00:00:00 2001 From: GoogleWiki Date: Sat, 28 Mar 2020 04:09:10 +0800 Subject: [PATCH] utils.clip_coords doesn't work as expected. (#961) * utils.clip_coords doesn't work as expected. Box coords may be negative or exceed borders. * Update utils.py Co-authored-by: Glenn Jocher --- utils/utils.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/utils/utils.py b/utils/utils.py index f8f192b4..c2ac95fc 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -160,8 +160,10 @@ def scale_coords(img1_shape, coords, img0_shape, ratio_pad=None): def clip_coords(boxes, img_shape): # Clip bounding xyxy bounding boxes to image shape (height, width) - boxes[:, [0, 2]].clamp_(0, img_shape[1]) # clip x - boxes[:, [1, 3]].clamp_(0, img_shape[0]) # clip y + boxes[:, 0].clamp_(0, img_shape[1]) # x1 + boxes[:, 1].clamp_(0, img_shape[0]) # y1 + boxes[:, 2].clamp_(0, img_shape[1]) # x2 + boxes[:, 3].clamp_(0, img_shape[0]) # y2 def ap_per_class(tp, conf, pred_cls, target_cls):