首页|基于多尺度特征提取与融合的垃圾分类模型研究

基于多尺度特征提取与融合的垃圾分类模型研究

扫码查看
以YOLOv4 算法为基础,了解YOLOv4 算法的基本结构及其在垃圾分类实际应用场景中存在的问题,提出了YOLOv4 算法改进策略,通过在CBM模块中设定两个卷积核来实现不同尺寸垃圾的特征提取.对改进的多尺度特征提取与融合的垃圾分类模型进行训练与验证,发现改进的YOLOv4 算法可以很好地满足多尺度特征提取与融合的基本需求,对小尺寸垃圾目标的分辨效率可达 85%以上,处理时间也能很好地满足实际应用场景的基本需求,并且卷积核值设定越小,垃圾目标识别精确度越高,处理时间也会相应增加,可以很好地促进垃圾分类产业向智能化转变.
Research on garbage classification model based on multi-scale feature extraction and fusion
Taking the YOLOv4 algorithm as the foundation,the paper proposes improvement strategies for the YOLOv4 algorithm by understanding its basic structure and addressing the issues encountered in practical applications of garbage classification.The improvement is achieved by setting two convolutional kernels in the CBM module to facilitate the feature extraction process for garbage of different sizes.Through training and validation of the improved garbage classification model with multi-scale feature extraction and fusion,it is found that the improved YOLOv4 algorithm can effectively meet the basic requirements of multi-scale feature extraction and fusion.The recognition efficiency for small-sized garbage targets can reach 85%over,and the processing time can meet the basic requirements of practical applications.Furthermore,it is observed that as the convolutional kernel size decreases,the accuracy of garbage target identification increases while the processing time also increases accordingly.This research contributes to the advancement of the garbage classification industry towards intelligence.

YOLOv4feature extractiongarbage classificationconvolutional kernel

李浩、李可、李英杰

展开 >

河南工程学院 计算机学院,河南 郑州 451191

河南工程学院 工程训练中心,河南 郑州 451191

YOLOv4 特征提取 垃圾分类 卷积核

2024

河南工程学院学报(自然科学版)
河南工程学院

河南工程学院学报(自然科学版)

影响因子:0.26
ISSN:1674-330X
年,卷(期):2024.36(3)