基于神经网络的垃圾分类方法及装置

基本信息

申请号 CN202110594291.1 申请日 -
公开(公告)号 CN113496254A 公开(公告)日 2021-10-12
申请公布号 CN113496254A 申请公布日 2021-10-12
分类号 G06K9/62(2006.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I 分类 计算;推算;计数;
发明人 王铎 申请(专利权)人 闻泰通讯股份有限公司
代理机构 北京开阳星知识产权代理有限公司 代理人 唐博
地址 314006浙江省嘉兴市南湖区亚中路777号(嘉兴科技城)
法律状态 -

摘要

摘要 本申请涉及一种基于神经网络的垃圾分类方法、装置、计算机设备和计算机可读存储介质。所述方法包括:获取训练样本集,所述训练样本集包括多个样本垃圾图像以及所述多个样本垃圾图像中各个样本垃圾图像对应的类别标签;加载预训练分类网络模型,所述预训练分类网络模型为基于源数据对卷积神经网络模型进行训练得到的图像分类模型;基于迁移学习,通过所述训练样本集对所述预训练分类网络模型的输出层网络结构进行训练,获取垃圾分类模型;所述输出层网络结构包括至少一个全连接层;根据所述垃圾分类模型进行垃圾分类。采用本方法构建小样本的垃圾分类模型,减少参数数量,降低计算的复杂度,从而能够实现垃圾的准确、快速分类。