如何使用TrashNet数据集?2527张垃圾图像的完整获取与预处理指南 📅 2026/8/8 19:36:24 如何使用TrashNet数据集2527张垃圾图像的完整获取与预处理指南【免费下载链接】trashnetDataset of images of trash; Torch-based CNN for garbage image classification项目地址: https://gitcode.com/gh_mirrors/tr/trashnetTrashNet是一个包含2527张垃圾图像的开源数据集支持玻璃、纸张、 cardboard、塑料、金属和 trash 六大类别的垃圾分类研究。本文将详细介绍如何获取这个高质量数据集并完成专业预处理帮助你快速开展垃圾图像分类项目。 TrashNet数据集核心信息TrashNet数据集由斯坦福大学CS 229课程项目开发包含六大类垃圾图像具体分布如下玻璃glass501张纸张paper594张硬纸板cardboard403张塑料plastic482张金属metal410张其他垃圾trash137张所有图像均在统一的白色背景下拍摄使用Apple iPhone 7 Plus、5S和SE等设备采集原始数据集大小约3.5GB。图像已标准化为512×384像素可通过data/constants.py调整尺寸参数非常适合用于训练卷积神经网络(CNN)进行垃圾分类研究。 数据集获取方法1. 克隆项目仓库首先通过Git克隆TrashNet项目仓库到本地git clone https://gitcode.com/gh_mirrors/tr/trashnet cd trashnet2. 解压预处理图像项目已提供预处理后的图像压缩包直接解压即可使用cd data unzip dataset-resized.zip3. 获取原始数据集可选如需使用原始高分辨率图像3.5GB可从Hugging Face Datasets下载dataset-original.zip并将解压后的文件夹放置在data目录下。原始图像需经过预处理才能用于模型训练。️ 图像预处理完整流程预处理脚本解析TrashNet提供了专业的图像预处理工具data/resize.py主要功能包括统一图像尺寸默认512×384自动旋转调整图像方向批量处理多类别图像自定义预处理步骤1. 修改尺寸参数如需调整输出图像尺寸编辑data/constants.py文件# 原始参数 DIM1 384 # 高度 DIM2 512 # 宽度2. 执行预处理删除现有预处理结果并重新运行预处理脚本cd data rm -rf dataset-resized # 注意这会删除现有预处理图像 python resize.py⏱️ 预处理过程约需30分钟脚本会自动处理dataset-original目录下的所有图像并将结果保存到dataset-resized文件夹。3. 预处理原理data/resize.py的核心处理流程遍历dataset-original下的各类别文件夹筛选出.jpg格式图像根据最长边自动旋转调整图像方向按指定尺寸缩放图像保存到dataset-resized对应类别文件夹 数据集文件结构预处理完成后数据集目录结构如下data/ ├── dataset-resized/ # 预处理后的图像 │ ├── glass/ # 玻璃类图像 │ ├── paper/ # 纸张类图像 │ ├── cardboard/ # 硬纸板类图像 │ ├── plastic/ # 塑料类图像 │ ├── metal/ # 金属类图像 │ └── trash/ # 其他垃圾类图像 ├── constants.py # 预处理参数配置 └── resize.py # 预处理脚本 数据集应用场景TrashNet数据集可广泛应用于垃圾分类AI模型训练计算机视觉算法研究环保科技产品开发学术研究与教育项目使用该数据集时请引用项目仓库。如需开展垃圾图像分类模型训练可结合项目提供的Torch-based CNN代码model.lua进行实验。 总结TrashNet提供了一个高质量、多类别的垃圾图像数据集通过本文介绍的方法你可以轻松获取2527张标准化垃圾图像并完成预处理。无论是学术研究还是商业应用这个数据集都能为你的垃圾分类项目提供坚实的数据基础。立即开始使用TrashNet数据集探索AI在环保领域的无限可能吧【免费下载链接】trashnetDataset of images of trash; Torch-based CNN for garbage image classification项目地址: https://gitcode.com/gh_mirrors/tr/trashnet创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考