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

--single-cls segments fix May resolve ultralytics#10230 Signed-off-by: Glenn Jocher <[email protected]> Signed-off-by: Glenn Jocher <[email protected]>
yourapple · Nov 22, 2022 · b32f67f · b32f67f
1 parent 915bbf2
commit b32f67f
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/utils/dataloaders.py b/utils/dataloaders.py
@@ -537,8 +537,6 @@ def __init__(self,
                     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: