Numerical stability fix for Albumentations (#3958)
This commit is contained in:
1 parent
443af8b25a
commit
80299a57e2
2 files changed
+12
-14
No files matched your search
+1
-1
@@ -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
|
||||
|
||||
Reference in new issue
Block a user