Win11复现 PointNet ++ 分类任务 || 三维点云深度学习第一篇

Windows 11 Github 上 复现 PointNet ++Classification 任务

下载代码和安装依赖环境

安装git

知乎教程

Git clone

源码地址: PointNet Github

Alt

点击Code 复制HTTPS

到目标目录

shift-右键 Git Bash Here

输入

Git Clone <url>

​​​​​​
Alt

安装nodejs

Node.js — Download (nodejs.org) 根据电脑型号选择

Alt

到目标目录继续输入以下命令:

npm init -f
npm install

Alt
会在你的项目文件夹里看到node_modules文件 即成功

下载ModelNet40数据集

ModelNet40数据集
格式:txt
打开看看:顺便看看路径 别放错啦
红色是X Y Z 橙色是 normal features
在这里插入图片描述

将下载路径置于项目文件 新建的 data文件夹里

(备注:下载过慢可以安装IDM Internet Download Manager软件和插件到浏览器)

打开Anaconda Prompt 配置环境

激活虚拟环境 安装相应包

conda activate pytorch 

(pytorch是我命名虚拟环境的名字 这里默认大家虚拟环境安装好了)

cd /d <目标路径>

也就是项目文件所在路径 同前面的路径

以tqdm为例

pip install tqdm 

Alt

运行代码

README里面有简介 比如想用的模型(pointNet 还是pointNet ++(的ssg 还是msg )) 以及用的数据集、是否用法向量(除了xyz以外的三个特征)用不用gpu 此外还要记得给定batch_size

示例:

python train_classification.py --model pointnet2_cls_ssg --log_dir pointnet2_cls_ssg --batch_size 8 --use_cpu

在这里插入图片描述

There you go !!!
但是实在跑的太慢了QAQ
后续可能会出服务器使用教程!欢迎关注~

  • 19
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
以下是使用PyTorch复现PointNet的基本步骤: 1. 导入必要的库 ```python import torch import torch.nn as nn import torch.nn.functional as F ``` 2. 定义PointNet的基本操作 ```python class TNet(nn.Module): def __init__(self, k=3): super(TNet, self).__init__() self.conv1 = nn.Conv1d(k, 64, 1) self.conv2 = nn.Conv1d(64, 128, 1) self.conv3 = nn.Conv1d(128, 1024, 1) self.fc1 = nn.Linear(1024, 512) self.fc2 = nn.Linear(512, 256) self.fc3 = nn.Linear(256, k*k) def forward(self, x): batch_size = x.size(0) x = F.relu(self.conv1(x)) x = F.relu(self.conv2(x)) x = F.relu(self.conv3(x)) x = torch.max(x, 2, keepdim=True)[0] x = x.view(-1, 1024) x = F.relu(self.fc1(x)) x = F.relu(self.fc2(x)) x = self.fc3(x) iden = torch.eye(x.size(1)).view(1, x.size(1)*x.size(1)).repeat(batch_size,1) if x.is_cuda: iden = iden.cuda() x = x + iden x = x.view(-1, x.size(1), x.size(1)) return x ``` 3. 定义PointNet的分类模型 ```python class PointNetCls(nn.Module): def __init__(self, k=2): super(PointNetCls, self).__init__() self.tnet1 = TNet(k=3) self.conv1 = nn.Conv1d(3, 64, 1) self.conv2 = nn.Conv1d(64, 64, 1) self.tnet2 = TNet(k=64) self.conv3 = nn.Conv1d(64, 128, 1) self.conv4 = nn.Conv1d(128, 1024, 1) self.fc1 = nn.Linear(1024, 512) self.fc2 = nn.Linear(512, 256) self.fc3 = nn.Linear(256, k) def forward(self, x): batch_size = x.size(0) x = self.tnet1(x) x = torch.bmm(x, x.transpose(2,1)) x = x.unsqueeze(1) x = F.relu(self.conv1(x)) x = F.relu(self.conv2(x)) x = torch.max(x, 2, keepdim=True)[0] x = x.repeat(1, 1, 1024) x = torch.cat([x, self.tnet2(x)], 1) x = F.relu(self.conv3(x)) x = self.conv4(x) x = torch.max(x, 2, keepdim=True)[0] x = x.view(-1, 1024) x = F.relu(self.fc1(x)) x = F.relu(self.fc2(x)) x = self.fc3(x) return F.log_softmax(x, dim=1) ``` 4. 定义损失函数和优化器 ```python model = PointNetCls() criterion = nn.NLLLoss() optimizer = torch.optim.Adam(model.parameters(), lr=0.001) ``` 5. 进行训练 ```python for epoch in range(50): running_loss = 0.0 for i, data in enumerate(trainloader, 0): inputs, labels = data optimizer.zero_grad() outputs = model(inputs) loss = criterion(outputs, labels) loss.backward() optimizer.step() running_loss += loss.item() print('Epoch %d loss: %.3f' %(epoch+1, running_loss/len(trainloader))) ``` 6. 进行测试 ```python correct = 0 total = 0 with torch.no_grad(): for data in testloader: inputs, labels = data outputs = model(inputs) _, predicted = torch.max(outputs.data, 1) total += labels.size(0) correct += (predicted == labels).sum().item() print('Accuracy: %.2f %%' % (100 * correct / total)) ``` 这就是使用PyTorch复现PointNet的基本步骤。可以根据需要进行调整和改进,以获得更好的性能。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值