Research on the Classification Method of Tea Tree Seeds Quality Based on Mid-Infrared Spectroscopy and Improved DenseNet

Precise quality screening of tea tree seeds is crucial for the development of the tea industry. This study proposes a high-precision quality classification method for tea tree seeds by integrating mid-infrared (MIR) spectroscopy with an improved deep learning model. Four types of tea tree seeds in d...

Full description

Saved in:
Bibliographic Details
Main Authors: Di Deng, Hao Li, Jiawei Luo, Jiachen Jiang, Hongbo Mu
Format: Article
Language:English
Published: MDPI AG 2025-06-01
Series:Applied Sciences
Subjects:
Online Access:https://www.mdpi.com/2076-3417/15/13/7336
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Precise quality screening of tea tree seeds is crucial for the development of the tea industry. This study proposes a high-precision quality classification method for tea tree seeds by integrating mid-infrared (MIR) spectroscopy with an improved deep learning model. Four types of tea tree seeds in different states were prepared, and their spectral data were collected and preprocessed using Savitzky–Golay (SG) filtering and wavelet transform. Aiming at the deficiencies of DenseNet121 in one-dimensional spectral processing, such as insufficient generalization ability and weak feature extraction, the ECA-DenseNet model was proposed. Based on DenseNet121, the Batch Channel Normalization (BCN) module was introduced to reduce the dimensionality via 1 × 1 convolution while preserving the feature extraction capabilities, the Attention–Convolution Mix (ACMix) module was integrated to combine convolution and self-attention, and the Efficient Channel Attention (ECA) mechanism was utilized to enhance the feature discriminability. Experiments show that ECA-DenseNet achieves 99% accuracy, recall, and F1-score for classifying the four seed quality types, outperforming the original DenseNet121, machine learning models, and deep learning models. This study provides an efficient solution for tea tree seeds detection and screening, and its modular design can serve as a reference for the spectral classification of other crops.
ISSN:2076-3417