一种图像检测模型的多尺度anchor初始化方法与装置制造方法及图纸

本发明专利技术涉及一种图像检测模型的多尺度anchor初始化方法，方法包括：将每种待检测图像对应的样本标签划分到对应的尺度内；根据每个尺度内的样本标签数量阈值，判断是否保留尺度；确定保留的每个尺度下的样本标签面积和宽高情况；根据样本标签面积和宽高情况，动态调整anchor数量并生成每个尺度下的初始化anchor；通过评价指标比较面积和宽高两种方法确定的anchor的情况，选择较达到评价指标的一组结果。本发明专利技术生成的初始化anchor既能够指导算法采用最合适的尺度，节省效率，避免尺度间的竞争；又能够与样本的分布情况一致，提升检测精度。

【技术实现步骤摘要】
一种图像检测模型的多尺度anchor初始化方法与装置
本申请涉及互联网
，尤其涉及一种图像检测模型的多尺度anchor初始化方法与装置。
技术介绍
目标检测算法(Detection)是深度学习计算机视觉(CV)领域一个非常重要的方向，算法过程是通过一系列带有标签的图像进行训练，提取目标特征，用于预测新的图像是否含目标并用矩形框标示，对目标图像进行标注，带有标签的图像的矩形框为目标矩形框。如图1所示，含有三个目标。目标检测训练过程中，Anchor需要设置有多个参数，其中非常重要的一个参数就是锚框(anchor)，其形态是矩形框。其原理是将一系列(常见的是9个)初始化的anchor做为基准，找到IoU(锚框和目标矩形框的面积交集/并集，如图2所示)和目标矩形框最接近的一个，并学习锚框和目标矩形框之间的变换关系。这种学习可近似看作线性变换，所以初始化的anchor和目标矩形框越接近，学习的效果越好，预测精度会越高。其中，目标矩形框，即标注的矩形框，通常称为标签。Anchor由两个参数构成，数量和大小。其中，anchor的数量一般本文档来自技高网...

【技术保护点】
1.一种图像检测模型的多尺度anchor初始化方法，其特征在于，所述方法包括：/n将每种待检测图像对应的样本标签划分到对应的尺度内；/n根据每个所述尺度内的样本标签数量阈值，判断是否保留所述尺度；/n确定保留的每个尺度下的样本标签面积和宽高情况；/n根据所述样本标签面积和宽高情况，确定anchor数量并生成所述尺度下的初始化anchor。/n

【技术特征摘要】
1.一种图像检测模型的多尺度anchor初始化方法，其特征在于，所述方法包括：
将每种待检测图像对应的样本标签划分到对应的尺度内；
根据每个所述尺度内的样本标签数量阈值，判断是否保留所述尺度；
确定保留的每个尺度下的样本标签面积和宽高情况；
根据所述样本标签面积和宽高情况，确定anchor数量并生成所述尺度下的初始化anchor。


2.根据权利要求1所述的图像检测模型的多尺度anchor初始化方法，其特征在于，将每种待检测图像对应的样本标签划分到对应的尺度内，包括：
根据尺度阈值，将每种待检测图像对应的样本标签划分到对应的尺度内，
所述尺度阈值为样本标签面积范围，所述样本标签面积范围为通过实验获取的能够对所述尺度下的样本标签有最佳检测结果的范围。


3.根据权利要求1所述的图像检测模型的多尺度anchor初始化方法，其特征在于，根据所述样本标签面积和宽高情况，确定anchor数量并生成所述尺度下的初始化anchor，包括：
根据每个尺度下的样本标签面积和宽高情况，通过不同的聚类的距离计算方式和动态调整anchor数量方法，生成所述尺度下的初始化anchor。


4.根据权利要求3所述的图像检测模型的多尺度anchor初始化方法，其特征在于，根据每个尺度下的样本标签面积和宽高情况，通过不同的聚类的距离计算方式和动态调整anchor数量方法，生成所述尺度下的初始化anchor，包括以下步骤：
根据每个尺度下的样本标签的面积排序，在每个尺度下等距离的选取预设数量的样本标签作为初始化聚类中心；
分别根据样本标签的面积和宽高两方面进行聚类的计算；
根据面积和宽高的聚类计算方法，两方面进行计算样本标签与所有初始化聚类中心的距离，将所述样本标签分类到距离最近的初始化聚类中心；
根据所述标签分类步骤生成的新的初始化聚类中心，取平均值作为新聚类中心；
计算每一类样本标签与新聚类中心距离的标准偏差和计算每一类别样本标签间的新聚类中心的距离，选择增加或者减小聚类中心个数；
通过面积和宽高两种距离的计算方法，得到的两组不同的聚类结果中，判断聚类中心个数是否变化；
若聚类中心个数不变化，则通过类内和类间的评价指标，从两组不同的聚类结果中，选取达到评价指标的一组聚类中心，作为初始化anchor。


5.根据权利要求1所述的图像检测模型的多尺度anchor初始化方法，其特征在于，通过面积和宽高两种距离的计算方法，得到的两组不同的聚类结果中，判断聚类中心个数是否变化，包括：
通过面积和宽高两种距离的计算方法，得到的两组不同的聚类结果中，判断聚类中心个数是否变化；
若聚类中心个数变化，则再次进行根据面积和宽高的聚类计算方法，两方面进行计算样本标签与所有初始化聚类中心的距离，将所述样本标签分类到距离最近的初始化聚类中心、根据所述标签分类步骤生成的新的初始化聚类中心，取平均值作为新聚类中心、计算每一类样本标签与新聚...

【专利技术属性】
技术研发人员：刘铎，
申请(专利权)人：北京凌云光技术集团有限责任公司，
类型：发明
国别省市：北京;11