GitHub - zztianzz/PF-Net-Point-Fractal-Network: CVPR2020 PF-Net: Point Fractal Network for 3D Point Cloud Completion源码链接
下载数据集
https://pan.baidu.com/s/1MavAO_GHa0a6BZh4Oaogug 提取码:3hoe
cpu环境
baidu-aip==2.2.18.0
cached-property==1.5.2
certifi==2021.10.8
charset-normalizer==2.0.12
h5py==3.6.0
idna==3.3
numpy==1.21.5
Pillow==9.1.0
requests==2.27.1
torch==1.11.0
torchvision==0.12.0
typing_extensions==4.1.1
urllib3==1.26.8
gpu 环境
cuda 11.1的
安装 torch 记得去测能不能加速
pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
数据这么放 shapenetcore_partanno_segmentation_benchmark_v0
创一个文件夹放权重
修改 把1,2 该 为0
修改权重路径
右键 run 就可以了 这是cpu的
还在训练中。
gpu
内存不够 修改为3的倍数