Update segment2box() comment (ultralytics#2600)

(cherry picked from commit d4456e4)
Lechtr · Mar 28, 2021 · 8371f64 · 8371f64
1 parent 5eb4500
commit 8371f64
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/general.py b/utils/general.py
@@ -289,7 +289,7 @@ def segment2box(segment, width=640, height=640):
     x, y = segment.T  # segment xy
     inside = (x >= 0) & (y >= 0) & (x <= width) & (y <= height)
     x, y, = x[inside], y[inside]
-    return np.array([x.min(), y.min(), x.max(), y.max()]) if any(x) else np.zeros((1, 4))  # cls, xyxy
+    return np.array([x.min(), y.min(), x.max(), y.max()]) if any(x) else np.zeros((1, 4))  # xyxy
 
 
 def segments2boxes(segments):