LibreDPFlowm-flow-sintel

Это checkpoint DPFlow (CVPR 2025) для оценки оптического потока, упакованный под задачу flow в LibreYOLO. Версия sintel дообучена на смеси FlyingThings3D, MPI-Sintel, KITTI 2015 и HD1K. Веса происходят из ptlflow; изменения — только обёртка state-dict, learned parameters не тронуты.

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

• 👁 Возвращает поток между двумя кадрами: r.flow.data — (H, W, 2) float32 в пикселях, (dx, dy) от кадра 0 к кадру 1.
• 🏔 Dual-pyramid encoder сам выбирает глубину по диагонали изображения: три уровня около 1K и ещё один на каждое удвоение.
• 🔥 Работает в нативном разрешении кадров и рассчитан на обобщение до 4K и 8K.
• ✅ Сохраняет upstream-расписание теста: 4 итерации refinement на уровень.
• ⬇ Репозиторий для LibreYOLO: модель автоматически скачивается из этого репо по имени файла.
• ♥ Веса некоммерческие — CC BY-NC-SA 4.0; код DPFlow — Apache-2.0.

👁 Пример

from libreyolo import LibreYOLO

model = LibreYOLO("LibreDPFlowm-flow-sintel.pt")  # auto-downloads from this repo
r = model(["frame_000.png", "frame_001.png"])[0]
r.flow.data      # (H, W, 2) float32, pixels, (dx, dy)
r.flow.to_rgb()  # Middlebury colour wheel

Подробнее

Проверка parity на RTX 3090: 23 пары 640×360, fp32, TF32 off, deterministic cuDNN — max abs diff 0.0 px, mean 0.0 px, bitwise identical.
Лицензия весов: CC BY-NC-SA 4.0, только non-commercial; код DPFlow — Apache-2.0.
Источник: ptlflow, dpflow-sintel-b44b072c.ckpt; изменения — только metadata wrapping state-dict.