YOLO模型的垃圾检测方法、系统、设备、产品及介质
申请号:CN202510090618
申请日期:2025-01-21
公开号:CN119540782B
公开日期:2025-04-08
类型:发明专利
摘要
本发明涉及计算机视觉技术领域,提供YOLO模型的垃圾检测方法、系统、设备、产品及介质,包括获取检测数据集,对检测数据集进行重分类,得到分类集;选取骨干网络,构建特征提取卷积模块,将特征提取卷积模块嵌入骨干网络中;选取颈部网络,构建SAC‑Net模块,将SAC‑Net模块嵌入颈部网络中,并将颈部网络与骨干网络连接;构建包括EIOU损失函数的边界框损失模块,通过边界框损失模块构建检测头,并将检测头与颈部网络连接;构建包括骨干网络、颈部网络和检测头的初始分类模型;根据分类集对初始分类模型进行配置,对初始分类模型进行训练,得到目标分类模型。本发明有效提升了分类模型分类的准确率。
技术关键词
垃圾检测方法
空间特征提取
卷积模块
特征提取模块
YOLO模型
检测头
垃圾检测系统
计算机程序产品
网络模块
计算机视觉技术
网络结构
通道
数据
处理器
图像
精度