From b32f67f6beb4a921c98301fe7724003e23103728 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 22 Nov 2022 20:30:14 +0100
Subject: [PATCH] `--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>
---
 utils/dataloaders.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/utils/dataloaders.py b/utils/dataloaders.py
index e107d1a2bcc..cc5f8843ef1 100644
--- 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: