zhangguanghui1 / PF-Net-Point-Fractal-Network

CVPR2020 PF-Net: Point Fractal Network for 3D Point Cloud Completion

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

PF-Net-Point-Fractal-Network

This repository is still under constructions.

If you have any questions about the code, please email me. Thanks!

This is the Pytorch implement of CVPR2020 PF-Net: Point Fractal Network for 3D Point Cloud Completion.

##0) Environment Pytorch 1.0.1 Python 3.7.4

##1) Dataset

  cd dataset
  bash download_shapenet_part16_catagories.sh

##2) Train

python Train_FPNet.py 

Change ‘crop_point_num’ to control the number of missing points. Change ‘point_scales_list ’to control different input resolutions. Change ‘D_choose’to control without using D-net.

##3) Evaluate the Performance on ShapeNet

python show_recon.py

Show the completion results, the program will generate txt files in 'test-examples'.

python show_CD.py

Show the Chamfer Distances and two metrics in our paper.

##4) Visualization of csv File

We provide some incomplete point cloud in file 'test_one'. Use the following code to complete a incomplete point cloud of csv file:

python Test_csv.py

change ‘infile’and ‘infile_real’to select different incomplete point cloud in ‘test_one’

##5) Visualization of Examples

Using Meshlab to visualize the txt files.

About

CVPR2020 PF-Net: Point Fractal Network for 3D Point Cloud Completion

License:MIT License


Languages

Language:Python 99.7%Language:Shell 0.3%