Pytorch
文章平均质量分 51
白色小靴
学生
展开
-
最新激光雷达闭环检测/地点识别算法CVTNet(2023)
CVTNet以激光点云多类投影生成的二维图为输入,利用cross transformer将多类信息交叉融合,为激光点云提取强特异性描述子,实现SLAM闭环检测或全局定位功能。此外,CVTNet生成的全局描述子具备车辆yaw角旋转不变性,提升车辆多视角地点识别精度。各项试验表明,CVTNet达到了SOTA的地点识别精度,且实时性适用于自动驾驶要求。代码地址:https://github.com/BIT-MJY/CVTNet。论文地址:https://arxiv.org/abs/2302.01665。原创 2023-04-01 10:26:49 · 572 阅读 · 1 评论 -
Pytorch使用GPU
pytorch使用gpu的机制是,将所建立的模型、以及训练集的tensor直接建立在gpu上。具体实现如下:net = Net()device = torch.device('cuda:0') # ****************************net.to(device) # ****************************。。。。。。。。。。。。。。 for i, data in enumerate(trainloader, 0): input原创 2020-10-16 13:50:46 · 648 阅读 · 0 评论 -
ubuntu20.04 配置pytorch深度学习环境 + SSD目标检测
0、准备工作0.1 gcc降级因为Ubuntu20.04自带的gcc版本为9.3,而cuda10不支持gcc-9,因此要手动安装gcc-7,命令如下:sudo apt-get install gcc-7 g++-7设置优先使用gcc7sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 9sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++原创 2020-10-11 15:59:07 · 4149 阅读 · 2 评论