[人工智能]jetson nano 用 tensorrt 运行 nanodet(kitti数据集)

题目是目标,我们先一步一步来做,第一步是训练神经网络, 我用的是 kitti 数据集,训练顺序为,第一步,拿到kitti数据集,第二步,把kitti数据集修改为voc数据集格式,第三步,修改配置文件进行训练

关于前两步,是copy自

KITTI车辆检测数据集转VOC格式（亲测成功，附KITTI云盘连接以及完整格式转换代码）- KITTI车辆检测数据集看着一篇就够了！_Hurri_cane的博客-CSDN博客_kitti数据集车辆检测

他这个转xml的脚本稍微改改路径就可以直接运行了,然后就是训练,修改一下配置文件即可开始训练,参考我写的笔记

nanodet训练voc数据集_zhqh100的博客-CSDN博客_nanodet 训练voc

我就不全粘贴了,只粘贴一小部分配置文件

num_classes: 4
...
num_classes: 4
...
class_names: &class_names ['Car', 'Pedestrian', 'Cyclist', 'Truck']
....
img_path: /home/luke/Download/kitti_object/training/train/image_2  

开通VIP 解锁文章