Awesome
SAPCU
【Code of CVPR 2022 paper】
Self-Supervised Arbitrary-Scale Point Clouds Upsampling via Implicit Neural Representation
Paper address: https://arxiv.org/abs/2204.08196
Environment
Pytorch 1.9.0
CUDA 10.2
Evaluation
a. Download models
Download the pretrained models from the link and unzip it to ./out/
https://pan.baidu.com/s/1OPVnCHq129DBMWh5BA2Whg
access code: hgii
or
https://drive.google.com/file/d/12TifjDW2L7r2LK3AGDgabHdN3f0xY5DR/view?usp=share_link
b. Compilation
Run the following command for compiling dense.cpp which generates dense seed points. Note that the size of the input point cloud is currently limited to 5000, if you want to change this limit, you need to modify the size of the kdtree.
g++ -std=c++11 dense.cpp -O2 -o dense
c. Evaluation
You can now test our code on the provided point clouds in the test
folder. To this end, simply run
python generate.py
The 4X upsampling results will be created in the testout
folder.
Ground truth are provided by Meta-PU
Training
Download the training dataset from the link and unzip it to the working directory
https://pan.baidu.com/s/1VQ-3RFO02fQfcLBfqvCBZA
access code: vpfm
or
https://1drv.ms/f/s!AsP2NtMX-kUTml4U3DYUD6Hy9FJn?e=8QfJTH
Then run the following commands for training our network
python trainfn.py
python trainfd.py
Dataset
We present a fast implementation for building the dataset, which is based on occupancy_networks.
a. Preprocessing
Follow the link occupancy_networks to obtain pointclouds and watertight meshes, notice that we only use ShapeNet dataset v1.
b. Building and installing
Then move scripts to occupancy_networks/scripts run the following commands for building traindata for fd and fn:
bash dataset_shapenet/build-fd.sh
bash dataset_shapenet/build-fn.sh
bash dataset_shapenet/installfd.sh
bash dataset_shapenet/installfn.sh
Evaluation
The code for evaluation can be download from:
https://github.com/pleaseconnectwifi/Meta-PU/tree/master/evaluation_code
https://github.com/jialancong/3D_Processing
Citation
If the code is useful for your research, please consider citing:
@inproceedings{sapcu,
title = {Self-Supervised Arbitrary-Scale Point Clouds Upsampling via Implicit Neural Representation},
author = {Wenbo Zhao, Xianming Liu, Zhiwei Zhong, Junjun Jian, Wei Gao, Ge Li, Xiangyang Ji},
booktitle = {Proceedings IEEE Conf. on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2022},
pages = {1999-2007}
}
Acknowledgement
The code is based on occupancy_networks and DGCNN, If you use any of this code, please make sure to cite these works.