Eff-GRot

Eff-GRot — предобученные веса для одноимённой модели оценки поворота (rotation estimation) на трансформерах, представленной на WACV 2026. Репозиторий fanismathioulakis содержит компактную PyTorch-модель примерно на 20M параметров (0.02B) и отсылает к коду и статье.

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

• 🔍 Задача — оценка поворота объектов с помощью трансформеров.
• ⚙ Компактный размер: около 20M параметров, библиотека PyTorch.
• 📊 В карточке заявлены настройки оценки из статьи: 64 референса, farthest-point reference selection, allocentric-позы и in-plane reference augmentation для LINEMOD.
• 🖥 Для запуска есть готовый eval.py и инструкции в GitHub-репозитории.

💻 Пример

hf download fanismathioulakis/eff-grot model.safetensors config.json --local-dir ckpts/eff-grot

python eval.py --dataset linemod --checkpoint ckpts/eff-grot/model.safetensors --verbose

Подробнее

Модель: ~20M параметров, PyTorch.
В карточке нет цифр бенчмарков и требований к VRAM.
Оценка по умолчанию воспроизводит настройки статьи для LINEMOD: 64 референса, farthest-point selection, allocentric-позы, in-plane аугментация.
Для CO3D и подготовки датасетов нужно смотреть GitHub-репозиторий.