LibreWAFTs-flow-dav2

Переупакованные веса WAFT для оценки optical flow: вариант backbone dav2 размера s с энкодером Depth Anything V2 Small и 56,9 млн параметров. Модель подготовлена LibreYOLO для одноимённой библиотеки; изменена только упаковка метаданных чекпойнта, обученные параметры и их имена не тронуты.

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

• 🔍 Считает optical flow между кадрами: N входных кадров дают N − 1 потоков.
• ⚙ Архитектура WAFT (a2) с энкодером Depth Anything V2 Small — 56,9 млн параметров.
• 🔗 Чекпойнт получен из zero-shot набора WAFT: TartanAir, затем FlyingChairs, затем FlyingThings3D.
• 📊 Выход — поток (H, W, 2) в пикселях, компоненты (dx, dy).
• © Лицензии BSD 3-Clause для WAFT и Apache 2.0 для энкодера Depth Anything V2 Small.

💻 Пример

from libreyolo import LibreYOLO

model = LibreYOLO("LibreWAFTs-flow-dav2.pt")
results = model(["frame_000.png", "frame_001.png"])  # N кадров → N - 1 потоков
flow = results[0].flow.data  # (H, W, 2), пиксели, (dx, dy)

Подробнее

Параметров: 56,9 млн.
Чекпойнт из zero-shot набора WAFT, обучение шло на TartanAir, FlyingChairs и FlyingThings3D.
В репозитории изменена только обёртка метаданных, веса не переобучались.
Лицензия: BSD 3-Clause + Apache 2.0.