🧠 基于 CNN 的垃圾识别分类系统

本项目通过卷积神经网络（CNN）实现生活垃圾的自动识别与分类，旨在提升垃圾分类的智能化与自动化水平。
支持图片上传识别、模型训练、验证与可视化展示。

---

📁 项目结构

---

🚀 项目功能

• ✅ 支持 4 类垃圾自动识别（可回收、有害、厨余、其他）
• 🧠 使用 ResNet18 / MobileNetV2 模型结构
• 🧩 支持 迁移学习、数据增强、轻量化优化
• 📊 自动输出训练曲线与准确率
• 🌐 提供 Web 界面展示（基于 Streamlit）

---

🧱 环境依赖

推荐 Python 版本：3.9 - 3.11
使用 PyTorch 进行模型构建与训练。

安装依赖：

pip install -r requirements.txt
data/
├── train/
│   ├── recyclable/
│   ├── hazardous/
│   ├── kitchen/
│   └── other/
└── val/
    ├── recyclable/
    ├── hazardous/
    ├── kitchen/
    └── other/

先生成模型:
**python src\train.py --train_dir "data\train" --val_dir "data\val"**
在执行推理脚本:
python src\infer.py