1.首先去官网下载对应的工具库https://github.com/linClubs/nuscenes2kitti.git
2.官网有具体的教程,不过编译的时候需要删除nus2kitti_ws/src的Cmake文件,除此之外在命令窗口进行软链接export LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libffi.so.7
3.详细过程如下:
# 记得切一个能运行BEVFusion,fastbev,bevdet,mmdetetion3d的python虚拟环境运行该工程
# 报错需要什么库就编译什么库# 1 激活自己的虚拟环境 本人使用bevdet
conda activate bevdet# 2 安装依赖包
pip install pyyaml nuscenes-devkit python-opencv pypcd# 3 安装ros发布box的依赖 需要ros可视化才需要安装
pip install pyyaml rospkg empy catkin_pkg
sudo apt install ros-noetic-jsk-recognition ros-noetic-jsk-rviz-plugins# 4 创建工作空间
mkdir -p nus2kitti_ws/src && cd nus2kitti_ws/src# 5 拉取源码
git clone https://github.com/linClubs/nuscenes2kitti.git# 6 编译
cd nus2kitti_ws && catkin_make# 7 运行时报错
ModuleNotFoundError: No module named 'cStringIO'
# 修改如下 python3中StringIO是从io导入
try:
from cStringIO import StringIOfrom io import StringIO
# 1. 只发布前视图和点云
roslaunch nus_pkg nus_pub_cloud.launch# 2. 发布环视,点云, 3Dbox
roslaunch nus_pkg nus_pub_box3d.launch# 3 nus2kitti
rosrun nus_pkg nus2kitti.py
# 或者
cd nus_pkg/scripts
python nus2kitti.py
4.这个库写的超详细还可以制作自己的数据集,可以自己研究研究
5.之后运行后生成的标签是这样的,我们进行转换
6.之后我们写个下面的脚本转换一下
nuscenes2kitti
7.转换之后就是正常的数据集格式:
8.最后一步重要的是需要改变角度要不最后检测结果会偏90°,相应的程序代码为:
import os
import math# 文件夹路径
folder_path = '/home/pc/2026BOSS/OpenPCDet/data/custom/training/label_2'# 遍历文件夹中的所有文件
for filename in os.listdir(folder_path):# 只处理 .txt 文件if filename.endswith('.txt'):file_path = os.path.join(folder_path, filename)# 读取文件内容with open(file_path, 'r') as file:lines = file.readlines()# 处理每一行的最后一个数字new_lines = []for line in lines:parts = line.strip().split()if parts:# 将最后一个数字加上 pi/2parts[-1] = str(float(parts[-1]) + math.pi / 2)new_line = ' '.join(parts)new_lines.append(new_line)# 将处理后的内容写回文件with open(file_path, 'w') as file:file.write('\n'.join(new_lines) + '\n')print("处理完成")
9 ,转换之后的标签如下所示:
10.这里写了一个点云可视化脚本,比较方便。
import sys
import os
import numpy as np
from PyQt5.QtWidgets import (QApplication, QWidget, QPushButton, QLabel,QFileDialog, QVBoxLayout, QHBoxLayout, QComboBox, QSlider
)
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QScreen
from mayavi.core.ui.api import MayaviScene, SceneEditor
from mayavi.tools.mlab_scene_model import MlabSceneModel
from traits.api import HasTraits, Instance
from traitsui.api import View, Item, Group# 嵌入 Mayavi 的 Traits 视图
class MayaviView(HasTraits):scene = Instance(MlabSceneModel, ())view = View(Group(Item('scene', editor=SceneEditor(scene_class=MayaviScene), show_label=False),),resizable=True,width=800,height=600)def __init__(self, **kwargs):super(MayaviView, self).__init__(**kwargs)# 设置背景颜色为黑色self.scene.mayavi_scene.scene.background = (0, 0, 0)def plot(self, x, y, z, point_size):# 清空当前图self.scene.mlab.clf()# 绘制所有点为白色self.scene.mlab.points3d(x, y, z,mode='point',color=(1, 1, 1), # 白色scale_factor=point_size,figure=self.scene.mayavi_scene)class PointCloudViewer(QWidget):def __init__(self):super().__init__()self.setWindowTitle("BIN 点云可视化")# 设置窗口大小为屏幕大小的百分比以实现自适应screen = QScreen.availableGeometry(QApplication.primaryScreen())self.resize(screen.width() * 0.8, screen.height() * 0.8)self.mayavi_widget = Noneself.x = self.y = self.z = None# 主布局main_layout = QVBoxLayout()# 文件标签和按钮布局file_layout = QHBoxLayout()self.label = QLabel("请选择一个 .bin 文件")file_layout.addWidget(self.label)self.select_button = QPushButton("打开文件")self.select_button.clicked.connect(self.open_file)file_layout.addWidget(self.select_button)main_layout.addLayout(file_layout)# 点尺寸滑块布局config_layout = QHBoxLayout()config_layout.addWidget(QLabel("点尺寸:"))self.size_slider = QSlider(Qt.Horizontal)self.size_slider.setMinimum(1)self.size_slider.setMaximum(20)self.size_slider.setValue(5)self.size_slider.valueChanged.connect(self.update_plot)config_layout.addWidget(self.size_slider)main_layout.addLayout(config_layout)self.setLayout(main_layout)def open_file(self):filename, _ = QFileDialog.getOpenFileName(self, "选择点云 .bin 文件", "", "BIN files (*.bin)")if filename:self.label.setText(f"已加载: {os.path.basename(filename)}")self.load_bin(filename)self.show_mayavi()def load_bin(self, filepath):raw = np.fromfile(filepath, dtype=np.float32)if raw.size % 4 == 0:pc = raw.reshape(-1, 4)elif raw.size % 3 == 0:pc = raw.reshape(-1, 3)pc = np.concatenate([pc, np.zeros((pc.shape[0], 1), dtype=np.float32)], axis=1)else:raise ValueError(f"无法识别的 BIN 格式,总元素数 {raw.size}")self.x, self.y, self.z = pc[:, 0], pc[:, 1], pc[:, 2]def show_mayavi(self):if self.mayavi_widget is None:self.mayavi_view = MayaviView()self.mayavi_widget = self.mayavi_view.edit_traits(parent=self, kind='subpanel').control# 将 Mayavi 控件添加到主布局下方self.layout().addWidget(self.mayavi_widget)self.update_plot()def update_plot(self):if self.x is None:returnpoint_size = self.size_slider.value() * 0.1 # 调整缩放因子self.mayavi_view.plot(self.x, self.y, self.z, point_size)if __name__ == '__main__':app = QApplication(sys.argv)viewer = PointCloudViewer()viewer.show()sys.exit(app.exec_())
11.训练后的检测效果: