This commit is contained in:
Glenn Jocher 2019-07-15 16:48:02 +02:00
parent 7c2623dd4f
commit 6893f1daf8
2 changed files with 2 additions and 2 deletions

View File

@ -89,7 +89,7 @@ def test(
continue
# Clip boxes to image bounds
# clip_coords(pred, shapes[si])
clip_coords(pred, (height, width))
# Append to text file
# with open('test.txt', 'a') as file:

View File

@ -130,7 +130,7 @@ def scale_coords(img1_shape, coords, img0_shape):
coords[:, [0, 2]] -= (img1_shape[1] - img0_shape[1] * gain) / 2 # x padding
coords[:, [1, 3]] -= (img1_shape[0] - img0_shape[0] * gain) / 2 # y padding
coords[:, :4] /= gain
# clip_coords(coords, img0_shape)
clip_coords(coords, img0_shape)
return coords