Unverified 提交 b32f67f6 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

`--single-cls` segments fix (#10260)

--single-cls segments fix May resolve #10230 Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
上级 915bbf29
......@@ -537,8 +537,6 @@ class LoadImagesAndLabels(Dataset):
self.segments[i] = segment[j]
if single_cls: # single-class training, merge all classes into 0
self.labels[i][:, 0] = 0
if segment:
self.segments[i][:, 0] = 0
# Rectangular Training
if self.rect:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论