Skip to content

The official implementation of "Not All Voxels Are Equal: Hardness-Aware Semantic Scene Completion with Self-Distillation" (CVPR 2024)

License

Notifications You must be signed in to change notification settings

songw-zju/HASSC

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

HASSC

Song Wang, Jiawei Yu, Wentong Li, Wenyu Liu, Xiaolu Liu, Junbo Chen*, Jianke Zhu*

This is the official implementation of Not All Voxels Are Equal: Hardness-Aware Semantic Scene Completion with Self-Distillation (CVPR 2024) [Paper] [Video].

Preparation

SemanticKITTI Download

Environment Setup

We release the HASSC implementation with VoxFormer-T, please refer the environment setup in the original repo.

Run and Eval

Train the SSC model with our proposed HASSC on 4 GPUs

./tools/dist_train.sh ./projects/configs/hassc/hassc-voxformer-T.py 4

Eval the SSC model with our proposed HASSC on 4 GPUs

./tools/dist_test.sh ./projects/configs/hassc/hassc-voxformer-T.py ./path/to/ckpts.pth 4

Acknowledgement

Many thanks to these excellent open source projects: VoxFormer, mmdetction3d, PointRend

Citations

@inproceedings{wang2024not,
      title={Not All Voxels Are Equal: Hardness-Aware Semantic Scene Completion with Self-Distillation},
      author={Wang, Song and Yu, Jiawei and Li, Wentong and Liu, Wenyu and Liu, Xiaolu and Chen, Junbo and Zhu, Jianke},
      booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
      year={2024}
}

About

The official implementation of "Not All Voxels Are Equal: Hardness-Aware Semantic Scene Completion with Self-Distillation" (CVPR 2024)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages