Skip to content

densechen/CASS

master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
lib
 
 
 
 
 
 
 
 
 
 
 
 

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