windows 下复习 PF-Net

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

windows 下复习 PF-Net_第1张图片

数据这么放   shapenetcore_partanno_segmentation_benchmark_v0

windows 下复习 PF-Net_第2张图片

创一个文件夹放权重

windows 下复习 PF-Net_第3张图片

 修改    把1,2 该 为0

windows 下复习 PF-Net_第4张图片

    

修改权重路径

windows 下复习 PF-Net_第5张图片

 右键    run  就可以了  这是cpu的

windows 下复习 PF-Net_第6张图片

 还在训练中。

gpu

windows 下复习 PF-Net_第7张图片

windows 下复习 PF-Net_第8张图片

内存不够    修改为3的倍数

windows 下复习 PF-Net_第9张图片

 windows 下复习 PF-Net_第10张图片

 

你可能感兴趣的:(复现,pycharm,python)