Переглянути джерело

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

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
modifyDataloader
HRan2004 GitHub 2 роки тому
джерело
коміт
cf298fb72f
Не вдалося знайти GPG ключ що відповідає даному підпису Ідентифікатор GPG ключа: 4AEE18F83AFDEB23
1 змінених файлів з 1 додано та 0 видалено
  1. +1
    -0
      utils/general.py

+ 1
- 0
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

Завантаження…
Відмінити
Зберегти