AutoBrep — авторегрессивная генерация геометрии и топологии B-Rep
AutoBrep — модель-трансформер для авторегрессивной генерации геометрии и топологии в формате B-Rep. Она предназначена для исследователей и разработчиков, работающих с 3D-моделированием на основе последовательности токенов, описывающих геометрические и топологические характеристики.
Возможности и состав проекта
Модель генерирует B-Rep геометрию и топологию, используя порядок обхода в ширину (BFS) графа топологии B-Rep. Геометрическая информация токенизируется как ограничивающие рамки в сочетании с закодированными UV-grid shape codes. Топологическая структура представлена специальными идентификаторами граней, которые связывают смежности граней и рёбер с референсными токенами.
Начало работы и требования
Для работы с проектом необходимо установить среду с помощью conda и активировать её. Доступны предварительно обученные модели, которые можно загрузить из Hugging Face для использования в инференсе. Для обучения модели требуется загрузить набор данных ABC-1M и указать путь к нему, а также к предварительно обученным FSQ-моделям в файле конфигурации.
Ограничения и выбор сценария
AutoBrep является исследовательской моделью, а не параметрической CAD-системой с инженерными гарантиями. Сгенерированную топологию и геометрию нужно проверять CAD-инструментами на корректность и пригодность для задачи. Обучение требует подготовленного ABC-1M и FSQ-компонентов; загрузка одного checkpoint не воспроизводит весь обучающий процесс. Права на данные и зависимости проверяются отдельно.
Лицензия и условия использования
GitHub определяет лицензию репозитория как MIT. Перед использованием и распространением проверьте полный текст LICENSE и отдельные условия зависимостей, данных или моделей: лицензия репозитория не заменяет их лицензии.
