--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:
Glenn Jocher authored and GitHub committed 2022-11-22 20:30:14 +01:00
1 parent 915bbf294b
commit b32f67f6be
1 file changed
-2
-2
View File
@@ -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: