# TrashClassifier

**Repository Path**: zhangp2020/TrashClassifier

## Basic Information

- **Project Name**: TrashClassifier
- **Description**: 基于EfficientNet 网路的Keras垃圾分类
- **Primary Language**: Python
- **License**: 0BSD
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2021-08-26
- **Last Updated**: 2021-08-26

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# TrashClassifier

#### Description
基于EfficientNet 网络的Keras垃圾分类

#### Installation

安装[EfficientNet](https://github.com/qubvel/efficientnet)

```bash
pip install efficientnet==1.1.0
```

安装Keras，版本大于2.0

```bash
pip install keras
```

安装tensorflow

```bash
pip install tensorflow-gpu==1.15.0
```

安装其他一些工具

```bash
pip install tqdm matplotlib numpy opencv-python
```

#### Instructions

训练模型

```bash
python train.py
```

预测

```bash
python predict.py
```

预测效果

prediction 为预测结果，labels 为真实标签

![image-20200427151444215](assets/image-20200427151444215.png)

可视化

![image-20200427151539374](assets/image-20200427151539374.png)