[发明专利]一种锚框参数的确定方法、装置、设备及可读存储介质

权利要求书

1.一种锚框参数的确定方法，其特征在于，包括：

获取图像，对图像进行预处理，所述图像包括多个目标标注框；

根据输出特征图的跨度、预处理后图像的第一尺寸和预设的锚框参数生成各特征图的锚框；所述根据输出特征图的跨度、各所述特征图的第二尺寸及预设的锚框参数生成各所述特征图的锚框的步骤包括：根据输出特征图的跨度及预设的锚框参数生成各所述特征图对应的基础锚框；根据各所述特征图的第二尺寸及所述基础锚框生成各所述特征图的锚框；锚框参数为锚框尺度和锚框高宽比；输出特征图的跨度为输入图像与特征图的尺寸之间的比值；

根据多个所述目标标注框从多个所述锚框中选择出作为正样本的多个目标锚框，得到所述目标锚框与所述目标标注框的多个匹配对；

根据所述目标锚框与所述目标标注框的多个匹配对计算损失；

根据预设规则对所述锚框参数进行更新，并返回获取图像，对图像进行预处理，所述图像包括多个目标标注框的步骤，直至所述损失收敛，得到目标锚框参数。

2.根据权利要求1所述的锚框参数的确定方法，其特征在于，所述根据输出特征图的跨度、预处理后图像的第一尺寸和预设的锚框参数生成各特征图的锚框的步骤包括：

根据输出特征图的跨度和预处理后图像的第一尺寸计算各特征图的第二尺寸；

根据输出特征图的跨度、各所述特征图的第二尺寸和预设的锚框参数生成各所述特征图的锚框。

3.根据权利要求1所述的锚框参数的确定方法，其特征在于，所述根据输出特征图的跨度及预设的锚框参数生成各所述特征图对应的基础锚框的步骤包括：

根据输出特征图的跨度及预设的锚框参数计算各所述特征图对应的基础锚框参数；

根据各所述基础锚框参数生成各所述特征图的基础锚框。

4.根据权利要求1所述的锚框参数的确定方法，其特征在于，所述根据所述目标锚框与所述目标标注框的多个匹配对计算损失的步骤包括：

计算所述目标锚框与所述目标标注框的每个匹配对的GIoU损失；

根据每个匹配对的GIoU损失计算GIoU损失均值及GIoU损失标准差；

将所述GIoU损失均值与所述GIoU损失均值的第一权重的第一乘积、所述GIoU损失标准差与所述GIoU损失标准差的第二权重的第二乘积进行累加，得到所述损失。

5.一种锚框参数的确定装置，其特征在于，包括：

获取单元，用于获取图像，对图像进行预处理，所述图像包括多个目标标注框；

生成单元，用于根据输出特征图的跨度、预处理后图像的第一尺寸和预设的锚框参数生成各特征图的锚框；所述生成子单元，包括：第一生成子单元，用于根据输出特征图的跨度及预设的锚框参数生成各所述特征图对应的基础锚框；第二生成子单元，用于根据各所述特征图的第二尺寸及所述基础锚框生成各所述特征图的锚框；锚框参数为锚框尺度和锚框高宽比；输出特征图的跨度为输入图像与特征图的尺寸之间的比值；

选择单元，用于根据多个所述目标标注框从多个所述锚框中选择出作为正样本的多个目标锚框，得到所述目标锚框与所述目标标注框的多个匹配对；

计算单元，用于根据所述目标锚框与所述目标标注框的多个匹配对计算损失；

更新单元，用于根据预设规则对锚框参数进行更新，并返回获取图像，对图像进行预处理，所述图像包括多个目标标注框的步骤，直至所述损失收敛，得到目标锚框参数。

6.根据权利要求5所述的锚框参数的确定装置，其特征在于，所述生成单元包括：

计算子单元，用于根据输出特征图的跨度和预处理后图像的第一尺寸计算各特征图的第二尺寸；

生成子单元，用于根据输出特征图的跨度、各所述特征图的第二尺寸和预设的锚框参数生成各所述特征图的锚框。

7.一种电子设备，其特征在于，包括：

至少一个处理器；以及与所述至少一个处理器通信连接的存储器；其中，所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器执行如权利要求1-4任意一项所述的锚框参数的确定方法。

8.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质存储有计算机指令，所述计算机指令用于使计算机执行如权利要求1-4任意一项所述的锚框参数的确定方法。