Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

inplace clip_coords() clamp

This commit is contained in:
Glenn Jocher 2020-03-13 10:47:00 -07:00
parent 731305142b
commit 6aae5aca64
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/utils.py
View File

@ -160,8 +160,8 @@ 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]] = boxes[:, [0, 2]].clamp(min=0, max=img_shape[1]) # clip x
boxes[:, [1, 3]] = boxes[:, [1, 3]].clamp(min=0, max=img_shape[0]) # clip y
boxes[:, [0, 2]].clamp_(0, img_shape[1]) # clip x
boxes[:, [1, 3]].clamp_(0, img_shape[0]) # clip y
def ap_per_class(tp, conf, pred_cls, target_cls):