updates
This commit is contained in:
parent
7c2623dd4f
commit
6893f1daf8
2
test.py
2
test.py
|
@ -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:
|
||||
|
|
|
@ -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
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue