Sep, 2023

MFL-YOLO: 损坏交通标志物的目标检测模型

TL;DR我们提出了基于 YOLOv5s 的改进目标检测方法 MFL-YOLO(互补特征级别损失增强 YOLO),通过设计简单的跨级别损失函数,使得模型的每个级别都具有自己的作用,有助于学习更多多样化的特征并提高细粒度。相比 YOLOv5s,我们的 MFL-YOLO 在 F1 得分和 mAP 上分别提高了 4.3 和 5.1,同时减少了 8.9% 的 FLOPs。Grad-CAM 热图可视化显示我们的模型能更好地关注损坏交通标志的局部细节。此外,我们还对 CCTSDB2021 和 TT100K 进行了实验证明我们模型的泛化性。