Skip to content
/ CASS Public

CASS:Learning Canonical Shape Space for Category-Level 6D Object Pose and Size Estimation

Notifications You must be signed in to change notification settings

densechen/CASS

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CASS:Learning Canonical Shape Space for Category-Level 6D Object Pose and Size Estimation

Evaluation Steps

  1. Install the following requirements:
    open3d==0.8.0.0
    opencv-python==4.1.1.26
    torch==1.2.0
    torchvision==0.4.0
    tqdm==4.32.1
    trimesh==3.2.20
  1. Compile "./metrics" for re-evaluating reconstructed models. You can skip this step and delete line 25-28 in ./tools/eval.py, if you have downloaded our results in next step.

  2. Download predicted masks and pretrained models.

    You can download our pretrained models, results and segmentation masks of real test dataset in NOCS from Google Driver.

    If you want to re-calculate CASS's results, please download the NOCS real test dataset and 3d models.

  3. Evaluate CASS and NOCS

    1. Unzip predicted results, and specified --save_dir in eval.sh. You will get evaluation results of CASS and NOCS at the same time.
    2. If you want to recalculate CASS's results, please place segmentation mask of NOCS, which is contained in the Google Driver, to the real-test dataset folder along with their color images. Refer to 1-2 line in ./eval.sh about how to start the evaluation.

Acknowledgement

We have referred to part of the code from NOCS_CVPR2019, FoldingNet, DenseFusion, Open3D and PointFlow.

About

CASS:Learning Canonical Shape Space for Category-Level 6D Object Pose and Size Estimation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published