[发明专利]一种anchor-free算法转换为anchor-based算法的方法及系统

权利要求书

1.一种anchor-free算法转换为anchor-based算法的方法，其特征在于，包括：

步骤1：计算anchor-free算法中所有grid信息，以及获取anchor-free算法的预测值；

步骤2：根据anchor-based算法生成anchor的具体规则，生成与anchor-free算法中grid的形状及数量相同的anchor；

步骤3：根据anchor-free算法的grid预测值和anchor-based算法的anchor预测值，建立anchor-free算法转换为anchor-based算法的转换关系；

步骤4：根据所述转换关系，在神经网络中设置相应的层，实现将anchor-free预测值转换为anchor-based预测值。

2.根据权利要求1所述的方法，其特征在于，所述计算anchor-free算法中所有grid信息，包括：

用于预测物体的特征图，其下采样倍数在其对应的原图中平铺即为grid。

3.根据权利要求2所述的方法，其特征在于，所述获取anchor-free算法的预测值，包括：

将所述原图输入已经训练好的anchor-free模型，其输出值即预测值。

4.根据权利要求1所述的方法，其特征在于，所述步骤2的具体步骤为：

设置anchor-based算法中对原图的下采样倍数、anchor尺寸和长宽，生成与anchor-free算法中grid的形状和数量均相同的anchor。

5.根据权利要求1所述的方法，其特征在于，所述anchor预测值是对目标物体的中心点坐标、长和宽的编码值。

6.根据权利要求5所述的方法，其特征在于，所述grid预测值为grid的中心点分别到目标物体的真实位置框的四条边的距离。

7.根据权利要求6所述的方法，其特征在于，所述grid预测值和anchor预测值是对同一个目标物体的不同表示。

8.根据权利要求1所述的方法，其特征在于，所述步骤4包括：

将由anchor-free主体结构和anchor-free后处理构成的anchor-free模型转为由anchor-free主体结构、转换结构和anchor-based后处理构成的anchor-based结构；所述主体结构包括卷积、批归一化、激活、池化、反卷积、相加、变形。

9.根据权利要求8所述的方法，其特征在于，所述anchor-free主体结构的输出值为anchor-free算法中grid的输出值；所述转换结构用于将anchor-free主体结构的输出值转换得到的anchor-based算法中对应的anchor预测值作为其输出的值，并将该输出的值作为anchor-based后处理所需要的输入值。

10.一种anchor-free算法转换为anchor-based算法的系统，其特征在于，包括：

计算获取模块，用于计算anchor-free算法中所有grid信息，以及获取anchor-free算法的预测值；

生成模块，用于根据anchor-based算法生成anchor的具体规则，生成与anchor-free算法中grid的形状及数量相同的anchor；

转换模块，用于根据anchor-free算法的grid预测值和anchor-based算法的anchor预测值，建立anchor-free算法转换为anchor-based算法的转换关系；

实现模块，用于根据所述转换关系，在神经网络中设置相应的层，实现将anchor-free预测值转换为anchor-based预测值。