LibreMEMFOFm-flow-kitti

Переупаковка чекпоинта MEMFOF (Tartan-T-TSKH-kitti) — файнтюна на бенчмарке KITTI — под библиотеку LibreYOLO. Модель читает кадры t-1, t и t+1 и предсказывает обратный и прямой flow среднего кадра. Тензоры upstream не изменены: это только обёртка метаданных в чекпоинт LibreYOLO v1.0.

✨ Ключевые особенности

• 👁 Три кадра на вход — два результата: flow вперёд и назад для кадра t.
• 💻 Задача flow семейства LibreMEMFOF: результат в Results.flow как поле (H, W, 2) в пикселях на исходном холсте изображения.
• 🎨 Встроенная визуализация to_rgb() по цветовому кругу Middlebury.
• 🔨 Инференс по умолчанию как у upstream: fp32, 8 итераций уточнения, нативное разрешение с replicate-паддингом до кратности 16.
• 📓 Статья MEMFOF принята на ICCV 2025 (arXiv:2506.23151), код и веса под BSD-3-Clause.

💻 Пример

from libreyolo import LibreYOLO

model = LibreYOLO("LibreMEMFOFm-flow-kitti.pt")
results = model(["frame_000.png", "frame_001.png", "frame_002.png"])
flow = results[0].flow.data       # (H, W, 2), dx и dy в пикселях
rgb = results[0].flow.to_rgb()    # цветовой круг Middlebury

Подробнее

Числовых бенчмарков в карточке нет — только ссылка на статью ICCV 2025.
Требования к VRAM не указаны; инференс в fp32, 8 итераций уточнения, паддинг до кратности 16.
Изменены только метаданные: task: flow, nc: 1, model_family: memfof, size: m, variant: kitti.
Лицензия весов отдельно upstream не заявлена и наследуется от кода — BSD-3-Clause.