search

GAP-LAB-CUHK-SZ / Total3DUnderstanding

Implementation of CVPR'20 Oral: Total3DUnderstanding: Joint Layout, Object Pose and Mesh Reconstruction for Indoor Scenes from a Single Image
MIT License
418 stars 50 forks source link

The visualization #3

Closed ysunbp closed 4 years ago

ysunbp commented 4 years ago

It seems that the test command can only generate meshes in .obj format and their metadata. Is there an easy way to generate the images(screen shot?) of the resulting 3d scene?

yinyunie commented 4 years ago

I updated the utils/visualize.py.

After the metadata is generated in testing, you can visualize a 3D scene (only in the test set of SUN RGB-D) by giving the sample id as: python utils/visualize.py --result_path out/total3d/a_folder_named_with_script_time/visualization --sequence_id 274

You can also save the screenshots by turn on the 'if_save' in the visualize.py file. The results will be saved in demo/sunrgbd/.