生活垃圾目标检测数据集:5类别、6,000张图像 | 目标检测
源码数据分享
通过网盘分享的文件:生活垃圾目标检测数据集
链接: https://pan.baidu.com/s/1iSlUKBEFQ2yO8NllReoOOA?pwd=5h95
提取码: 5h95
文章继续

一、城市生活垃圾治理的挑战与机遇
1.1 城市垃圾问题的全球视角
城市生活垃圾管理是全球城市治理面临的共同挑战。世界银行发布的《What a Waste 2.0》报告指出,全球每年产生约20.1亿吨城市固体废弃物,预计到2050年将增长70%达到34.0亿吨。其中,东亚和太平洋地区是最大的废弃物产生区域,中国作为该地区最大的经济体,城市生活垃圾年产生量超过2亿吨。
垃圾管理不善带来的问题远不止环境卫生:
- 资源浪费:大量可回收材料被填埋或焚烧,无法进入循环经济体系
- 环境污染:垃圾渗滤液污染地下水,焚烧产生有害气体
- 温室效应:有机垃圾分解产生的甲烷是强效温室气体
- 土地消耗:填埋场占用大量土地资源
- 社会成本:环卫工人劳动强度大,健康风险高
1.2 中国垃圾分类推进历程
中国垃圾分类经历了从倡导到强制的渐进过程:
- 2000年:8个试点城市启动垃圾分类收集
- 2017年:国务院发布《生活垃圾分类制度实施方案》
- 2019年:上海率先实施强制垃圾分类
- 2020年:46个重点城市基本建成垃圾分类处理系统
- 2025年:全国地级及以上城市基本建成垃圾分类处理系统
尽管政策推动力度空前,但实际执行层面仍面临巨大挑战:
- 居民分类意识参差不齐:分类准确率波动大
- 前端分类后端混装:打击居民分类积极性
- 人工分拣成本高:末端精细分拣依赖大量人工
- 监管难度大:违规投放行为难以及时发现
1.3 智能视觉技术重塑垃圾治理
AI视觉检测技术为垃圾分类提供了全新的技术路径:
前端智能分类:
- 智能垃圾桶:实时识别投放物类别,语音引导正确投放
- 智能投放站:自动识别并分拣,减少人工干预
中端智能监测:
- 垃圾满溢检测:实时监测垃圾桶/站满溢状态,优化清运路线
- 违规投放检测:识别未按规定分类的投放行为
末端智能分拣:
- 自动分拣流水线:基于视觉识别的机器人分拣系统
- 质量检测:检测可回收物的纯度和质量
而构建高性能垃圾分类模型的核心前提,是拥有一个高质量、大规模、场景真实的垃圾检测数据集。

二、数据集深度解析
2.1 核心规格参数
| 参数项 | 具体数值/描述 |
|---|---|
| 图像总量 | 6000张 |
| 类别数量 | 5类(玻璃、金属、纸张、塑料、其他垃圾) |
| 标注方式 | YOLO格式边界框标注 |
| 数据来源 | 真实生活环境采集 |
| 数据划分 | train / valid / test |
| 适配模型 | YOLOv5/v8/v11等主流目标检测模型 |
2.2 五类生活垃圾详解
| 类别ID | 类别名称 | 典型物体 | 视觉特征 | 识别难点 |
|---|---|---|---|---|
| 0 | 玻璃 | 玻璃瓶、玻璃碎片 | 透明/半透明,反光 | 透明边界模糊,碎片形态不规则 |
| 1 | 金属 | 易拉罐、金属罐头 | 金属光泽,硬质 | 压扁后变形大,氧化后颜色变化 |
| 2 | 纸张 | 纸盒、报纸、纸板 | 白色/棕色,柔软 | 揉皱后纹理复杂,与纸板区分 |
| 3 | 塑料 | 塑料瓶、塑料袋 | 透明/彩色,轻质 | 塑料袋折叠后形态多变 |
| 4 | 其他垃圾 | 食物残渣、陶瓷碎片 | 形态多样,颜色混杂 | 类内差异大,边界定义模糊 |
2.3 "其他垃圾"类别的设计考量
数据集中特别设置了"其他垃圾"类别,这一设计在工程实践中具有重要意义:
避免无法分类的尴尬:在实际场景中,总有一些垃圾难以归入特定类别(如污损的纸张、混合材料制品等),如果没有"其他垃圾"类别,模型会被迫将其归入某一类别,导致误分类。
提升整体分类可靠性:有了"其他垃圾"作为"兜底"类别,前4类的分类精度可以得到有效保障,因为模糊样本不会被错误地归入它们。
符合实际业务需求:在垃圾分类业务流程中,"其他垃圾"是一个真实存在的处理类别,需要与可回收物、有害垃圾、厨余垃圾并列处理。
2.4 数据集场景覆盖
本数据集的图像来源于真实生活环境,覆盖了城市垃圾管理的典型场景:
家庭场景:
- 厨房垃圾桶:厨余垃圾与可回收物混合
- 客厅废纸篓:纸张类垃圾为主
- 阳台回收区:分类后的可回收物暂存
小区垃圾投放点:
- 分类垃圾桶:四分类垃圾桶的内部视角
- 垃圾投放口:居民投放时的抓拍视角
- 桶外堆放:投放不规范的情况
街头垃圾收集区域:
- 路边果皮箱:行人投放的各类垃圾
- 街道清扫物:散落的各类垃圾
- 临时堆放点:大件垃圾和建筑垃圾
2.5 数据多样性保障
为确保模型具备良好的泛化能力,数据集在以下维度实现了多样性:
光照条件多样性:
- 室内人工光:白炽灯、荧光灯、LED灯
- 户外自然光:晴天、阴天、晨昏
- 弱光环境:夜间、地下车库
视角多样性:
- 俯视拍摄:垃圾桶内部视角
- 平视拍摄:行人视角
- 近距离拍摄:手机拍照距离
垃圾状态多样性:
- 完整物品:未使用的包装
- 使用后物品:饮用后的瓶子、食用后的容器
- 压缩变形:踩扁的易拉罐、折叠的纸盒
- 破碎物品:碎玻璃、撕碎的纸张
堆叠状态多样性:
- 单独放置:无遮挡的清晰目标
- 部分遮挡:被其他垃圾部分覆盖
- 深度堆叠:大量垃圾堆叠在一起

