fix(utils): missing edge when resample segments (ultralytics#8092)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
tdhooghe · Jun 10, 2022 · 34d9a4b · 34d9a4b
1 parent df7e780
commit 34d9a4b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utils/general.py b/utils/general.py
@@ -739,6 +739,7 @@ def segments2boxes(segments):
 def resample_segments(segments, n=1000):
     # Up-sample an (n,2) segment
     for i, s in enumerate(segments):
+        s = np.concatenate((s, s[0:1, :]), axis=0)
         x = np.linspace(0, len(s) - 1, n)
         xp = np.arange(len(s))
         segments[i] = np.concatenate([np.interp(x, xp, s[:, i]) for i in range(2)]).reshape(2, -1).T  # segment xy