基于Anchor-free目标检测算法的非极大值抑制方法和装置制造方法及图纸

本发明专利技术提供了一种基于Anchor‑free目标检测算法的非极大值抑制方法和装置，其中，该基于Anchor‑free目标检测算法的非极大值抑制方法包括以下步骤：采用CenterNet网络模型对目标检测图像进行识别，以获取含有目标物的热图；根据热图获取目标物的目标特征点；采用N*N的max pooling对目标特征点以及目标特征点周围的(N

【技术实现步骤摘要】
基于Anchor-free目标检测算法的非极大值抑制方法和装置
本专利技术涉及深度学习和计算机视觉
，具体涉及一种基于Anchor-free目标检测算法的非极大值抑制方法和一种基于Anchor-free目标检测算法的非极大值抑制装置。
技术介绍
相关技术中，在目标检测的过程中，采用Anchor-free目标检测算法的CenterNet网络模型对待检测图像进行目标检测时，容易出现同一个目标被多个检测框框住的情况，从而大大降低了目标检测的效率。
技术实现思路
本专利技术为解决上述技术问题，提供了一种基于Anchor-free目标检测算法的非极大值抑制方法，能够有效地避免出现同一目标物存在多个检测框的情况，从而大大提高了目标检测的效率。本专利技术采用的技术方案如下：一种基于Anchor-free目标检测算法的非极大值抑制方法，包括以下步骤：采用CenterNet网络模型对目标检测图像进行识别，以获取含有目标物的热图；根据所述热图获取所述目标物的目标特征点；采用N*N的maxpooling对所述目标特征本文档来自技高网...

【技术保护点】
1.一种基于Anchor-free目标检测算法的非极大值抑制方法，其特征在于，包括以下步骤：/n采用CenterNet网络模型对目标检测图像进行识别，以获取含有目标物的热图；/n根据所述热图获取所述目标物的目标特征点；/n采用N*N的max pooling对所述目标特征点以及所述目标特征点周围的(N

【技术特征摘要】
1.一种基于Anchor-free目标检测算法的非极大值抑制方法，其特征在于，包括以下步骤：
采用CenterNet网络模型对目标检测图像进行识别，以获取含有目标物的热图；
根据所述热图获取所述目标物的目标特征点；
采用N*N的maxpooling对所述目标特征点以及所述目标特征点周围的(N2-1)个特征点进行非极大值抑制，以获取置信度最大的特征点，其中，N为大于等于3的正整数；
根据所述置信度最大的特征点的个数确定所述目标物的中心点；
根据所述目标物的中心点生成所述目标物的检测框。


2.根据权利要求1所述的基于Anchor-free目标检测算法的非极大值抑制方法，其特征在于，所述采用CenterNet网络模型对目标检测图像进行识别，以获取含有目标物的热图，包括：
对所述目标检测图像进行预处理；
根据预处理后的目标检测图像生成所述含有目标物的热图。


3.根据权利要求1所述的基于Anchor-free目标检测算法的非极大值抑制方法，其特征在于，所述根据所述置信度最大的特征点的个数确定所述目标物的中心点，包括：
判断所述置信度最大的特征点的个数是否大于1；
如果所述置信度最大的特征点的个数大于1，则从所述置信度最大的特征点中选取出一个特征点，以作为所述目标物的中心点；
如果所述置信度最大的特征点的个数等于1，则确定所述置信度最大的特征点为所述目标物的中心点。


4.一种基...

【专利技术属性】
技术研发人员：陈海波，邱云香，
申请(专利权)人：深兰人工智能芯片研究院江苏有限公司，
类型：发明
国别省市：江苏;32