基于改进的自适应Anchor生成的DK-YOLOv4模型制造技术

传统的自适应Anchor生成方法中，对于目标检测框出现宽高比极端变化的情况，会造成的参数过多、收敛慢、不能有效体现目标真实框的宽高比分布进而严重影响小目标检测精度的问题，提出了一种提出了基于改进的自适应Anchor生成的DK

【技术实现步骤摘要】
基于改进的自适应Anchor生成的DK
‑
YOLOv4模型


[0001]本专利技术目标检测
，具体涉及到一种基于改进的自适应Anchor生成的DK
‑
YOLOv4模型。

技术介绍

[0002]Anchor机制是大多数目标检测算法中不可或缺的一环，其主要是对数据集中的样本真实边界框进行聚类或其他方式生成的一种预测边界框参考值，通过设置预定义的先验框可以很好的来表示待检测目标的原始状态，得到更合理的数据样本边界框潜在分布。高质量的Anchor在小目标检测和后处理预测过程中可以起到优化调节作用。针对目标检测框宽高比极端变化问题，提出一种基于改进的自适应Anchor生成算法并结合YOLOv4目标检测模型得到该专利技术的DK
‑
YOLOv4模型。

技术实现思路

[0003]本专利技术旨在专利技术一种基于改进的自适应Anchor生成算法。
[0004]该算法的输入包括：真实框GT的数量Box＝{1,2,3
…
NP}，gt
wh
为训练数据中的所有真实框的宽高，thr设置为宽高比的阈值，T设置为最佳可能召回率BPR的下限阈值，输出一个最优的自适应Anchor个体集合，本专利技术包括如下的技术方案：
[0005](1)基于K
‑
means++聚类自适应Anchor生成算法
[0006]输入：真实框GT的数量Box＝{1,2,3
…
N}，Anchor聚类中心数量i，代表每一个真实框被选为下个聚类核心的概率区间
[0007]输出：K个自适应Anchor
[0008]1：从Box＝{1,2,3
…
N}中随机选取一个边界框作为初始聚类中心C1；
[0009]2：迭代选出剩下的K
‑
1个聚类中心
[0010]for i in range(k
‑
1)
[0011]3://每个边界框到聚类中心的距离
[0012]4:Sum_min_distance＝min(SUM(distance_Box))
[0013]5:P＝1
‑
IoU
B,C
//真实框和聚类中心IoU越大，距离越短；
[0014]6:P＝P+Sum_distance//越大该边界框作为下一个聚类中心的概率越大；
[0015]7：end for
[0016]8:将迭代选出K个边界框作为初始Anchor聚类中心C＝{C1,C2,....C
K
}；
[0017]9：迭代选出最终的K个自适应Anchor
[0018]10：for j in range(Box)
[0019]11:对于所有真实框，用IoU作为Anchor框和边界框的度量，将其中任意边界框分派给与其距离最小的Anchor框；
[0020]12：计算每个聚类簇中所有真实目标框宽和高的均值，更新Anchor；
[0021]13：end for
[0022]14：end for
[0023](2)基于差分进化的自适应Anchor生成算法
[0024]输入：真实框GT的数量i＝{1,2,3,
…
NP}，变量维度D，rand是[0,1]之间的随机数，j
rand
是[1,D]之间的随机数，CR是交叉概率，F是缩放因子。
[0025]输出：最优的自适应Anchor个体集合X
i
[0026]1.生成初始化种群NP
[0027]2.评价初始种群NP中所有个体适应值
[0028]3.While生成的Anchor框并不是最优预设边界框do
[0029]4.for i＝1to NP do
[0030]5.随机均匀选择r1≠r2≠r3≠i
[0031]6.L＝0
[0032]7.U
i
＝X
i
[0033]8.repeat
[0034]9.U
i
(j)＝X
r1
(j)+F
×
(X
r2
(j)
‑
X
r3
(j))
[0035]10.j
rand
＝(j
rand
+1)mod D
[0036]11.L＝L+1
[0037]12.until rand[0,1)>CR or L>D
[0038]13.end for
[0039]14.for i＝0to NP do
[0040]15.评价新个体U
i
[0041]16.if U
i
优于X
i then
[0042]17.X
i
＝U
i
[0043]18.end if
[0044]19.end for
[0045]end
[0046](3)基于K
‑
means++聚类和差分进化相结合的自适应Anchor生成算法输入：真实框GT的数量Box＝{1,2,3
…
NP}，gt
wh
为训练数据中的所有真实框的宽高，thr设置为宽高比的阈值，T设置为最佳可能召回率BPR的下限阈值输出：最优的自适应Anchor个体集合
[0047]1.使用算法1的K
‑
means++聚类方法迭代生成9个Anchor；
[0048]2.计算9个Anchor的宽高值Anchor
wh
；
[0049]3.计算gt
wh
和Anchor
wh
的比例值x，并将其统一至[0,1]的范围内；
[0050]4.计算最佳可能召回率BPR＝mean(max(min(x))＞thr)；
[0051]5.if BPR>T then
[0052]6.直接将第1步中得到9个Anchor作为最优解保存；
[0053]7.else使用K
‑
means++和DE算法进行Anchor重计算
[0054]8.for all Box＝{1,2,3
…
NP}do
[0055]9.按照算法2使用变异算子随机生成边界框；
[0056]10.计算变异边界框和Anchor的的交叉概率；
[0057]11.重计算样本个体的适应度函数值，选择最优值；
[0058]12.end for
[0059]13.使用K
‑
means++生成9个Anchor；
[0060]14.重复第2步操作，直到得到最优的9个最适合的Anchor。
[0061]验证该算法的数据集是MS C本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.针对目标检测框宽高比极端变化问题，提出一种基于改进的自适应Anchor生成算法并结合YOLOv4目标检测模型得到该发明的DK
‑
YOLOv4模型；该模型的技术方案如下：(1)基于K
‑
means++聚类自适应Anchor生成算法输入：真实框GT的数量Box＝{1,2,3
…
N}，Anchor聚类中心数量i，代表每一个真实框被选为下个聚类核心的概率区间输出：K个自适应Anchor(2)基于差分进化的自适应Anchor生成算法输入：真实框GT的数量i＝{1,2,3,
…
NP}，变量维度D，rand是[0,1]之间的随机数，j
rand
是[1,D]之间的随机数，CR是交叉概率，F...

【专利技术属性】
技术研发人员：杨晓雅，万冬厚，高辉，付苗苗，邓淼磊，张德贤，
申请(专利权)人：河南工业大学，
类型：发明
国别省市：