基于YOLOv5手机目标检测算法
首发时间:2024-04-03
摘要:随着移动设备的普及,移动手机的应用领域也日益多样化,然而在一些特定场所,手机的使用存在限制。手机因为目标小、背景干扰多,在检测过程中容易产生误检、漏检。针对这些难题,本文改进了YOLOv5s算法用于手机目标检测。该算法设计了基于Focal Loss的少数样本友好的损失函数,通过调节样本的权重,解决了数据集中手机目标占比较大、相似干扰目标数量较少的问题,从而提高了检测精度。改进了传统的NMS方法,通过提高手机目标框的置信度、各个类别检测框整体进行IoU筛选以及降低手机目标重叠区域的IoU判断阈值,有效地提高了手机目标的权重,减少了漏检情况。实验证明,采取的改进措施在自行采集的手机检测数据上取得了良好的效果,为手机的检测提供了有力的支撑。
关键词: 计算机应用技术 手机目标检测 损失函数 极大值抑制
For information in English, please click here
Mobile Object Detection Algorithm based on YOLOv5
Abstract:With the popularization of mobile devices, the application field of mobile phones is becoming increasingly diverse.However, the use of mobile phones is restricted in some specific places. Due to small object and high background interference, mobile phones are prone to false alarms and missed detections during the detection process. To address these challenges, this paper proposes improvements to the YOLOv5s algorithm for mobile phone object detection.The algorithm designs a focal loss-based loss function friendly to minority samples, which adjusts the weights of samples to address the problem of a relatively large proportion of mobile phone objects and a small number of similar interfering objects in the dataset, thereby improving detection accuracy. The traditional NMS method is also improved by enhancing the confidence of mobile phone bounding boxes, uniformly comparing bounding boxes of different categories, and reducing the IoU threshold for determining overlapping areas of mobile phone objects. This effectively increases the weight of mobile phone objects and reduces missed detections.Experimental results demonstrate that the proposed improvements achieve satisfactory results on self-collected mobile phone detection data, providing strong support for mobile phone detection.
Keywords: Computer Applocation Technology Mobile Phone Object Detection Minority-friendly Loss Function Non-Maximum Suppression
基金:
引用

No.****
动态公开评议
共计0人参与
勘误表
基于YOLOv5手机目标检测算法
评论
全部评论