diff --git a/utils/datasets.py b/utils/datasets.py index 6ce7a81b6..5a3b2110b 100755 --- 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: