Study on Spark Image Detection for Abrasive Belt Grinding via Transfer Learning with YOLOv8

Aiming to solve the problems of low precision and poor efficiency caused by relying on manual experience during the manual polishing of blades, a multi-view spark image detection method based on YOLOv8 transfer learning is proposed. A multi-pose spark image dataset including front, side, and 45° ang...

Full description

Saved in:
Bibliographic Details
Main Authors: Jian Huang, Guangpeng Zhang
Format: Article
Language:English
Published: MDPI AG 2025-05-01
Series:Sensors
Subjects:
Online Access:https://www.mdpi.com/1424-8220/25/9/2946
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1849312664981864448
author Jian Huang
Guangpeng Zhang
author_facet Jian Huang
Guangpeng Zhang
author_sort Jian Huang
collection DOAJ
description Aiming to solve the problems of low precision and poor efficiency caused by relying on manual experience during the manual polishing of blades, a multi-view spark image detection method based on YOLOv8 transfer learning is proposed. A multi-pose spark image dataset including front, side, and 45° angle views is constructed, and the cross-view detection task is achieved for the first time. The generalization ability of the model is enhanced through the following innovative strategies: (1) a cross-view transfer learning framework based on dynamic anchor box optimization is designed, and the parameters of the front spark detection model YOLOv8 are transferred to the side and 45°-angle detection tasks; (2) an attention-guided feature alignment module is introduced to alleviate the feature distribution shift caused by view differences; and (3) a curriculum learning strategy is adopted, where the datasets of different views are trained separately first and then sampled to reconstruct the dataset for further training, gradually increasing the weight of samples from complex views. The experimental results show that on the self-built multi-view dataset (containing 3000 annotated images), this method achieves an average detection accuracy of 98.7%, which is 14.2% higher than that of the original YOLOv8 model. The inference speed reaches 55 FPS on an NVIDIA RTX 4090, meeting the requirements of industrial online monitoring. The research results provide key technical support for the intelligent prediction of the material removal rate in the precision machining of blades and have the potential for rapid deployment in industrial scenarios.
format Article
id doaj-art-644fb27306ef4b43b53fa4ef299f3d8b
institution Kabale University
issn 1424-8220
language English
publishDate 2025-05-01
publisher MDPI AG
record_format Article
series Sensors
spelling doaj-art-644fb27306ef4b43b53fa4ef299f3d8b2025-08-20T03:53:01ZengMDPI AGSensors1424-82202025-05-01259294610.3390/s25092946Study on Spark Image Detection for Abrasive Belt Grinding via Transfer Learning with YOLOv8Jian Huang0Guangpeng Zhang1School of Mechanical and Precision Instrument Engineering, Xi’an University of Technology, Xi’an 710048, ChinaSchool of Mechanical and Precision Instrument Engineering, Xi’an University of Technology, Xi’an 710048, ChinaAiming to solve the problems of low precision and poor efficiency caused by relying on manual experience during the manual polishing of blades, a multi-view spark image detection method based on YOLOv8 transfer learning is proposed. A multi-pose spark image dataset including front, side, and 45° angle views is constructed, and the cross-view detection task is achieved for the first time. The generalization ability of the model is enhanced through the following innovative strategies: (1) a cross-view transfer learning framework based on dynamic anchor box optimization is designed, and the parameters of the front spark detection model YOLOv8 are transferred to the side and 45°-angle detection tasks; (2) an attention-guided feature alignment module is introduced to alleviate the feature distribution shift caused by view differences; and (3) a curriculum learning strategy is adopted, where the datasets of different views are trained separately first and then sampled to reconstruct the dataset for further training, gradually increasing the weight of samples from complex views. The experimental results show that on the self-built multi-view dataset (containing 3000 annotated images), this method achieves an average detection accuracy of 98.7%, which is 14.2% higher than that of the original YOLOv8 model. The inference speed reaches 55 FPS on an NVIDIA RTX 4090, meeting the requirements of industrial online monitoring. The research results provide key technical support for the intelligent prediction of the material removal rate in the precision machining of blades and have the potential for rapid deployment in industrial scenarios.https://www.mdpi.com/1424-8220/25/9/2946yolov8deep learningtransfer learningspark image
spellingShingle Jian Huang
Guangpeng Zhang
Study on Spark Image Detection for Abrasive Belt Grinding via Transfer Learning with YOLOv8
Sensors
yolov8
deep learning
transfer learning
spark image
title Study on Spark Image Detection for Abrasive Belt Grinding via Transfer Learning with YOLOv8
title_full Study on Spark Image Detection for Abrasive Belt Grinding via Transfer Learning with YOLOv8
title_fullStr Study on Spark Image Detection for Abrasive Belt Grinding via Transfer Learning with YOLOv8
title_full_unstemmed Study on Spark Image Detection for Abrasive Belt Grinding via Transfer Learning with YOLOv8
title_short Study on Spark Image Detection for Abrasive Belt Grinding via Transfer Learning with YOLOv8
title_sort study on spark image detection for abrasive belt grinding via transfer learning with yolov8
topic yolov8
deep learning
transfer learning
spark image
url https://www.mdpi.com/1424-8220/25/9/2946
work_keys_str_mv AT jianhuang studyonsparkimagedetectionforabrasivebeltgrindingviatransferlearningwithyolov8
AT guangpengzhang studyonsparkimagedetectionforabrasivebeltgrindingviatransferlearningwithyolov8