基于YOLOv4改进算法的目标检测方法、装置及系统

摘要

本发明公开了一种基于YOLOv4改进算法的目标检测方法、装置及系统，方法包括获取YOLOv4‑FCSPX网络结构；获取含目标物体的图片集，包括训练子集和测试子集；对训练子集中图片分别进行目标物体标注，获得对应的标注框；对所有标注框进行聚类，获得K个先验框；将所有先验框和图片集中的原始图片输入至YOLOv4‑FCSPX网络结构，生成特征图，得到特征图对应先验框的物体置信度；基于特征图对应先验框的物体置信度筛选出一定数量的候选框；对所有候选框进行非极大值抑制得到预测框；基于各预测框与标注框的损失值对YOLOv4‑FCSPX网络结构的权值进行更新；将测试子集中的图片输入至训练好的YOLOv4‑FCSPX网络结构，获得目标物体的大小、位置及类别。本发明能够大大提高对小目标的检测精度。