项目简介
本项目基于Python语言,运用图像分类模型Inception - v3和文本分类模型TextCNN构建了一个垃圾分类识别系统。用户下载模型后,使用测试图片就能验证系统的垃圾分类功能,系统会输出识别物品对应的垃圾类别。
项目的主要特性和功能
- 图像分类:采用Inception - v3模型对输入图像进行识别。
- 垃圾分类映射:利用TextCNN模型,结合数据标注的训练数据和测试数据,将识别出的物品映射到对应垃圾分类。
- 多语言类别映射:提供英文、中文和id的imagenet类别映射表。
- 方便测试:可通过命令行快速下载和测试模型,还能对指定图片进行垃圾分类识别。
安装使用步骤
假设用户已下载本项目的源码文件,按以下步骤操作:
1. 下载模型:在命令行中运行 python classify_image.py
。
2. 测试模型:在命令行中运行 python classify_image.py --image_file ./img/2.png
。
3. 进行垃圾分类识别:在命令行中运行 python rafuse.py img/2.png
,即可获取物品对应的垃圾类别输出结果。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】