Skip to content

Y
yolov5
Unverified 提交 d885799c authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者： GitHub
浏览文件
操作

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

上级 360eec69
隐藏空白字符变更
内嵌 并排
正在显示 包含 1 行增加 和 0 行删除
utils/datasets.py
浏览文件 @ d885799c
...@@ -611,6 +611,7 @@ class LoadImagesAndLabels(Dataset): ...@@ -611,6 +611,7 @@ class LoadImagesAndLabels(Dataset):
# Cutouts # Cutouts
# labels = cutout(img, labels, p=0.5) # labels = cutout(img, labels, p=0.5)
# nl = len(labels) # update after cutout
labels_out = torch.zeros((nl, 6)) labels_out = torch.zeros((nl, 6))
if nl: if nl:
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论