# tensorflow2.0-神经网络实现垃圾分类

**Repository Path**: K1903/classify_grabage

## Basic Information

- **Project Name**: tensorflow2.0-神经网络实现垃圾分类
- **Description**: 小白入门tensorflow写的一个简单程序，利用自建神经网络和vgg16两种方式实现垃圾分类。
- **Primary Language**: Python
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 28
- **Forks**: 3
- **Created**: 2020-06-20
- **Last Updated**: 2024-03-19

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# tensorflow2.0-神经网络实现垃圾分类

#### 介绍
小白入门tensorflow写的一个简单程序，利用自建神经网络和vgg16两种方式实现垃圾分类。
总共四种分类结果，训练集中每一种类别对应1000个样本，测试集中每一种类别对应100个样本。
#### 软件架构
classify_garbage.py为自建神经网络，trans_classify_garbage.py为vgg16迁移学习，test_garbage.py为模型测试
model.zip存放训练好的模型文件，data.zip存放数据集和测试集（需在百度云中下载）
#### 使用说明

1.  在文本文档中访问百度云链接，下载两个压缩包
2.  解压后将文件夹内的文件拷贝至工程文件夹
3.  运行代码