Maturity Classification of “Hupingzao” Jujubes with an Imbalanced Dataset Based on Improved MobileNet V2

Haixia Sun, Shujuan Zhang (), Rui Ren and Liyang Su
Additional contact information
Haixia Sun: College of Agricultural Engineering, Shanxi Agricultural University, Jinzhong 030801, China
Shujuan Zhang: College of Agricultural Engineering, Shanxi Agricultural University, Jinzhong 030801, China
Rui Ren: College of Agricultural Engineering, Shanxi Agricultural University, Jinzhong 030801, China
Liyang Su: College of Agricultural Engineering, Shanxi Agricultural University, Jinzhong 030801, China

Agriculture, 2022, vol. 12, issue 9, 1-16

Abstract: Fruits with various maturity levels coexist among the harvested jujubes, and have different tastes and uses. Manual grading has a low efficiency and a strong subjectivity. The number of “Hupingzao” jujubes between different maturity levels is unbalanced, which affects the performance of the classifier. To solve the above issue, the class balance loss (CB) was used to improve the MobileNet V2 network, and a transfer learning strategy was used to train the model. The model was optimized based on the selection of an optimizer and learning rate. The model achieved the best classification results using the AdamW optimizer and a learning rate of 0.0001. The application of transfer learning and class balance loss improved the model’s performance. The precision was 96.800~100.000%, the recall was 95.833~100.000%, and the F1 score was 0.963~1.000. To compare the CB-MobileNet V2 performance, the CB-AlexNet, CB-GoogLeNet, CB-ShuffleNet, CB-Inception V3, CB-ResNet 50, and CB-VGG 16 with transfer learning were used to build classification models. Achieving a validation accuracy of 99.058%, and a validation loss value of 0.055, the CB-MobileNet V2 model showed a better overall performance compared with other models. The maturity detection system of “Hupingzao” jujubes was developed to test the model. The testing accuracy of the CB-MobileNet V2 model was 99.294%. The research indicates that the CB-MobileNet V2 model improves the performance of maturity classification, and provides a theoretical basis for intelligent classification of the quality of “Hupingzao” jujubes.

Keywords: jujube; imbalanced dataset; MobileNet V2; maturity; classification (search for similar items in EconPapers)
JEL-codes: Q1 Q10 Q11 Q12 Q13 Q14 Q15 Q16 Q17 Q18 (search for similar items in EconPapers)
Date: 2022
References: View references in EconPapers View complete reference list from CitEc
Citations: View citations in EconPapers (1)

Downloads: (external link)
https://www.mdpi.com/2077-0472/12/9/1305/pdf (application/pdf)
https://www.mdpi.com/2077-0472/12/9/1305/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jagris:v:12:y:2022:i:9:p:1305-:d:897770

Access Statistics for this article

Agriculture is currently edited by Ms. Leda Xuan

More articles in Agriculture from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().