-
公开(公告)号:CN114282649A
公开(公告)日:2022-04-05
申请号:CN202111519608.1
申请日:2021-12-14
Applicant: 江苏省特种设备安全监督检验研究院
IPC: G06N3/04 , G06N3/08 , G06V10/774 , G06V10/80 , G06V10/74
Abstract: 本发明属于机器视觉领域,公开了基于双向注意力机制的YOLOV5的目标检测方法,将数据集按比例分为训练数据集和测试数据集;对训练数据集的图片归一化为固定大小的图片;将归一化后的图片先输入到增加输出层的骨干网络,再经过双向注意力机制模块后,在使用改进的损失函数计算的YOLOv5网络中训练和测试;将归一化后的实时数据图片输入到改进后的YOLOv5网络中,输出目标检测结果。本发明提高了对目标的检测能力,解决网络对特征差异不敏感的问题,解决训练过程目标框回归不稳定的问题,保证了anchor框和目标框之间的长宽比的一致性。