Soybean Seed Classification and Identification Based on Corner Point Multi-Feature Segmentation and Improved MoblieViT

Abstract

To address the challenges of high model complexity, substantial computational resource consumption, and insufficient classification accuracy in existing soybean seed identification research, we first perform soybean seed segmentation based on polygon features, constructing a dataset comprising five categories: whole seeds, broken seeds, seeds with epidermal damage, immature seeds, and spotted seeds. The MobileViT module is then optimized by employing Depthwise Separable Convolution (DSC) in place of standard convolutions, applying Transformer Half-Dimension (THD) for dimensional reconstruction, and integrating Dynamic Channel Recalibration (DCR) to reduce model parameters and enhance inter-channel interactions. Furthermore, by incorporating the CBAM attention mechanism into the MV2 module and replacing the ReLU6 activation function with the Mish activation function, the model’s feature extraction capability and generalization performance are further improved. These enhancements culminate in a novel soybean seed detection model, MobileViT-SD (MobileViT for Soybean Detection). Experimental results demonstrate that the proposed MobileViT-SD model contains only 2.09 million parameters while achieving a classification accuracy of 98.39% and an F1 score of 98.38%, representing improvements of 2.86% and 2.88%, respectively, over the original MobileViT model. Comparative experiments further show that MobileViT-SD not only outperforms several representative lightweight models in both detection accuracy and efficiency but also surpasses a number of mainstream heavyweight models. Its highly optimized, lightweight architecture combines efficient inference performance with low resource consumption, making it well-suited for deployment in computing-constrained environments, such as edge devices.

Plant phenotyping relevance

大豆種子の状態・損傷を画像セグメンテーションと軽量分類モデルで推定する手法開発が中心であり、植物器官の形態・品質状態を取得するフェノタイピング手法に該当する。

Code and data availability

The preprint describes a self-collected soybean seed image dataset (five defect categories, ~5,600 images) and a MobileViT-SD model, but no supplied block contains any data or code availability statement, public repository link, or author URL for the dataset, images, segmentation code, or trained model. All URLs in the

No evidence-backed public reproduction asset is currently recorded.