首页|期刊导航|农业装备与车辆工程|基于改进ShuffleNet V2的板栗智能分级方法研究

基于改进ShuffleNet V2的板栗智能分级方法研究OA

Research on intelligent grading method of chestnuts based on improved ShuffleNet V2

中文摘要英文摘要

为实现板栗外观缺陷的精准检测,改进现有ShuffleNet V2 网络,通过引入卷积块注意力机制(Convolutional Block Attention Module,CBAM),增强模型对霉斑、虫眼等细微缺陷特征的提取能力;同时采用Focal Loss损失函数,缓解因正负样本不均衡所导致的训练偏倚问题.实验结果表明,CBAM模块有效提升了模型对缺陷区域的识别敏感度,而Focal Loss加强了对难例样本的关注,使缺陷级板栗的识别召回率从 85.6%提升至 91.8%.所提出模型在保持高分类精度的同时具备良好的轻量化特性,其参数量仅为 3.6 M,单张图像推理时间为 3.2 ms.

To achieve accurate detection of appearance defects in Chinese chestnuts,the existing ShuffleNet V2 network was improved.By introducing the Convolutional Block Attention Module(CBAM),the model's ability to extract subtle defect features such as mold spots and insect holes was enhanced.Meanwhile,the Focal Loss function was adopted to alleviate the training bias caused by the imbalance between positive and negative samples.Experimental results showed that the CBAM module effectively improves the model's sensitivity to identifying defect regions,and Focal Loss strengthens the focus on hard-to-classify samples,increasing the recall rate of defect-level Chinese chestnut recognition from 85.6%to 91.8%.The proposed model maintains high classification accuracy while possessing excellent lightweight characteristics,with a parameter count of only 3.6 M and an inference time of 3.2 ms per image.

陈倩娥;焦洪磊

河北科技师范学院数学与信息科技学院,河北 秦皇岛 066004河北科技师范学院数学与信息科技学院,河北 秦皇岛 066004

轻工纺织

板栗等级划分缺陷检测ShuffleNet V2注意力模块

classification of chestnuts gradesdefect detectionShuffleNet V2attention module

《农业装备与车辆工程》 2026 (1)

36-42,48,8

10.3969/j.issn.1673-3142.2026.01.005

评论