[发明专利]候选框处理方法、装置、设备及介质

权利要求书

1.一种候选框处理方法，其特征在于，应用于目标检测网络，所述目标检测网络包括候选框生成模块和目标检测模块，所述方法包括：

预测所述目标检测模块对所述候选框生成模块输出的各个候选框的喜好程度；

根据所述目标检测网络的当前使用阶段，将所述候选框生成模块输出的各个候选框中相应喜好程度的候选框输入到所述目标检测模块；其中，在不同的使用阶段，输入到所述目标检测模块的候选框的喜好程度不同。

2.根据权利要求1所述的方法，其特征在于，根据所述目标检测网络的当前使用阶段，将所述候选框生成模块输出的各个候选框中相应喜好程度的候选框输入到所述目标检测模块，包括：

在所述目标检测网络的当前使用阶段为训练阶段的情况下，从所述候选框生成模块输出的各个候选框中，采样喜好程度低于预设阈值的候选框并输入到所述目标检测模块。

3.根据权利要求1所述的方法，其特征在于，根据所述目标检测网络的当前使用阶段，将所述候选框生成模块输出的各个候选框中相应喜好程度的候选框输入到所述目标检测模块，包括：

在所述目标检测网络的当前使用阶段为应用阶段的情况下，从所述候选框生成模块输出的各个候选框中，采样喜好程度不低于所述预设阈值的候选框并输入到所述目标检测模块。

4.根据权利要求1-3任一项所述的方法，其特征在于，所述方法还包括：

在所述候选框生成模块的输出端串联预测模块；

在所述目标检测网络的当前使用阶段为训练阶段的情况下，根据所述目标检测模块的目标检测损失值，生成喜好程度标签；

根据所述预测模块输出的喜好程度预测结果与所述喜好程度标签，对所述预测模块进行训练，得到喜好程度预测模块，所述喜好程度预测模块用于预测所述目标检测模块对所述候选框生成模块输出的各个候选框的喜好程度。

5.根据权利要求4所述的方法，其特征在于，所述目标检测模块包括目标分类子模块和目标定位子模块；根据所述目标检测模块的目标检测损失值，生成喜好程度标签，包括：

根据所述目标分类子模块的分类损失值，生成喜好程度标签；或

根据所述目标分类子模块的分类损失值和所述目标定位子模块的定位损失值，生成喜好程度标签。

6.根据权利要求4所述的方法，其特征在于，根据所述预测模块输出的喜好程度预测结果与所述喜好程度标签，对所述预测模块进行训练，得到喜好程度预测模块，包括：

根据所述预测模块输出的喜好程度预测结果与所述喜好程度标签，确定喜好程度损失值；

根据所述喜好程度损失值，对所述预测模块进行参数更新。

7.根据权利要求6所述的方法，其特征在于，在所述目标检测网络的当前使用阶段为训练阶段的情况下，所述方法还包括：

根据所述目标检测模块输出的目标预测结果与所述候选框生成模块输出的候选框对应的标签，生成所述目标检测损失值；

根据所述喜好程度损失值和所述目标检测损失值，对所述候选框生成模块和所述目标检测模块进行参数更新。

8.根据权利要求1-7任一所述的方法，其特征在于，所述方法还包括：

在所述目标检测网络的当前使用阶段为应用阶段的情况下，获得所述目标检测模块输出的对相应的候选框进行检测的检测结果；

输出所述检测结果。

9.一种候选框处理装置，其特征在于，应用于目标检测网络，所述目标检测网络包括候选框生成模块和目标检测模块，所述装置包括：

预测模块，用于预测所述目标检测模块对所述候选框生成模块输出的各个候选框的喜好程度；

输入模块，用于根据所述目标检测网络的当前使用阶段，将所述候选框生成模块输出的各个候选框中相应喜好程度的候选框输入到所述目标检测模块；其中，在不同的使用阶段，输入到所述目标检测模块的候选框的喜好程度不同。