23 lines (16 loc) · 853 Bytes

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 或本地虚拟环境。需要运行时，请自行准备数据集与模型权重。