Скрипты для выполнения распределенного рендеринга объектов Objaverse в Blender на многих графических процессорах и процессах.
Мы тестировали сценарии рендеринга только на компьютерах Ubuntu с графическими процессорами NVIDIA.
Если у вас возникнут какие-либо проблемы, пожалуйста, откройте проблему! :)
wget https://download.blender.org/release/Blender3.2/blender-3.2.2-linux-x64.tar.xz
tar -xf blender-3.2.2-linux-x64.tar.xz
rm blender-3.2.2-linux-x64.tar.xz
# this is needed to download urls in blender
# https://github.com/python-poetry/poetry/issues/5117#issuecomment-1058747106
sudo update-ca-certificates --fresh
export SSL_CERT_DIR=/etc/ssl/certs
pip install -r requirements.txt
sudo apt-get install xserver-xorg
sudo python3 scripts/start_xserver.py start
python3 scripts/download_objaverse.py --start_i 0 --end_i 100
python3 scripts/distributed.py
--num_gpus < NUM_GPUs >
--workers_per_gpu < WORKERS_PER_GPU >
--input_models_path < INPUT_MODELS_PATH >
Затем изображения будут отображены в каталоге views
.
В скрипте scripts/distributed.py
мы используем Wandb для регистрации результатов рендеринга. Вы можете создать бесплатную учетную запись, а затем установить для переменной среды WANDB_API_KEY
свой ключ API.
Мы также используем AWS S3 для загрузки обработанных изображений. Вы можете создать бесплатную учетную запись, а затем установить переменные среды AWS_ACCESS_KEY_ID
и AWS_SECRET_ACCESS_KEY
для своих учетных данных.
Objaverse — это проект с открытым исходным кодом, созданный командой ПРИОР из Института искусственного интеллекта Аллена (AI2). AI2 — это некоммерческий институт, миссия которого состоит в том, чтобы внести свой вклад в развитие человечества посредством эффективных исследований и разработок в области искусственного интеллекта.