Numerical stability fix for Albumentations (#3958)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-07-10 19:50:53 +02:00
1 parent 443af8b25a
commit 80299a57e2
2 files changed
+12 -14

No files matched your search

+1 -1
View File
@@ -550,7 +550,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
nl = len(labels) # number of labels
if nl:
labels[:, 1:5] = xyxy2xywhn(labels[:, 1:5], w=img.shape[1], h=img.shape[0]) # xyxy to xywh normalized
labels[:, 1:5] = xyxy2xywhn(labels[:, 1:5], w=img.shape[1], h=img.shape[0], clip=True, eps=1E-3)
if self.augment:
# Albumentations