用於在 Blender 中跨多個 GPU 和進程執行 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 是由艾倫人工智慧研究所 (AI2) 的 PRIOR 團隊所建構的開源專案。 AI2 是一家非營利機構,其使命是透過高影響力的人工智慧研究和工程為人類做出貢獻。