Commit
•
d885799
1
Parent(s):
360eec6
Update `nl` after `cutout()` (#5873)
Browse files- utils/datasets.py +1 -0
utils/datasets.py
CHANGED
|
@@ -611,6 +611,7 @@ class LoadImagesAndLabels(Dataset):
|
|
| 611 |
|
| 612 |
# Cutouts
|
| 613 |
# labels = cutout(img, labels, p=0.5)
|
|
|
|
| 614 |
|
| 615 |
labels_out = torch.zeros((nl, 6))
|
| 616 |
if nl:
|
|
|
|
| 611 |
|
| 612 |
# Cutouts
|
| 613 |
# labels = cutout(img, labels, p=0.5)
|
| 614 |
+
# nl = len(labels) # update after cutout
|
| 615 |
|
| 616 |
labels_out = torch.zeros((nl, 6))
|
| 617 |
if nl:
|