基于轻量卷积网络的马铃薯外部缺陷无损分级

目前马铃薯外部缺陷检测方法主要依靠人工提取特征,且检测精度不高,为了更好地对马铃薯外部缺陷进行快速、准确在线分级,本实验提出一种基于轻量卷积网络的在线分级方法.首先,利用ImageNet数据集训练Xception网络模型,建立马铃薯预训练网络模型;然后,重新构建5类缺陷全连接层,并通过迁移学习在预训练网络模型上训练马铃...

Nondestructive Classification of Defects in Potatoes Based on Lightweight Convolutional Neural Network

At present, the detection of external defects in potatoes mainly depends on manual feature extraction and is consequently inaccurate. A classification method for fast and accurate online grading of potatoes based on the lightweight convolutional neural network is proposed in this paper. First, the Xception network model was trained using ImageNet dataset to establish a pre-trained network model. Then, based on the trained Xception network, this method replaced the Softmax classifier in the original Xception network with category 5 tags, and the potato defect dataset was trained in the Xception framework with transfer learning．Finally, based on the trained external defect recognition model, the classification performance of 5 tags defects was tested. The experimental results showed that when the learning rate was 0.000 01, the overall performance of the network model was optimal, the training accuracy rate was 98.88%, and the loss value was 0.034 9. Compared with nine other neural networks with different depths, the proposed lightweight network model had better recognition performance with average recognition accuracy of 96.04% under the same sample conditions. The model’s processing time was shorter than that of the ResNet152 network, with better recognition effect, and the recognition rate of the network model was 6.4 frames/s. This study could provide theoretical support for online classification of potatoes.