轻量级模型P-YOLOv8实现高精度实时分心驾驶检测
P-YOLOv8: Efficient and Accurate Real-Time Detection of Distracted Driving
- 基于预训练YOLOv8优化,架构创新实现小模型大性能
- 仅2.84MB模型大小,99.46%准确率,支持嵌入式部署
- 适合资源受限设备,适用于实时交通安全监控场景
分心驾驶是全球范围内的重大安全隐患。本文提出一种高效且实时的机器学习检测系统P-YOLOv8,针对传统模型计算成本高、延迟大的问题进行优化。在State Farm的分心驾驶员检测数据集上,该模型涵盖22,424张图像和10类行为,实现了99.46%的高准确率。其模型仅2.84 MB,参数量1,451,098,显著低于VGG16、VGG19、ResNet等传统模型。得益于创新架构,P-YOLOv8在保持高精度的同时大幅降低计算开销,为低成本嵌入式设备上的实时部署提供了可行方案。实验验证了其在物体检测与图像分类任务中的优异表现,展现出在TinyML场景下的应用潜力。
原文摘要 · Abstract (English)
Distracted driving is a critical safety issue that leads to numerous fatalities and injuries worldwide. This study addresses the urgent need for efficient and real-time machine learning models to detect distracted driving behaviors. Leveraging the Pretrained YOLOv8 (P-YOLOv8) model, a real-time object detection system is introduced, optimized for both speed and accuracy. This approach addresses the computational constraints and latency limitations commonly associated with conventional detection models. The study demonstrates P-YOLOv8 versatility in both object detection and image classification tasks using the Distracted Driver Detection dataset from State Farm, which includes 22,424 images across ten behavior categories. Our research explores the application of P-YOLOv8 for image classification, evaluating its performance compared to deep learning models such as VGG16, VGG19, and ResNet. Some traditional models often struggle with low accuracy, while others achieve high accuracy but come with high computational costs and slow detection speeds, making them unsuitable for real-time applications. P-YOLOv8 addresses these issues by achieving competitive accuracy with significant computational cost and efficiency advantages. In particular, P-YOLOv8 generates a lightweight model with a size of only 2.84 MB and a lower number of parameters, totaling 1,451,098, due to its innovative architecture. It achieves a high accuracy of 99.46 percent with this small model size, opening new directions for deployment on inexpensive and small embedded devices using Tiny Machine Learning (TinyML). The experimental results show robust performance, making P-YOLOv8 a cost-effective solution for real-time deployment. This study provides a detailed analysis of P-YOLOv8's architecture, training, and performance benchmarks, highlighting its potential for real-time use in detecting distracted driving.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。