LibreWAFTm-flow

LibreYOLO выложил LibreWAFTm-flow — веса оптического потока WAFT (a1) с энкодером Depth Anything V2 Small на 64,2 млн параметров, перепакованные для библиотеки LibreYOLO. Это канонический чекпойнт WAFT, который README исходного проекта рекомендует для downstream-задач. Модель принимает последовательность кадров и оценивает движение между соседними кадрами.

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

• 🎞 Считает optical flow: N кадров дают N−1 потоков.
• 🔥 Канонический WAFT a1 — вариант, рекомендованный в README исходного проекта для downstream.
• 💻 Перепакован для LibreYOLO: обученные параметры и их имена не менялись, изменена только обёртка метаданных.
• 📚 Энкодер — Depth Anything V2 Small, всего 64,2 млн параметров.
• ✅ Лицензии: BSD 3-Clause для WAFT и Apache 2.0 для энкодера.

💻 Пример

from libreyolo import LibreYOLO

model = LibreYOLO('LibreWAFTm-flow.pt')
results = model(['frame_000.png', 'frame_001.png'])  # N frames give N - 1 flows
flow = results[0].flow.data                           # (H, W, 2), pixels, (dx, dy)

Подробнее

Обучение: zero-shot по схеме TartanAir → FlyingChairs → FlyingThings3D.
Параметров: 64,2 млн.
Лицензии: BSD 3-Clause (WAFT) + Apache 2.0 (Depth Anything V2 Small).
Статья: WAFT, ICLR 2026.
Бенчмарков и требований к VRAM в карточке нет.