三、模型训练与性能优化
3.1 数据集配置
# domestic_waste.yaml
path: database/生活垃圾目标检测数据集(垃圾分类检测数据集)
train: train/images
valid: valid/images
test: test/images
nc: 5
names:
0: 玻璃
1: 金属
2: 纸张
3: 塑料
4: 其他垃圾
3.2 模型训练
from ultralytics import YOLO
model = YOLO('yolov8s.pt')
results = model.train(
data='domestic_waste.yaml',
epochs=200,
imgsz=640,
batch=32,
patience=40,
lr0=0.01,
mosaic=1.0,
mixup=0.1,
copy_paste=0.05,
degrees=15,
translate=0.1,
scale=0.9,
fliplr=0.5,
hsv_h=0.015,
hsv_s=0.7,
hsv_v=0.4,
project='domestic_waste',
name='yolov8s_waste'
)
3.3 针对垃圾检测的关键优化
遮挡与堆叠处理:
垃圾场景中遮挡严重是常态,需要专门优化:
# Soft-NMS配置
results = model.val(
data='domestic_waste.yaml',
iou=0.65, # 适当提高IoU阈值
conf=0.3, # 降低置信度阈值提升召回
max_det=50 # 增加最大检测数
)
透明物体检测:
玻璃和透明塑料是检测难点:
- 在训练数据中增加透明物体的样本比例
- 使用边缘增强预处理突出透明物体边界
- 引入深度信息辅助判断(如使用RGB-D相机)
- 增加反光场景的数据增强
变形物体适应:
垃圾的变形程度远超一般目标检测场景:
- 大幅度的仿射变换增强
- 弹性变形增强(Elastic Deformation)
- 可变形卷积网络(DCN)
- 形状上下文特征增强
3.4 类别不均衡处理
"其他垃圾"的类内差异大,且样本可能偏少,需要特别处理:
# 类别权重配置
class_weights = {
0: 1.0, # 玻璃
1: 1.0, # 金属
2: 1.0, # 纸张
3: 1.0, # 塑料
4: 1.5, # 其他垃圾 - 增加权重
}
3.5 后处理优化
class WasteDetectionPostProcessor:
def __init__(self, conf_threshold=0.5, iou_threshold=0.45):
self.conf_threshold = conf_threshold
self.iou_threshold = iou_threshold
def process(self, detections):
"""后处理优化检测结果"""
# 1. 置信度过滤
filtered = [d for d in detections if d['confidence'] >= self.conf_threshold]
# 2. 基于面积的合理性检查
valid = []
for det in filtered:
area = self._calc_area(det['bbox'])
if area > 100: # 过滤过小的检测框
valid.append(det)
# 3. 时序一致性检查(视频场景)
# ...
return valid
def _calc_area(self, bbox):
return (bbox[2] - bbox[0]) * (bbox[3] - bbox[1])
四、智能垃圾分类系统设计
4.1 智能垃圾桶方案
智能垃圾桶是最贴近居民的垃圾分类终端:
硬件配置:
- 高清摄像头:安装在投放口上方
- 计算模块:RK3588或Jetson Nano
- 语音播报模块:分类结果语音提示
- 显示模块:LCD屏显示分类引导
- 分拣执行机构:电动翻板或旋转桶
工作流程:
- 用户靠近,传感器触发
- 摄像头拍摄投放物品
- AI模型实时识别垃圾类别
- 语音播报识别结果
- 分拣机构引导垃圾进入对应仓室
- 记录投放数据
用户体验设计:
- 识别速度<1秒,避免用户等待
- 语音提示友好,避免命令式语气
- 分类错误时温和提醒,不拒绝投放
- 提供分类知识科普功能
4.2 垃圾分拣流水线方案
在垃圾分拣中心,视觉检测系统用于自动分拣:
系统架构:
- 传送带:匀速输送垃圾
- 多角度摄像头阵列:全方位拍摄
- GPU/NPU推理服务器:实时检测
- 机器人/气阀执行机构:分拣动作
- 质检摄像头:分拣质量检测
关键技术指标:
- 传送带速度:1-3米/秒
- 检测帧率:>30FPS
- 分拣精度:>95%
- 处理能力:>1吨/小时
4.3 环卫监测系统方案
基于视觉的环卫监测系统用于城市级垃圾管理:
功能模块:
- 垃圾桶满溢检测:识别垃圾桶是否已满
- 违规投放检测:识别未按规定分类的投放
- 垃圾散落检测:识别路面散落的垃圾
- 清运路线优化:基于满溢数据优化清运路线
部署方案:
- 垃圾桶安装低功耗摄像头+NB-IoT通信
- 城管巡逻车安装移动检测设备
- 无人机定期巡检重点区域
4.4 系统集成与数据平台
构建城市级垃圾分类数据平台:
- 数据采集层:各类检测设备的数据接入
- 数据处理层:实时数据清洗、聚合和分析
- 业务应用层:分类统计、趋势分析、预警管理
- 可视化层:数据看板、热力图、报表

