-
公开(公告)号:CN117373057A
公开(公告)日:2024-01-09
申请号:CN202311141052.6
申请日:2023-09-06
Applicant: 无锡学院
IPC: G06V40/10 , G06V10/764 , G06V10/82 , G06V10/774 , G06V10/80 , G06N3/0464 , G06N3/08
Abstract: 本发明公开一种目标行人检测方法、系统、电子设备及介质,包括:S1:获取目标行人图像数据集,将目标行人图像数据集分为训练集和测试集;S2:设置目标行人检测模型:以YOLOv5算法为基本架构,对YOLOv5算法进行改进:将CBAM模块融合进YOLOv5主干网络中,替换掉原主干网络的C3模块;将改进后的FuseSoft‑NMS替换NMS,有利于更好的利用目标检测框的位置信息,提高目标定位的精度;S3:利用训练集对目标行人检测模型进行训练,直到所述图像分类模型的目标损失函数收敛或达到最大训练轮次,以获得训练好的图像分类模型;S4:利用训练好的目标行人检测模型对测试集进行检测,输出行人目标的目标行人图像检测结果,提高目标行人检测精度。