# GarbageClassification

**Repository Path**: wangzhengggao/garbage-classification

## Basic Information

- **Project Name**: GarbageClassification
- **Description**: 基于CNN的智能垃圾分类
- **Primary Language**: Python
- **License**: MulanPSL-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 3
- **Forks**: 0
- **Created**: 2024-08-14
- **Last Updated**: 2025-05-19

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# GarbageClassification

#### 介绍
这是脑洞大开AI实验室课程《基于CNN的智能垃圾分类》配套学习代码

#### 软件架构
- data    目录下是本次课程用到的数据集

- test    目录下是测试用的数据

- Net.py    是CNN模型架构和训练文件

- model.pth    是保存的训练好的模型

- garbage_classification.py    是模型加载和预测文件

#### 使用说明

- 运行Net.py会训练模型并将训练好的模型保存在model.pth

- 替换garbage_classification.py中测试图片的路径然后运行文件可以加载训练好的模型然后打印出预测结果