基于目标检测的户型图识别方法、装置、计算机设备

基本信息

申请号 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分支,得到初始模型,将训练户型图输入初始模型中进行训练,读取各个分支的损失函数,在各个分支的损失函数均取到最小值时,根据初始模型当前的模型参数确定检测模型,将待识别户型图输入检测模型,输出检测结果,识别检测结果中的矩形标注框,合并表征相同特征的矩形标注框,依据最终的矩形标注框确定待识别户型图的各部分特征,以准确确定待识别户型图的各部分特征,提高户型图的识别效率。