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

updates

This commit is contained in:
Glenn Jocher 2019-12-08 12:26:31 -08:00
parent b81beb0f5f
commit f373764e4d
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/datasets.py
View File

@ -677,8 +677,8 @@ def random_affine(img, targets=(), degrees=10, translate=.1, scale=.1, shear=10,
# xy = np.concatenate((x - w / 2, y - h / 2, x + w / 2, y + h / 2)).reshape(4, n).T # xy = np.concatenate((x - w / 2, y - h / 2, x + w / 2, y + h / 2)).reshape(4, n).T
# reject warped points outside of image # reject warped points outside of image
xy[:, [0, 2]] = xy[:, [0, 2]].clip(0, width) np.clip(xy[:, [0, 2]], 0, width, out=xy[:, [0, 2]])
xy[:, [1, 3]] = xy[:, [1, 3]].clip(0, height) np.clip(xy[:, [1, 3]], 0, height, out=xy[:, [1, 3]])
w = xy[:, 2] - xy[:, 0] w = xy[:, 2] - xy[:, 0]
h = xy[:, 3] - xy[:, 1] h = xy[:, 3] - xy[:, 1]
area = w * h area = w * h