许久没有更新了,3D目标检测在去年暑期做了语言,在今年年底获得了初步的实践过程,此处3D目标检测使用的是OpenPCDet框架,希望使用PointPillar进行训练,由于OpenPCDet都是基于Kitti数据集的.bin文件,因此通过realsense相机采集的.ply文件需要经过初步的处理才可以用于训练,具体的步骤如下。
首先需要安装好open3d,在OpenPCDet根目录下运行以下命令完成基本的环境配置
pip install open3d
pip install -r requirements.txt
python setup.py develop
在运行setup脚本时可能会出现权限不足的问题,可以改用 pip install -e .
对pcdet库进行安装。安装完成后可以下载部分kitti数据集并运行demo.py进行测试,或是在命令行尝试import pcdet
python tools/demo.py --cfg_file tools/cfgs/kitti_models/pointpillar.yaml --ckpt tools/pointpillar_7728.pth --data_path data/kitti/training/velodyne/000008.bin
数据集标定使用了kitti的格式,因此可以模仿kitti数据集结构分配文件,文件树结构如下:
OpenPCDet
├── data
│ ├── kitti
│ │ │── ImageSets
│ │ │ ├──train.txt & test.txt & val.txt
│ │ │── training
│ │ │ ├──calib & velodyne & label_2 & image_2
│ │ │── testing
│ │ │ ├──calib & velodyne & image_2 & label_2
ImageSets文件夹保存了用于训练、测试以及验证的数据集分类结果,将标定数据的文件名写入train.txt文件,并且需要将所对应的点云和label数据放到training文件夹的velodyne和label_2文件夹下。
kitti数据集的点云.bin文件中包含了点云的x、y、z空间坐标以及激光反射强度intensity数据,而realsense采集到的点云.ply文件为xyzrgb格式,为了快速使用OpenPCDet进行训练,因此可以将.ply文件通过tools/ply2bin.py进行转换。(此处可优化custom_dataset.py中的数据集加载进行修改,从而省去点云数据转换一步和生成pkl文件步骤)
import open3d as o3d
import numpy as np
import os
# load ply
def read_ply(filepath):
lidar = []
pcd = o3d.io.read_point_cloud(filepath, format='ply')
points = np.array(pcd.points)
for linestr in points:
if len(linestr) == 3: # only x,y,z
linestr_convert = list(map(float, linestr))
linestr_convert.append(0)
lidar.append(linestr_convert)
if len(linestr) == 4: # x,y,z,i
linestr_convert = list(map(float, linestr))
lidar.append(linestr_convert)
if len(linestr) == 6: # x,y,z,r,g,b
linestr_convert = list(map(float, linestr))
linestr_convert = linestr_convert[:3]
linestr_convert.append(0)
lidar.append(linestr_convert[:3])
return np.array(lidar)
def pcd2bin(pcd_fullname, bin_fullname):
pl = read_ply(pcd_fullname)
np_x = (np.array(pl[:,0], dtype=np.float32)).astype(np.float32)
np_y = (np.array(pl[:,1], dtype=np.float32)).astype(np.float32)
np_z = (np.array(pl[:,2], dtype=np.float32)).astype(np.float32)
np_i = (np.array(pl[:,3], dtype=np.float32)).astype(np.float32) / 256
points_32 = np.transpose(np.vstack((np_x, np_y, np_z, np_i)))
points_32.tofile(bin_fullname)
if __name__ == '__main__':
root_folder = "data/kitti/training/velodyne"
plylist = os.listdir(root_folder)
for plyfile in plylist:
plypath = os.path.join(root_folder, plyfile)
binpath = plypath.split('.')[0]+".bin"
# ply -> bin
pcd2bin(plypath,binpath)
完成了数据集文件的建立,需要对数据集进行初始化,即编写相对应的custom_dataset.py和.yaml文件。
仿照kitti_dataset.py进行修改,参考博文:https://blog.csdn.net/JulyLi2019/article/details/126351276
创建tools/cfgs/dataset_configs/cpte_dataset.yaml定义数据集相关参数
创建tools/cfgs/custom_models/pointpillar.yaml定义模型和训练相关参数
# 生成训练所需的pkl文件
python -m pcdet.datasets.custom.custom_dataset create_custom_infos /data/pl/OpenPCDet/tools/cfgs/dataset_configs/cpte_dataset.yaml
# 训练
python tools/train.py --cfg_file tools/cfgs/custom_models/pointpillar.yaml --batch_size=2 --epochs=300
训练结束后的验证方法,以及点云的坐标转换等,忙完毕业论文后又是工作上的如火如荼,后续尽量多多记录
文章浏览阅读1.2k次。vue2封装对话框el-dialog组件_
文章浏览阅读4.7k次,点赞5次,收藏6次。MFC 文本框换行 标签: it mfc 文本框1.将Multiline属性设置为True2.换行是使用"\r\n" (宽字符串为L"\r\n")3.如果需要编辑并且按Enter键换行,还要将 Want Return 设置为 True4.如果需要垂直滚动条的话将Vertical Scroll属性设置为True,需要水平滚动条的话将Horizontal Scroll属性设_c++ mfc同一框内输入二行怎么换行
文章浏览阅读832次。检查Linux是否是否开启所需端口,默认为6379,若未打开,将其开启:以root用户执行iptables -I INPUT -p tcp --dport 6379 -j ACCEPT如果还是未能解决,修改redis.conf,修改主机地址:bind 192.168.85.**;然后使用该配置文件,重新启动Redis服务./redis-server redis.conf..._redis-server doesn't support auth command or ismisconfigured. try
文章浏览阅读4.9k次。济大数电实验报告_数据选择器及其应用
文章浏览阅读1w次,点赞3次,收藏5次。题目描述所谓角谷猜想,是指对于任意一个正整数,如果是奇数,则乘3加1,如果是偶数,则除以2,得到的结果再按照上述规则重复处理,最终总能够得到1。如,假定初始整数为5,计算过程分别为16、8、4、2、1。程序要求输入一个整数,将经过处理得到1的过程输出来。输入 一个正整数N(N <= 2,000,000) 输出 从输入整数到1的步骤,每一步为一行,每一部中描述计算过程。最后一行输出"En..._角谷猜想c++代码
文章浏览阅读1k次。1: private float[,] LoadHeightData(Texture2D heightMap) 2: { 3: float minimumHeight = 255; 4: float maximumHeight = 0; 5: 6: int width = heightMap.Width; _positions indices 自动创建地形
文章浏览阅读5.5w次,点赞3次,收藏41次。加密方式主要有3种,摘要加密、对称加密与非对称加密。_java 加密 解密
文章浏览阅读6.5k次,点赞2次,收藏9次。sql语句中,以逗号分隔的字符串中是否包含某个特定字符串,类似于判断一个数组中是否包含某一个元素,例如:判断 ,a,b,c,d,e,f,g,中是否包含 a ,sql语句如何实现?SQL中没有类似VB中的Split的函数。此时,如要匹配 ,6,3,9,22,中的6。可用charindex函数巧妙实现:(注意红色的逗号)charindex( ',' +cast(6 as varchar)+ ',',‘,6,3,9,22,’)>0sqlserver:sqlserver中没有封装好的方法可以实现._sql server 逗号分隔包含
文章浏览阅读1.8k次。 第一次接触这个游戏,还是通过校内网。然后下载了一个单机绿色破解版的。感觉挺有意思,就慢慢玩了起来。后来打开游戏文件夹,发现base里面竟然都是一些xml、lua和jpg文件,感觉好熟悉,于是便开始了我的分析。 游戏目录结构如下:│ Around the World in 80 Days.exe│ bass.dll│ settings.ini│ ├─sc
文章浏览阅读8.4k次。百度普通IP定位是一套以HTTP/HTTPS形式提供的轻量级定位接口,用户可以通过该服务,根据IP定位来获取大致位置。1.首先需要成为百度地图开放平台的注册用户。http://lbsyun.baidu.com/index.php2. 申请Web服务API服务,在控制台创建一个应用:应用类型为浏览器端3.创建好应用后你,将会得到一个访问应用(AK).4.请求URLhttp://..._百度ip归属api
文章浏览阅读4.7k次。DataGrid颜色专题在Flex运用中经常提到的有关DataGrid问题是如何改变DataGrid单元格(cell),列(column)和行(row)的背景颜色(backgroundcolor)这里对这3种颜色做一个总结。 1. 设置行(row)的背景色 主要是通过对DataGrid扩展,对protected函数drawRowBackground()进行重写,具体代码如下:package com{ im_datagrid背景颜色
文章浏览阅读5k次。欢迎关注“小白玩转Python”,发现更多 “有趣”引言如果你刚刚开始学习 PyTorch,并想学习如何做一些基本的图像分类,那么这篇文章你一定不要错过哦~本文将通过组织自己的训练数据..._pytorch torchvision 训练自己的数据 图像分类