--single-cls segments fix (#10260)
--single-cls segments fix May resolve #10230 Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
1 parent
915bbf294b
commit
b32f67f6be
1 file changed
-2
@@ -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:
|
||||
|
||||
Reference in new issue
Block a user