-
公开(公告)号:CN118609087A
公开(公告)日:2024-09-06
申请号:CN202410728911.X
申请日:2024-06-06
Applicant: 云南省交通科学研究院有限公司
IPC: G06V20/58 , G06V10/40 , G06V10/80 , G06V10/764 , G06V10/82 , G06N3/0464 , G06N3/08
Abstract: 本发明涉及热红外交通目标检测技术领域,且公开了一种基于改进YOLOv8的热红外交通目标检测方法及设备。该基于改进YOLOv8的热红外交通目标检测方法收集交通场景下的热红外图像数据;基于Ak_C2f模块、MBConv模块、GAM模块和Shape‑IoU损失函数对YOLOv8网络模型进行改进,得到改进YOLOv8热红外交通目标检测模型;将收集的热红外数据集输入至改进的YOLOv8网络模型进行训练,得到权重文件;利用获得的权重文件对测试集进行测试,得到测试集中各个目标类别的锚框和置信度。在极端天气覆盖的交通场景下,既可以确保交通检测任务的实时性,也能够准确对目标类别进行识别,提升了目标检测的鲁棒性。