基于深度学习的水果缺陷实时检测方法

Real time detection method of fruit defects based on deep learning

目的:对CenterNet方法进行优化改进。方法:使用MobileNetV3的轻量化卷积神经网络替代CenterNet原有的骨干网络,加快检测速度,对MobileNetV3模块进行改进,增强模型对水果中小缺陷块的检测能力,并对CenterNet的预检测阶段进行优化,提高其检测准确度。结果:试验方法对显著缺陷如直径>4 mm的苹果识别率高达99.7%,检测速度为113帧/s,模型体积为1.31 MB。结论:与CenterNet_ResNet18和CenterNet_Shuffler模型相比,MO-CenterNet模型检测水果缺陷在训练时间、检测速度和准确率方面均衡性更好。

Objective:This study aimed to optimize and improve the CenterNet method for detecting fruit defects.Methods:the lightweight convolutional neural network of MobileNetV3 was used to replace the original backbone network of CenterNet accelerate the detection speed,improve the module of MobileNetV3,enhance the detection ability of the model for small and medium-sized defective blocks of fruit,and optimize the pre detection stage of CenterNet to increase its detection accuracy.Results:The recognition rate of significant defects such as apples with diameter>4 mm was 99.7%,and the detection speed was 113 FPS,with the model volume of 1.31 MB.Conclusion:Compared with models CenterNet_Resnet18 and CenterNet_Shuffler,model MO-CenterNet has better balance in training time,detection speed and accuracy.