基于目标检测的户型图识别方法、装置、计算机设备
基本信息
申请号 | CN202010031494.5 | 申请日 | - |
公开(公告)号 | CN111310755B | 公开(公告)日 | 2022-07-08 |
申请公布号 | CN111310755B | 申请公布日 | 2022-07-08 |
分类号 | G06V10/25(2022.01)I;G06V10/774(2022.01)I;G06V10/82(2022.01)I;G06K9/62(2022.01)I;G06N3/04(2006.01)I | 分类 | 计算;推算;计数; |
发明人 | 陈旋;吕成云;贺叶羊 | 申请(专利权)人 | 江苏艾佳家居用品有限公司 |
代理机构 | 南京新慧恒诚知识产权代理有限公司 | 代理人 | - |
地址 | 211100 江苏省南京市江宁区江南路9号招商高铁广场项目A座5层(南站片区) | ||
法律状态 | - |
摘要
摘要 | 本发明公开了一种基于目标检测的户型图识别方法、装置、计算机设备和存储介质,通过获取携带标签数据的训练户型图,在YOLOV3模型的DarkNet53网络后面加入Mask分支,并Head网络中加入Centerness分支,得到初始模型,将训练户型图输入初始模型中进行训练,读取各个分支的损失函数,在各个分支的损失函数均取到最小值时,根据初始模型当前的模型参数确定检测模型,将待识别户型图输入检测模型,输出检测结果,识别检测结果中的矩形标注框,合并表征相同特征的矩形标注框,依据最终的矩形标注框确定待识别户型图的各部分特征,以准确确定待识别户型图的各部分特征,提高户型图的识别效率。 |
