Awesome
Pytorch code for CVR [ICCV2021].
Xin Wei*, Yifei Gong*, Fudong Wang, Xing Sun, Jian Sun. Learning Canonical View Representation for 3D Shape Recognition with Arbitrary Views. ICCV, accepted, 2021. [pdf] [supp]
Citation
If you find our work useful in your research, please consider citing:
@InProceedings{Wei_2021_ICCV,
author = {Wei, Xin and Gong, Yifei and Wang, Fudong and Sun, Xing and Sun, Jian},
title = {Learning Canonical View Representation for 3D Shape Recognition With Arbitrary Views},
booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
month = {October},
year = {2021},
pages = {407-416}
}
Training
Requiement
This code is tested on Python 3.6 and Pytorch 1.8 +
Dataset
First download the arbitrary views ModelNet40 dataset and put it under data
https://drive.google.com/file/d/1RfE0aJ_IXNspVs610BkMgcWDP6FB0cYX/view?usp=sharing
Link of arbitrary views ScanObjectNN dataset:
https://drive.google.com/file/d/10xl-S8-XlaX5187Dkv91pX4JQ5DZxeM8/view?usp=sharing
Aligned-ScanObjectNN dataset: https://drive.google.com/file/d/1ihR6Fv88-6FOVUWdfHVMfDbUrx2eIPpR/view?usp=sharing
Rotated-ScanObjectNN dataset: https://drive.google.com/file/d/1GCwgrfbO_uO3Qh9UNPWRCuz2yr8UyRRT/view?usp=sharing