
dgcnn.pytorch:基于PyTorch的动态图CNN在点云学习中的应用(DGCNN)
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
DGCNN.pytorch是一款基于PyTorch框架实现的动态图卷积神经网络工具包,专注于点云数据的学习与处理,适用于各类图形识别和分类任务。
DGCNN.pytorch是用于点云学习的动态图CNN(DGCNN)的PyTorch实现。我们的代码框架是从其他资源借鉴而来的。请注意,在DGCNN论文中,分类网络结构在图3与第4.1节中的描述存在不一致之处。作者使用了第4.1节中的设置而不是图3,并且我们在PS的帮助下修正了图3的错误。
提示:点云实验的结果通常比2D图像更具随机性,请您多次运行实验并选择最佳结果以提高准确性。
所需环境:
- Python 3.7
- PyTorch 1.2
- CUDA 10.0
依赖包:
- glob,h5py,sklearn
点云分类操作:
训练脚本执行示例:
对于1024个点的实验: python main_cls.py --exp_name=cls_1024 --num_points=1024 --k=20
对于2048个点的实验: python main_cls.py (参数根据需要调整)
全部评论 (0)
还没有任何评论哟~


