search

3dlg-hcvc / cage

[CVPR 2024] Official Implementation of the paper "CAGE: Controllable Articulation GEneration"
https://3dlg-hcvc.github.io/cage/
MIT License
40 stars 2 forks source link

Saving generated meshes #1

Closed akomarichev closed 3 months ago

akomarichev commented 5 months ago

Hi,

Thank you for your great work! I am curious how to save generated objects as meshes / urdf in your pipeline once the model is fully trained?

SevenLJY commented 4 months ago

Hi @akomarichev, thanks for your interest in our work, and sorry for the late reply.. We used a simple retrieval strategy to obtain the meshes for each part. The code can be found in the retrieval folder. A "pseudo" usage example is as below.

# hashbook is under the same folder
obj_candidates = find_obj_candidates(json_data, gt_data_dir, hashbook_path)    
parts_to_render = pick_and_rescale_parts(json_data, obj_candidates, gt_data_dir)