基于YOLOv5的公共场景宠物犬牵绳检测

作者:耿帅帅; 廖涛; 刘唤唤
来源:现代信息科技, 2022, 6(11): 99-102.
DOI:10.19850/j.cnki.2096-4706.2022.011.025

摘要

针对公共场景下人们带宠物犬散步不牵绳而导致其袭击行人的问题,提出了一种基于YOLOv5的改进模型来提升对宠物犬检测的效果。以传统的YOLOv5检测模型为基础对其进行改进,结合坐标注意力模块SE,解决了模型对次要信息过度关注的问题,模型的精度得到提升。实验结果表明改进模型在自制数据集的平均精度得到提升,有效证明了该算法的有效性。

全文