Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Update nl after cutout() (#5873)

Browse Source
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-12-03 15:28:14 +01:00
1 parent 360eec6910
commit d885799c71
1 file changed
+1
Download Patch File Download Diff File Expand all files Collapse all files
utils/datasets.py
+1
View File
@@ -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: