Improving Walnut Images Segmentation Using Modified UNet3+ Algorithm

Jun Tie, Weibo Wu, Lu Zheng (), Lifeng Wu and Ting Chen
Additional contact information
Jun Tie: College of Computer Science, South-Central Minzu University, Wuhan 430074, China
Weibo Wu: College of Computer Science, South-Central Minzu University, Wuhan 430074, China
Lu Zheng: College of Computer Science, South-Central Minzu University, Wuhan 430074, China
Lifeng Wu: College of Computer Science, South-Central Minzu University, Wuhan 430074, China
Ting Chen: College of Computer Science, South-Central Minzu University, Wuhan 430074, China

Agriculture, 2024, vol. 14, issue 1, 1-16

Abstract: When aiming at the problems such as missed detection or misdetection of recognizing green walnuts in the natural environment directly by using target detection algorithms, a method is proposed based on improved UNet3+ for green walnut image segmentation, which incorporates the channel and spatial attention mechanism CBAM (convolutional block attention module) and cross-entropy loss function (cross-entropy loss) into the UNet3+ network structure, and introduces the five-layer CBAM in the encoder module to construct the improved UNet3+ network model. The model consists of an encoder module (down-sampling), a decoder module (up-sampling) and a full-scale skip connection module, a full-scale feature supervision module, and a classification guidance module. After utilizing data-enhanced approaches to expand the green walnut dataset, the improved UNet3+ model was trained. The experimental findings demonstrate that the improved UNet3+ network model achieves 91.82% average precision, 96.00% recall rate, and 93.70% F1 score in the green walnut segmentation task; the addition of five-layer CBAM boosts the model segmentation precision rate by 3.11 percentage points. The method can precisely and successfully segment green walnuts, which can serve as a guide and research foundation for precisely identifying and localizing green walnuts and finishing the autonomous sorting for intelligent robots.

Keywords: image segmentation; green walnut; UNet3+; CBAM (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: 2024
References: View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2077-0472/14/1/149/pdf (application/pdf)
https://www.mdpi.com/2077-0472/14/1/149/ (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:14:y:2024:i:1:p:149-:d:1322482

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