一种目标行人检测方法、系统、电子设备及介质

    公开(公告)号:CN117373057A

    公开(公告)日:2024-01-09

    申请号:CN202311141052.6

    申请日:2023-09-06

    Applicant: 无锡学院

    Abstract: 本发明公开一种目标行人检测方法、系统、电子设备及介质,包括:S1:获取目标行人图像数据集,将目标行人图像数据集分为训练集和测试集;S2:设置目标行人检测模型:以YOLOv5算法为基本架构,对YOLOv5算法进行改进:将CBAM模块融合进YOLOv5主干网络中,替换掉原主干网络的C3模块;将改进后的FuseSoft‑NMS替换NMS,有利于更好的利用目标检测框的位置信息,提高目标定位的精度;S3:利用训练集对目标行人检测模型进行训练,直到所述图像分类模型的目标损失函数收敛或达到最大训练轮次,以获得训练好的图像分类模型;S4:利用训练好的目标行人检测模型对测试集进行检测,输出行人目标的目标行人图像检测结果,提高目标行人检测精度。

Patent Agency Ranking