Fruit Detection Research Based on Near-Infrared Spectroscopy and Lightweight Neural Network
jinfu zhang,bin tang,jianxu wang,linfeng f. cai,junfeng miao,qing chen,hang liu,mingfu zhao
DOI: https://doi.org/10.1117/12.2656581
2022-01-01
Abstract:Spectroscopy is an important method in the field of nondestructive testing of fruits, and spectral pretreatment and feature screening are the difficulties and keys, which have a greater impact on the accuracy of the model, so this paper aims to use the lightweight neural network (MoblieNetV1) combined with near-infrared spectroscopy to improve the accuracy and stability of the model, reduce its computational amount, and verify its feasibility and effect. First, the fruit dataset is classified and labeled, and apples, pears, citrus, and peaches are labeled as 1, 2, 3, and 4, respectively, and then the sample spectra are put into model training in batches. MoblieNetV1 extracts features through the bottom layer, and then further extracts features at a deeper level, and finally obtains the classification of the target. Experimental results show that the moblieNetV1 used has achieved a high recognition accuracy, which is more suitable for use in the field of spectral detection than other neural networks, and avoids the cumbersome process of manually extracting features.