[发明专利]一种基于人工智能的垃圾分类识别方法

权利要求书

1.一种基于人工智能的垃圾分类识别方法，其特征是步骤如下：

S1建立数据集：收集垃圾照片，并对垃圾照片进行类别标注，构成数据集；

S2对数据集中的垃圾图片进行预处理；

S3由卷积深度神经网络构成垃圾分类模型：第一层由两层卷积层和一层池化层组成，第二层由两层卷积层和一层池化层组成，第三层由四层卷积层和一层池化层组成，第四层由四层卷积层和一层池化层组成，第五层由四层全连接层组成；

S4使用步骤S2中预处理后的数据集训练步骤S3中的垃圾分类模型；

S5将待分类的垃圾照片预处理后输入到训练后的垃圾分类模型，输出分类结果。

2.根据权利要求1所述的方法，其特征是S1中将数据集按照一定比例划分成训练测试集和验证集。

3.根据权利要求2所述的方法，其特征是S4中使用10折交叉验证方法将训练测试集划分为测试集和训练集，利用划分后的测试集和训练集训练垃圾分类模型，使用验证集根据代价函数计算垃圾分类模型的代价，根据代价选择最优的垃圾分类模型。

4.根据权利要求3所述的方法，其特征是具体步骤为：

S401使用十折交叉验证法对训练测试集进行划分和验证，训练测试集中的十分之一作为测试集，剩余的十分之九作为训练集；

S402使用训练集和测试集训练垃圾分类模型；

S403使用验证集根据代价函数计算垃圾分类模型的代价；

S404根据代价进行反向传播修改垃圾分类模型的参数直到得到最优的垃圾分类模型。

5.根据权利要求1-4任一所述的方法，其特征是S3中对垃圾分类模型使用batch-normal、prelu激活函数、dropout方法中一种或几种进行优化。

6.根据权利要求5所述的方法，其特征是S2中对数据集中进行图片降噪、二值化、字符切分以及归一化的预处理。

7.根据权利要求6所述的方法，其特征是S5中利用opencv读取待分类的垃圾照片预处理后输入到训练后的垃圾分类模型，输出分类结果。

8.一种基于人工智能的垃圾分类识别系统，其特征是包括数据集建立模块、图片预处理模块、垃圾分类模型建立模块、训练模块及垃圾分类模块，

数据集建立模块建立数据集：收集垃圾照片，并对垃圾照片进行类别标注，构成数据集；

图片预处理模块对数据集中的垃圾图片进行预处理；

垃圾分类模型建立模块利用卷积深度神经网络构成垃圾分类模型：第一层由两层卷积层和一层池化层组成，第二层由两层卷积层和一层池化层组成，第三层由四层卷积层和一层池化层组成，第四层由四层卷积层和一层池化层组成，第五层由四层全连接层组成；

训练模块使用图片预处理模块中预处理后的数据集训练垃圾分类模型建立模块中的垃圾分类模型；

垃圾分类模块将待分类的垃圾照片预处理后输入到训练后的垃圾分类模型，输出分类结果。