现代电子技术2024，Vol.47Issue(3)：51-58,8.DOI:10.16652/j.issn.1004-373x.2024.03.010
基于深度学习的陪护机器人目标检测方法
Object detection method of escort robot based on deep learning
汪振耀 1张礼华 1郑俭1
作者信息
- 1. 江苏科技大学长山校区 机械工程学院, 江苏 镇江 212100
- 折叠
摘要
Abstract
In view of the false detection and missed detection when escort robots carry out object detection tasks in complex environments,an escort robot object detection method based on deep learning is proposed.The improved WGAN network is used to generate high-quality images to enrich the diversity of data sets and lay a foundation for subsequent YOLOv5s training.The Gridmask data enhancement technology is used for the original YOLOv5s model to improve the recognition ability of the detection model for complex scenes and occlusion.In the part of Backbone,the module C2fSE is adopted,and LSTM network and YOLOv5s network are combined to reduce the number of network parameters,obtain more context information,and improve the accuracy of object detection.The dice coefficient(DC)loss is added to the total loss function.In comparison with the original algorithm,the detection accuracy,recall rate and mAP of the improved algorithm increases by 6.8%,4.6%and 8.3%,respectively.The experimental results show that the improved algorithm has been improved in the detection accuracy and missed detection of small objects and partially-occluded objects,which is suitable for the object detection by household escort robot.关键词
目标检测/YOLOv5s/长短时记忆网络/WGAN/C2fSE/数据增强/损失函数Key words
object detection/YOLOv5s/LSTM network/WGAN/C2fSE/data enhancement/loss function分类
信息技术与安全科学引用本文复制引用
汪振耀,张礼华,郑俭..基于深度学习的陪护机器人目标检测方法[J].现代电子技术,2024,47(3):51-58,8.
