[发明专利]一种垃圾分类方法及智能垃圾桶

权利要求书

1.一种垃圾分类方法，其特征在于，包括：

获取基础数据集，并对所述基础数据集进行预处理得到预处理数据集；

选取预训练模型进行迁移学习，将所述预处理数据集和垃圾分类标签输入至所述预训练模型中进行训练，构建垃圾训练模型；

采集待分类垃圾的图像数据，并将所述图像数据输入至所述垃圾训练模型中进行特征提取，得到所述待分类垃圾对应的垃圾特征；

将所述垃圾特征与所述垃圾训练模型中的特征进行匹配，根据匹配结果得到所述待分类垃圾的分类标签，将所述分类标签作为所述待分类垃圾的分类结果。

2.如权利要求1所述的垃圾分类方法，其特征在于，所述获取基础数据集，并对所述基础数据集进行预处理得到预处理数据集，具体为：

获取基础数据集，采用数据增广方法对所述基础数据集进行预处理得到预处理数据集。

3.如权利要求2所述的垃圾分类方法，其特征在于，采用数据增广方法对所述基础数据集进行预处理得到预处理数据集，包括：

对所述基础数据集进行水平翻转、垂直翻转、高斯噪声和高斯模糊操作扩充所述基础数据集；以及通过引入外部数据扩充所述基础数据集。

4.如权利要求1所述的垃圾分类方法，其特征在于，所述选取预训练模型进行迁移学习，将所述预处理数据集和垃圾分类标签输入至所述预训练模型中进行训练，构建垃圾训练模型，具体为：

选取卷积神经网络作为预训练模型，将CBAM注意力机制模块添加至所述预训练模型的首层卷积层，在所述预训练模型的全连接层添加Dropout层，并采用交叉熵损失函数作为损失函数，以及采用SGD作为优化函数；

将所述预处理数据集和垃圾分类标签输入至所述预训练模型中进行微调训练，构建垃圾训练模型。

5.如权利要求1所述的垃圾分类方法，其特征在于，还包括：在获得所述待分类垃圾的分类结果后，通过语音播放所述分类结果，在预设时间内没有检测到用户手动输入的分类指令时，控制所述分类结果对应的垃圾收集装置的阀门打开。

6.如权利要求1所述的垃圾分类方法，其特征在于，所述分类标签包括可回收垃圾、不可回收垃圾和其他垃圾。

7.一种智能垃圾桶，其特征在于，包括外壳、盖板、分类识别腔和若干个垃圾收集装置，所述盖板设置于所述外壳的顶部，所述盖板下方设置有用于对待分类垃圾进行分类识别的分类识别腔，所述分类识别腔的底部设置有通过传输管道连通至若干个所述垃圾收集装置的内门；

所述分类识别腔设置有图像采集模块和垃圾分类装置，所述图像采集模块用于采集所述分类识别腔的待分类垃圾的图像数据，所述垃圾分类装置用于执行如权利要求1-6中任意一项所述的垃圾分类方法。

8.如权利要求7所述的智能垃圾桶，其特征在于，还包括设置于所述盖板上的压力传感器，所述压力传感器用于检测所述盖板是否存在超过阈值大小的压力值。

9.如权利要求8所述的智能垃圾桶，其特征在于，所述垃圾分类装置还用于，在所述压力传感器检测到的压力值超过阈值大小时，控制所述盖板打开。

10.如权利要求7所述的智能垃圾桶，还包括设置在每一所述垃圾收集装置上的红外探测器，所述红外探测器用于通过红外光探测每一所述垃圾收集装置的垃圾容量是否达到预设值。