TripoSR: генератор 3D-моделей — что умеет и что нужно для запуска

TripoSR — это инструмент с открытым исходным кодом для быстрого создания трёхмерных объектов по одному изображению. Он решает задачу image-to-3D, то есть превращает плоскую картинку в объёмную модель, и ориентирован на высокую скорость работы.

Что умеет

По описанию разработчиков, TripoSR восстанавливает трёхмерную геометрию объекта по единственному входному изображению. Авторы делают акцент на скорости реконструкции, что отражено в названии проекта. Детали о поддерживаемых форматах выходных моделей и текстур в паспорте не раскрываются.

Что нужно для запуска

• Платформа: по описанию авторов, для работы требуется CUDA. Это означает, что необходим графический процессор NVIDIA.
• Видеопамять: авторы указывают, что для обработки одного изображения с параметрами по умолчанию требуется около 6 ГБ VRAM.
• Язык и окружение: код написан на Python. Для установки зависимостей авторы рекомендуют установить CUDA, если она доступна в системе.
• Файлы модели: веса модели занимают 1.6 ГБ. Самый большой файл также весит 1.6 ГБ.
• Лицензия: и код, и веса распространяются по лицензии MIT. Это разрешает свободное использование, включая коммерческое.
• Размещение: код доступен в репозитории github.com/VAST-AI-Research/TripoSR, веса модели — на huggingface.co/stabilityai/TripoSR.

Кому подойдёт

TripoSR будет полезен разработчикам и исследователям, которым нужна быстрая черновая реконструкция 3D-объектов из картинок, и кто готов запускать Python-скрипты на локальной машине с видеокартой NVIDIA. Открытая лицензия MIT позволяет встраивать этот генератор в свои проекты без ограничений.

Инструмент не подойдёт пользователям без дискретной видеокарты NVIDIA с объёмом памяти менее 6 ГБ. Также он не рассчитан на тех, кто ожидает готовое приложение с графическим интерфейсом — работа идёт через командную строку и скрипты на Python. Авторы не предоставляют облачной версии, запустить TripoSR локально не получится на компьютерах с GPU от AMD или Intel, если они не поддерживают CUDA.

TripoSR представляет собой компактный инструмент с чёткими системными требованиями. Его главная особенность, заявленная создателями, — скорость получения результата из одного снимка, что может быть критично в задачах, где время обработки важнее детализации.