Update `nl` after `cutout()` (#5873)

2021-12-03 15:28:14 +01:00 · 2021-12-03 15:28:14 +01:00 · d885799c71
parent 360eec6910
commit d885799c71
1 changed files with 1 additions and 0 deletions
--- a/utils/datasets.py
+++ b/utils/datasets.py
@ -611,6 +611,7 @@ class LoadImagesAndLabels(Dataset):

            # Cutouts
            # labels = cutout(img, labels, p=0.5)
+            # nl = len(labels)  # update after cutout

        labels_out = torch.zeros((nl, 6))
        if nl: