fix(utils): missing edge when resample segments (#8092)
Browse filesCo-authored-by: Glenn Jocher <[email protected]>
- utils/general.py +1 -0
utils/general.py
CHANGED
@@ -738,6 +738,7 @@ def segments2boxes(segments):
|
|
738 |
def resample_segments(segments, n=1000):
|
739 |
# Up-sample an (n,2) segment
|
740 |
for i, s in enumerate(segments):
|
|
|
741 |
x = np.linspace(0, len(s) - 1, n)
|
742 |
xp = np.arange(len(s))
|
743 |
segments[i] = np.concatenate([np.interp(x, xp, s[:, i]) for i in range(2)]).reshape(2, -1).T # segment xy
|
|
|
738 |
def resample_segments(segments, n=1000):
|
739 |
# Up-sample an (n,2) segment
|
740 |
for i, s in enumerate(segments):
|
741 |
+
s = np.concatenate((s, s[0:1, :]), axis=0)
|
742 |
x = np.linspace(0, len(s) - 1, n)
|
743 |
xp = np.arange(len(s))
|
744 |
segments[i] = np.concatenate([np.interp(x, xp, s[:, i]) for i in range(2)]).reshape(2, -1).T # segment xy
|