UML图片的识别方法及系统

基本信息

申请号 CN202110364911.2 申请日 -
公开(公告)号 CN112801058A 公开(公告)日 2021-06-29
申请公布号 CN112801058A 申请公布日 2021-06-29
分类号 G06K9/00;G06K9/62;G06N3/04;G06N3/08 分类 计算;推算;计数;
发明人 杨晨;滕峰 申请(专利权)人 艾伯资讯(深圳)有限公司
代理机构 深圳市壹壹壹知识产权代理事务所(普通合伙) 代理人 师勇
地址 518000 广东省深圳市南山区高新区北区朗山路13号清华紫光科技园4层C401-C405
法律状态 -

摘要

摘要 本发明实施例公开了一种UML图片的识别方法及系统,所述方法包括:步骤1:收集UML图片数据,形成数据集,对数据集划分,并进行数据预处理;步骤2:构建模型;步骤3:训练模型;步骤4:调整模型;步骤5:下载互联网上的图片;将图片转换成模型的合法输入;输入模型;使用模型识别图片;将识别出的UML图片存入UML图库。本发明在卷积模块的基础上,通过增加通道注意力模块和空间注意力模块,并在注意力模块后采用全局平均池化层进行连接,提升了UML图像识别的效率和准确率,实现了能够从互联网海量图片中有效识别UML图片的技术效果。