diff --git a/utils/datasets.py b/utils/datasets.py index 6ce7a81b69e7..5a3b2110b2e0 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -611,6 +611,7 @@ def __getitem__(self, index): # Cutouts # labels = cutout(img, labels, p=0.5) + # nl = len(labels) # update after cutout labels_out = torch.zeros((nl, 6)) if nl: