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

Co-authored-by: Glenn Jocher <[email protected]>
tc5z · Jun 9, 2022 · cf298fb · cf298fb
1 parent 2dd3db0
commit cf298fb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utils/general.py b/utils/general.py
@@ -738,6 +738,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