A lightweight network for phone surface defect detection with industrial deployment on RK3568 edge devices

Shulong Zhuo, Xiaojian Zhou, Jiajing Cai, Hao Bai, Xu Duan and YiQun Ma

PLOS ONE, 2025, vol. 20, issue 12, 1-27

Abstract: In smartphone production, detecting small defects on screens remains challenging due to low detection accuracy, high missed detection rates, and slow processing speeds. To address these issues, this paper presents a Lightweight Network Based on YOLOv8 (LNB-YOLO) for defect detection, with several key enhancements. First, a Feature Pyramid Network based on Context-Guided Spatial Feature Reconstruction (CGRFPN) is integrated to improve the perception of multi-level features and enhance small target recognition in complex backgrounds. Second, the Efficient Local Attention (ELA) module is incorporated into the Backbone’s C2F module to improve localization precision, while the Minimum Point Distance based IoU (MPDIoU) loss function is employed to prevent gradient explosion. Third, a lightweight Detail-Enhanced Convolution and Shared Convolutional Detection Head (LSDECD) is designed to capture fine details while reducing parameters and computational complexity. Finally, model pruning and knowledge distillation techniques are applied to further optimize efficiency. Experimental results on the PKU-Market-Phone dataset show that LNB-YOLO achieves a mAP@0.5 of 97.5% and a mAP@.5:.95 of 68.8%, surpassing the original YOLOv8 by 6.1% and 9.3%, respectively. The model also reduces parameters by 80% and computational cost by 63%, effectively meeting precision requirements for smartphone production lines.

Date: 2025
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0336971 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 36971&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0336971

DOI: 10.1371/journal.pone.0336971

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().