This invention discloses a method for sorting fresh tea leaves based on a frequency-domain tree-structured topology network, along with a
computer device and storage medium. The method encompasses a complete process:
image acquisition, preprocessing, frequency-domain
decomposition, depth modeling, map construction, and classification. First,
image quality is improved through
color normalization and
edge enhancement. Then, frequency-domain tree-structured
decomposition using
wavelet transform and
discrete cosine transform is performed to extract multi-scale features. Subsequently, long- and short-range dependency modeling and residual
convolution modules are integrated to achieve multi-level feature representation. Finally, a tree-structured topology attention path and structure map simulating the
bud-leaf-
vein relationship are constructed to enhance semantic understanding. A tree-structure-aware classification function is used to achieve fine-grained classification of single buds, one
bud and one leaf, one
bud and two leaves, and one bud and multiple leaves. During training, cross-entropy loss, data augmentation, and regularization strategies are combined to improve model robustness. This method demonstrates high classification accuracy and stability on multiple tea image datasets, effectively improving the practicality of automatic sorting of fresh tea leaves.