摘要详情

ID / 提交时间

46 / 2024-08-15 12:21:09

标题

CDC-YOLO for Gear Defect Detection: Make YOLOv8 Faster and Lighter on RK3588

关键字

defect detection, YOLO, dual convolution, embedded platform.

主题及专题

lAI Applications in Engineering

状态

全文录用

作者

YaoJiachen / Nanjing University of Science and Technology；School of Mechanical Engineering

WangManyi / School of Mechanical Engineering； NanJing University of Science and Technology

MaoXujing / The King’s School

摘要

Defect detection on the gear surface is crucial for preventing faults in mechanical systems. However, most detection models are not extremely effective on embedded platforms. To address this issue, we present a lightweight detection model called CDC-YOLO, which is based on YOLOv8 and specifically designed for embedded platforms. It utilizes our proposed CDC module as a residual structure for extracting multi-scale features, allowing for better adaptation to different platforms. Additionally, we achieve model lightweighting by using a dual convolutional architecture. Experimental results on both computers and embedded platforms demonstrate that our proposed method outperforms the baseline YOLOv8.