一种基于YOLO的交通场景下多目标检测方法和装置

摘要

本发明涉及一种基于YOLO的交通场景下多目标检测方法和装置，方法采用YOLO算法进行交通场景下的多目标检测，其特征在于，方法包括以下步骤：S1：获取数据集，并采用数据增广的方法扩增数据集；S2：对扩增后的数据集进行重新聚类，以更新用于YOLO算法的先验锚框尺寸；S3：将扩增后的数据集载入预先建立并训练好的改进YOLO算法中，获取数据集的目标检测结果；改进YOLO算法采用焦点损失作为损失函数进行模型训练与现有技术相比，本发明提高了模型的训练效果和检测效果，改善了样本类别不均衡的问题，具有检测速度快且检测精度高等优点。