Deep Learning Garbage Classification

基于深度学习的垃圾图像分类系统，包含模型训练、Flask 推理服务与 Spring Boot Web 后端。

功能概览

• 使用 ResNet50 进行 12 类垃圾图片分类训练。
• 提供 Flask REST API，用于图片上传与模型推理。
• 提供 Spring Boot Web 应用，用于分类记录管理与前端交互。
• 提供 MySQL、SQL Server 与通用 SQL 初始化脚本。

项目结构

database/          数据库初始化脚本
garbage-web/       Spring Boot Web 后端
model-service/     Flask 模型推理服务
model-training/    PyTorch 模型训练代码

说明

仓库仅包含源码与必要配置，不包含训练数据集、模型权重、运行日志、构建产物、JDK 或本地虚拟环境。需要运行时，请自行准备数据集与模型权重。