Skip to content

Commit

Permalink
ObjV1
Browse files Browse the repository at this point in the history
  • Loading branch information
hw-liang committed May 27, 2024
1 parent df9c394 commit 44e773e
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 5 deletions.
7 changes: 4 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,8 +12,9 @@ vast 3D data corpus of [Objaverse-1.0](https://objaverse.allenai.org/objaverse-1
3. (To be uploaded) Rendered 4D images by our team to save you GPU time.

## 4D Dataset ID
We curated a total of 54K dynamic 3D assets from Objaverse-1.0 and Objaverse-xl. We provide the selected ids from Objaverse-1.0 in `rendering/src/ObjV1_curated.txt`. More about the dataset and curation scrips coming soon!
We curated a total of 54K dynamic 3D assets from Objaverse-1.0 and Objaverse-xl. We provide the selected ids from Objaverse-1.0 in `rendering/src/ObjV1_curated.txt`.
For text-to-4D generation, the captions are obtained from the work [Cap3D](https://huggingface.co/datasets/tiange/Cap3D).
More about the dataset and curation scrips coming soon!

## 4D Dataset Rendering Script
1. Clone the repository and enter the rendering directory:
Expand All @@ -38,13 +39,13 @@ pip install objaverse
python download.py --id_path src/sample.txt
```

Right-click the objaverse package in `download.py` and open the `objaverse/__init__.py` file. Change `BASE_PATH='./obj_data'` at line 14 or another path you prefer. Otherwise, it will download to the default system path.
Please change `objaverse._VERSIONED_PATH` in `download.py` to the path you prefer to store the glb files. By default, it will be downloaded to `obj_v1/`.

4. Render 4D images

Default setting
```bash
python render.py --obj_path "./obj_data/hf-objaverse-v1/glbs" \
python render.py --obj_path "./obj_v1/glbs" \
--save_dir './output' \
--gpu_num 8 \
--frame_num 24 \
Expand Down
4 changes: 2 additions & 2 deletions rendering/download.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,8 @@
# print(object_uids)
print('file number:', len(object_uids))

objaverse.BASE_PATH = '/home/projects/Diffusion4D/rendering/'
objaverse._VERSIONED_PATH = 'objaverse_v1/'
# objaverse.BASE_PATH = '/home/projects/Diffusion4D/rendering/'
objaverse._VERSIONED_PATH = 'obj_v1/'
objaverse.load_objects(
uids=object_uids,
download_processes=multiprocessing_cpu_count
Expand Down

0 comments on commit 44e773e

Please sign in to comment.