基于双向注意力机制增强YOLO V5的目标检测方法

    公开(公告)号:CN114282649A

    公开(公告)日:2022-04-05

    申请号:CN202111519608.1

    申请日:2021-12-14

    Abstract: 本发明属于机器视觉领域,公开了基于双向注意力机制的YOLOV5的目标检测方法,将数据集按比例分为训练数据集和测试数据集;对训练数据集的图片归一化为固定大小的图片;将归一化后的图片先输入到增加输出层的骨干网络,再经过双向注意力机制模块后,在使用改进的损失函数计算的YOLOv5网络中训练和测试;将归一化后的实时数据图片输入到改进后的YOLOv5网络中,输出目标检测结果。本发明提高了对目标的检测能力,解决网络对特征差异不敏感的问题,解决训练过程目标框回归不稳定的问题,保证了anchor框和目标框之间的长宽比的一致性。

Patent Agency Ranking