From ac30294ed994c429d5326909e98852a57911328a Mon Sep 17 00:00:00 2001 From: Huu Quan Date: Tue, 17 Aug 2021 22:58:42 +0900 Subject: [PATCH 1/2] fix missing labels after augmentation --- utils/datasets.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/utils/datasets.py b/utils/datasets.py index 52b028994325..5997de72dc96 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -569,6 +569,8 @@ def __getitem__(self, index): # Albumentations img, labels = self.albumentations(img, labels) + nl = len(labels) # update number of labels after augmentation + # HSV color-space augment_hsv(img, hgain=hyp['hsv_h'], sgain=hyp['hsv_s'], vgain=hyp['hsv_v']) From dbfdf250b77eda8ad6c4d6b20b5ef2cbcb58facb Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 18 Aug 2021 12:04:19 +0200 Subject: [PATCH 2/2] Update datasets.py Cleanup --- utils/datasets.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/utils/datasets.py b/utils/datasets.py index 5997de72dc96..25a2ba6f9561 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -568,8 +568,7 @@ def __getitem__(self, index): if self.augment: # Albumentations img, labels = self.albumentations(img, labels) - - nl = len(labels) # update number of labels after augmentation + nl = len(labels) # update after albumentations # HSV color-space augment_hsv(img, hgain=hyp['hsv_h'], sgain=hyp['hsv_s'], vgain=hyp['hsv_v'])