HRan2004 glenn-jocher commited on
Commit
cf298fb
·
unverified ·
1 Parent(s): 2dd3db0

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

Browse files

Co-authored-by: Glenn Jocher <[email protected]>

Files changed (1) hide show
  1. 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