Skip to content

ahyunSeo/RotSymDETR

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

2 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

Leveraging 3D Geometric Priors in 2D Rotation Symmetry Detection (CVPR 2025)

Ahyun Seo, Minsu Cho

[Paper] Β |Β  [Project Page]

This is the official PyTorch implementation of
Leveraging 3D Geometric Priors in 2D Rotation Symmetry Detection, accepted at CVPR 2025.


πŸ› οΈ Environment Setup

conda create -n rotsymdetr python=3.8.18 pip -y
conda activate rotsymdetr
bash setup.sh
mkdir weights sym_datasets

πŸ“‚ Datasets and Pretrained Weights

Directory structure after setup:

.
β”œβ”€β”€ sym_datasets
β”‚   └── DENDI
β”œβ”€β”€ weights
β”‚   β”œβ”€β”€ baseline_3d_best.pth
β”‚   └── prior_3d_best.pth
β”œβ”€β”€ projects 
└── setup.sh

πŸš€ Running the Code

πŸ”Ž Inference (with pretrained weights)

./tools/dist_test.sh projects/configs/prior_3d.py weights/prior_3d_best.pth $ngpu

πŸ‹οΈβ€β™‚οΈ Training

./tools/dist_train.sh projects/configs/prior_3d.py $ngpu
  • For the 3D baseline, use baseline_3d.py and baseline_3d_best.pth.

πŸ™ Acknowledgement

Many thanks to the following project for inspiration and support:

  • BEVFormer (ECCV 2022) β€” GitHub

πŸ“– Citation

If you find this work helpful, please consider citing:

@inproceedings{seo2025rotsymdetr,
  author    = {Seo, Ahyun and Cho, Minsu},
  title     = {Leveraging 3D Geometric Priors in 2D Rotation Symmetry Detection},
  booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
  year      = {2025}
}

About

(CVPR 2025) Official PyTorch implementation of "Leveraging 3D Geometric Priors in 2D Rotation Symmetry Detection."

Resources

Stars

Watchers

Forks

Contributors