五、业务价值与经济效益
5.1 分类效率提升
| 指标 | 传统人工 | AI辅助 | 提升幅度 |
|---|---|---|---|
| 分类准确率 | 70-80% | 90-95% | +15-20% |
| 分类速度 | 5-10件/分钟 | 30-60件/分钟 | 5-10倍 |
| 人工成本 | 高 | 低 | -60%以上 |
| 工作时间 | 8小时/天 | 24小时/天 | 3倍 |
5.2 回收价值提升
更精准的分类意味着更高的回收品质和更大的回收价值:
- 可回收物纯度提升 → 回收价格提高
- 混入率降低 → 减少二次分拣成本
- 回收率提升 → 更多资源进入循环经济
- 末端处理成本降低 → 填埋/焚烧量减少
5.3 社会与环境效益
- 减少垃圾填埋量,延长填埋场使用寿命
- 减少焚烧排放,改善空气质量
- 提升居民环保意识和参与度
- 创造AI+环保领域就业机会
- 推动循环经济和绿色城市发展
六、技术前沿探索
6.1 多模态融合分类
单一RGB图像在垃圾识别中存在固有局限(如透明物体、遮挡场景)。融合多种传感器信息可以显著提升识别能力:
- RGB-D:深度信息辅助判断物体厚度和形状
- 近红外(NIR):区分材质(塑料vs纸张)
- 高光谱:精确识别材料成分
- 重量传感器:辅助判断空瓶vs满瓶
6.2 端云协同推理
将轻量级模型部署在边缘设备进行快速初步判断,对不确定的样本上传云端由大模型二次分析,兼顾实时性和精度。
6.3 持续学习与自适应
垃圾分类场景中的垃圾类型和形态会随季节、节日、地区等因素变化,模型需要具备持续学习和自适应能力:
- 在线学习:根据新的误检案例持续更新模型
- 域自适应:跨场景迁移(如从社区到商业区)
- 少样本扩展:快速适应新的垃圾类型
6.4 生成式AI辅助
利用扩散模型生成合成训练数据,特别是针对罕见类别和极端场景的数据补充:
- 生成特定类型的垃圾图像
- 模拟各种光照和天气条件
- 合成遮挡和堆叠场景
- 数据增强的多样化
七、项目落地注意事项
7.1 数据合规性
- 确保图像数据采集符合隐私保护法规
- 避免采集包含个人信息的图像
- 数据使用需获得相应授权
7.2 环境适应性
- 室外设备需考虑防水防尘(IP65以上)
- 高温/低温环境的设备稳定性
- 强光/逆光环境的光学补偿
- 夜间场景的补光设计
7.3 用户接受度
- 语音提示音量和语气需合理设置
- 分类结果的展示需直观易懂
- 错误分类的纠正需温和友好
- 系统故障时需有降级方案
7.4 运维可持续性
- 模型定期更新迭代
- 硬件定期巡检维护
- 数据质量持续监控
- 用户反馈及时响应
八、总结
本5类生活垃圾检测数据集以6000张高质量标注图像,覆盖玻璃、金属、纸张、塑料、其他垃圾5类核心生活垃圾,为城市垃圾分类AI系统提供了坚实的数据基础。数据集来源于真实生活场景,类别设计贴合垃圾分类业务需求,数据结构标准化程度高,可直接用于YOLO系列模型训练与部署。
从更宏观的视角来看,AI技术正在深刻改变城市垃圾治理的方式。从"人海战术"到"智能识别",从"事后处理"到"源头分类",AI赋能的垃圾分类系统将推动城市垃圾治理从粗放式管理向精细化运营转型,为绿色城市建设和循环经济发展提供有力支撑。