Pytorch_PCT
5星
- 浏览量: 0
- 大小:None
- 文件类型：None
简介：
PyTorch_PCT是基于PyTorch框架开发的一种点云处理技术，专注于提升三维数据的机器学习模型性能。
PCT：点云变压器
这是PCT的Pytorch实现。
要求：
- Python >= 3.7
- 火炬(Torch) >= 1.6
- h5py
- scikit学习
安装命令：
```
pip install pointnet2_ops_lib/
```
该代码来自楷模。我们在ModelNet40验证数据集上获得了93.2%的准确性。
模型路径在./checkpoints/train/models/model.t7中。
培训和测试示例：
训练:
```python
python main.py --exp_name=train --num_points=1024 --use_sgd=True --batch_size 32 --epochs 250 --lr 0.0001
```
测试：
```python
python main.py --exp_name=test --num_points=1024 --use_sgd，其余参数根据实际情况填写。
```
全部评论 (0)
还没有任何评论哟~
