LibreNeuFlowv2s-flow

Это веса NeuFlow v2 из mixed-прогона — upstream-чекпоинт по умолчанию для инференса, переупакованный под задачу flow в LibreYOLO. Сеть на 9M параметров предназначена для вычисления оптического потока на edge-устройствах. Архитектура не новая: изменена только обёртка state-dict.

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

• 👉 9M параметров — компактная модель оптического потока для edge-устройств.
• 💪 Mixed-тренировка на нескольких датасетах; это upstream default inference checkpoint.
• 💳 Совместимость с LibreYOLO: task=flow, canvas 432×768, variant=mixed.
• 🐾 Ключи и обученные параметры не изменены — только metadata-wrap в схеме LibreYOLO.
• 👀 На RTX 3090 в fp32 результат совпадает с upstream бит в бит: max abs diff 0.0 px.

💪 Пример

from libreyolo import LibreYOLO

model = LibreYOLO("LibreNeuFlowv2s-flow.pt")
results = model(["frame_000.png", "frame_001.png"])
flow = results[0].flow.data        # (H, W, 2), (dx, dy)
rgb = results[0].flow.to_rgb()     # Middlebury colour wheel

Подробнее

Веса получены из neuflow_mixed.pth (NeuFlow_v2, commit 204b5e3).
Лицензия Apache 2.0; авторы upstream не давали отдельной лицензии на веса — это интерпретация LibreYOLO.
Для запуска нужен LibreYOLO; parity-проверка проведена на RTX 3090 в fp32.