垃圾分类 Garbage_Classification(G_C)

1. 函数说明

main.py
模型训练主程序
args.py
- 保存模型参数
data_preprocess.py
- 数据预处理相关
forecast.py
- 模型预测使用，Web接口方式
- 结构格式：
  - 请求地址：{Host}:{Point}/forecast
  - 请求方式：Post
  - Body：
    - KEY: image
    - VALUE: 图像文件
models（Folder）
- 存放各模型
utils（Folder）
- 存放其他相关功能函数

2.测试版本

类型	版本
Python	3.7
PyTorch	1.5
CUDA	10.2

3.模型下载

已训练好的模型保存在Google Drive中，点击ResNeXt_32x16d_WSL_Attention.pth下载。