[发明专利]图像分割模型的处理方法和处理装置

权利要求书

1.一种图像分割模型的处理方法，所述图像分割模型包括特征提取子模型和图像分割子模型，所述特征提取子模型用于提取图像的特征，所述图像分割子模型用于根据所述提取的特征对所述图像进行分割，其特征在于，包括：

对所述特征提取子模型进行层宽调整，以得到第一特征提取子模型；

根据所述第一特征提取子模型和所述图像分割子模型得到目标图像分割模型。

2.如权利要求1所述的处理方法，其特征在于，所述对所述特征提取子模型进行层宽调整，包括：

增大所述特征提取子模型的通道数量，得到第二特征提取子模型；

为所述第二特征提取子模型生成K个不同的第一二值化编码，所述第一二值化编码包括多个二值化数值，所述多个二值化数值与所述第二特征提取子模型的多个通道一一对应，所述多个二值化数值中的每个二值化数值用于指示所述每个二值化数值对应的通道保留还是去除，K为大于1的整数；

根据所述K个不同的第一二值化编码保留或者去除所述第二特征提取子模型的通道，得到K个第三特征提取子模型；

根据所述K个第三特征提取子模型中每个第三特征提取子模型对图像进行特征提取时的交并比和计算量，从所述K个第三特征提取子模型中选择M个第三特征提取子模型，M为大于1的整数；

对所述M个第三特征提取子模型对应的M个第一二值化编码进行交叉和/或变异处理，得到S个第二二值化编码，S为大于1整数；

根据所述S个第二二值化编码中的每个第二二值化编码保留或者去除所述第二特征提取子模型的通道，得到S个第四特征提取子模型；

将所述第四特征提取子模型作为所述第三特征提取子模型，取K＝S，重复执行上述第四个操作至第六个操作T次；

将最后一次得到的S个第四特征提取子模型的一个作为所述第一特征提取子模型。

3.根据权利要求2所述的处理方法，其特征在于，所述M个第三特征提取子模型中的第j个第三特征提取子模型被选择的概率满足如下公式：

其中，Pr(b_j)表示所述第j个第三特征提取子模型被选择的概率，f(b_j)满足以下公式：

f(b_j)＝mIoU(j)+α/N(j)

其中，mIoU(j)表示所述第j个第三特征提取子模型对图像进行特征提取时的交并比，N(j)表示所述第个j第三特征提取子模型对所述图像进行特征提取时的计算量，α是一个预设参数。

4.根据权利要求1至3中任一项所述的处理方法，其特征在于，所述根据所述第一特征提取子模型和所述图像分割子模型得到目标图像分割模型，包括：

使用教师特征提取模型对所述第一特征提取模型进行知识蒸馏，得到第五特征提取子模型；

根据所述第五特征提取子模型和所述图像分割子模型得到所述目标图像分割模型。

5.如权利要求4所述的处理方法，其特征在于，所述第五特征提取子模型的损失函数满足如下关系：

其中，G_ti表示所述教师特征提取模型输出的T个尺度特征中的第i个尺度特征，G_si表示所述第五特征提取子模型输出T个尺度特征中的第i个尺度特征，T为正整数，所述教师特征提取模型输出的第i个尺度特征与所述第五特征提取子模型输出的第i个尺度特征的尺度相同，“|| ||”表示矩阵范数。

6.如权利要求4或5所述的处理方法，其特征在于，所述根据所述第五特征提取子模型和所述图像分割子模型得到所述目标图像分割模型，包括：

根据教师图像分割模型对所述图像分割子模型进行知识蒸馏，得到目标图像分割子模型，其中，所述目标图像分割模型包括所述第五特征提取子模型和所述目标图像分割子模型。