여러 GPU 및 프로세스에 걸쳐 Blender에서 Objaverse 개체의 분산 렌더링을 수행하는 스크립트입니다.
우리는 NVIDIA GPU가 있는 Ubuntu 시스템에서만 렌더링 스크립트를 테스트했습니다.
문제가 발생하면 문제를 열어주세요! :)
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는 Allen Institute for AI(AI2)의 PRIOR 팀이 구축한 오픈 소스 프로젝트입니다. AI2는 영향력 있는 AI 연구와 엔지니어링을 통해 인류에 공헌한다는 사명을 가진 비영리 기관입니다.