The Rapid and Accurate Detection of Kidney Bean Seeds Based on a Compressed Yolov3 Model

Yu Wang, Hongyi Bai (), Laijun Sun, Yan Tang, Yonglong Huo and Rui Min
Additional contact information
Yu Wang: College of Electronics and Engineering, Heilongjiang University, Harbin 150080, China
Hongyi Bai: College of Electronics and Engineering, Heilongjiang University, Harbin 150080, China
Laijun Sun: College of Electronics and Engineering, Heilongjiang University, Harbin 150080, China
Yan Tang: College of Electronics and Engineering, Heilongjiang University, Harbin 150080, China
Yonglong Huo: College of Electronics and Engineering, Heilongjiang University, Harbin 150080, China
Rui Min: Center for Cognition and Neuroergonomics, State Key Laboratory of Cognitive Neuroscience and Learning, Beijing Normal University at Zhuhai, Zhuhai 519087, China

Agriculture, 2022, vol. 12, issue 8, 1-21

Abstract: Due to their rich nutritional value, kidney beans are considered one of the major products of international agricultural trade. The conventional method used for the manual detection of seeds is inefficient and may damage the test object. To locate and classify different kidney bean seeds rapidly and accurately, the Yolov3 network has been improved to realize seed detection in the current paper. Firstly, a dataset of 10 varieties of kidney bean seeds was produced and 1292 images were collected. Then, the dataset was divided into the training, validation, and test sets with the assigned ratio of 8:1:1. The kidney bean seeds dataset was trained using the Yolov3 model. Additionally, the implemented speed needed to be guaranteed while satisfying the detection accuracy. To meet such detection requirements, the Yolov3 model was pruned using the scaling factors of the batch normalization layer as a measure of channel importance, and finally fine-tuned with the aid of knowledge distillation. Then, the Yolov3, Yolov3-tiny, Yolov4, and the improved Yolov3 were used to detect the images in the test set. Subsequently, the performances of these four networks were compared. The results show that the model pruning method can compress the model to a great extent, and the number of model parameters is reduced by 98%. The detection time is shortened by 59%, and the average accuracy reaches 98.33%. Considering the speed and mAP, the improved Yolov3 detected the best results. The experimental results demonstrate that the method can accomplish the rapid and accurate detection of kidney bean seeds. It can provide a solid foundation for the marketing and planting of kidney bean seeds.

Keywords: kidney bean seeds detection; Yolov3; model compression; knowledge distillation (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/8/1202/pdf (application/pdf)
https://www.mdpi.com/2077-0472/12/8/1202/ (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:8:p:1202-:d:885889

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 ().