一种基于图卷积的目标检测后处理方法及装置
实质审查的生效
摘要
本发明公开了一种基于图卷积的目标检测后处理方法及装置,该方法具体包括如下步骤:步骤1,训练阶段:训练得到图卷积神经网络模型;(1)筛选出最佳匹配的预测框;(2)使用图卷积预测最佳匹配的矩形框集合;步骤2,预测阶段:对于每张检测图片,将目标检测模型预测出的且经过类别得分阈值过滤后的矩形框集合记为B,将B构造为图,使用训练好的图卷积神经网络模型,如果其预测类别为1的得分大于预先设定的阈值,则保留该结点对应的矩形框,所有保留的矩形框集合即是最终的输出结果。通过使用图卷积操作替代操作,不需要预先设置的阈值,利用了矩形框自身的特征信息,以及矩形框的上下文信息,可以很大程度提升目标检测模型的后处理性能。
基本信息
专利标题 :
一种基于图卷积的目标检测后处理方法及装置
专利标题(英):
暂无
公开(公告)号 :
CN114359627A
申请号 :
CN202111536248.6
公开(公告)日 :
2022-04-15
申请日 :
2021-12-15
授权号 :
暂无
授权日 :
暂无
发明人 :
李军
申请人 :
南京视察者智能科技有限公司
申请人地址 :
江苏省南京市秦淮区永智路6号南京白下高新技术产业园区四号楼A栋102室
代理机构 :
代理人 :
优先权 :
CN202111536248.6
主分类号 :
G06V10/764
IPC分类号 :
G06V10/764 G06V10/82 G06K9/62 G06N3/04 G06N3/08
法律状态
2022-05-03 :
实质审查的生效
IPC(主分类) : G06V 10/764
申请日 : 20211215
申请日 : 20211215
2022-04-15 :
公开
注:本法律状态信息仅供参考,即时准确的法律状态信息须到国家知识产权局办理专利登记簿副本。
文件下载
暂无PDF文件可下载