[发明专利]基于生成对抗网络的具有可伸缩性模块化图像识别方法有效
申请号: | 202010448712.5 | 申请日: | 2020-05-25 |
公开(公告)号: | CN111461264B | 公开(公告)日: | 2023-06-13 |
发明(设计)人: | 俞扬;詹德川;周志华;仲耀晖 | 申请(专利权)人: | 南京大学 |
主分类号: | G06V10/764 | 分类号: | G06V10/764;G06V10/82;G06N3/0475;G06N3/094 |
代理公司: | 南京乐羽知行专利代理事务所(普通合伙) 32326 | 代理人: | 李玉平 |
地址: | 210023 江苏*** | 国省代码: | 江苏;32 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 基于 生成 对抗 网络 具有 伸缩性 模块化 图像 识别 方法 | ||
1.一种基于生成对抗网络的具有可伸缩性模块化图像识别方法,其特征在于,训练图像识别模型,利用图像识别模型识别图像;所述训练图像识别模型包括以下步骤:
步骤1:将图像数据按类别进行整理,有N个类别就将其分为N类,每类中只包含相同类别的图像数据;
步骤2:根据类别个数N准备N个生成对抗网络模型;
步骤3:训练第i个类别的图像数据,i∈{1,2,...,N};训练生成器Gi,直到生成对抗网络训练完成;
步骤4:固定生成对抗网络中判别器Di的参数,按梯度逆方向训练生成器Gi;
步骤5:固定生成对抗网络中生成器Gi的参数,训练判别器Di,直到训练完成;
步骤6:对所有类别图像数据重复步骤3至步骤5,直到N个判别器训练完成;
步骤7:将N个判别器并联排列组合成判别器组;
所述步骤3中,生成器Gi损失函数的计算方式为最小化其中,x~Pr为从真实图像数据集取出的样本,x~Pg为从生成的图像数据集中取出的样本,D为判别器;
所述步骤4中,生成器Gi损失函数的计算方式为最大化
所述步骤5中,判别器Di损失函数的计算方式为最小化
2.根据权利要求1所述的基于生成对抗网络的具有可伸缩性模块化图像识别方法,其特征在于,利用所述图像识别模型识别图像:
将图片输入到判别器组,每个判别器将对图片输出预测结果;计算最终预测结果。
3.根据权利要求1所述的基于生成对抗网络的具有可伸缩性模块化图像识别方法,其特征在于,所述的步骤3、4和5中,训练所使用的优化器为RMSProp优化器。
4.根据权利要求2所述的基于生成对抗网络的具有可伸缩性模块化图像识别方法,其特征在于,计算最终预测结果的具体过程为:若只有一个判别器将其标注为正样本,则将其归入该判别器所代表的类别;若多个判别器将其标注为正样本或者没有判别器将其标注为正样本,则将其归入负样本。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于南京大学,未经南京大学许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202010448712.5/1.html,转载请声明来源钻瓜专利网。