confhead

confhead — обученная голова уверенности для замороженной Depth Anything V2 vitb на nuScenes. Она предсказывает p_error — вероятность того, что глубина ячейки сетки неверна относительно лидара. В репозитории один прогон run1: 68 входных каналов (features, depth, temporal, radar), сетка 132×74.

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

• 🔒 Работает поверх замороженной Depth Anything V2 vitb; сама модель глубины не переобучается.
• 📡 Входы: features, depth, temporal и radar, всего 68 каналов.
• 🎯 Предсказывает p_error для каждой ячейки; положительная метка — абсолютная относительная ошибка выше tau=0.1, глубина 1–80 м.
• 🗂 В репозитории: best.pt, last.pt, config.json, history.json и steps.json.
• ⚙ Обучение: AdamW, cosine schedule, BCE на ячейках с лидарной меткой, 8 эпох.
• ❗ Лицензия наследует CC BY-NC-SA 4.0 и условия использования nuScenes.

⌨ Пример

hf download nadayusc/confhead <run>/best.pt --local-dir .

from confhead import load_head
head, inputs, fields = load_head("<run>/best.pt")

Подробнее

Лучший dev loss run1 — 0.5562 на 8-й эпохе из 8; обучение заняло 3,5 часа.
Доля положительных ячеек — 37,6%, class weight — 1,658.
Обучение шло на CUDA с torch 2.14.1+cu126; VRAM в карточке не указан.
Обучение и метки — nuScenes; других доменов в карточке нет.