representación objaversa

Representación objaversa

Scripts para realizar renderizado distribuido de objetos Objaverse en Blender en muchas GPU y procesos.

Requisitos del sistema

Solo hemos probado los scripts de renderizado en máquinas Ubuntu que tienen GPU NVIDIA.

Si tiene algún problema, abra un problema. :)

Instalación

1. Instalar licuadora

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

2. Actualizar certificados de Blender para descargar URL

 # 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

3. Instalar dependencias de Python

pip install -r requirements.txt

4. (Opcional) Si está ejecutando el renderizado en una máquina sin cabeza, deberá iniciar un servidor x. Para hacer esto, ejecute:

sudo apt-get install xserver-xorg
sudo python3 scripts/start_xserver.py start

Representación

1. Descarga los objetos:

python3 scripts/download_objaverse.py --start_i 0 --end_i 100

2. Inicie el script de renderizado distribuido:

python3 scripts/distributed.py 
  --num_gpus < NUM_GPUs > 
  --workers_per_gpu < WORKERS_PER_GPU > 
  --input_models_path < INPUT_MODELS_PATH >

Esto luego renderizará las imágenes en el directorio views .

(Opcional) Registro y carga

En el script scripts/distributed.py , usamos Wandb para registrar los resultados del renderizado. Puede crear una cuenta gratuita y luego configurar la variable de entorno WANDB_API_KEY en su clave API.

También utilizamos AWS S3 para cargar las imágenes renderizadas. Puede crear una cuenta gratuita y luego configurar las variables de entorno AWS_ACCESS_KEY_ID y AWS_SECRET_ACCESS_KEY según sus credenciales.

Nuestro equipo

Objaverse es un proyecto de código abierto creado por el equipo PRIOR del Instituto Allen de IA (AI2). AI2 es un instituto sin fines de lucro con la misión de contribuir a la humanidad a través de la investigación e ingeniería de IA de alto impacto.