[发明专利]一种抠图网络训练方法及抠图方法

权利要求书

1.一种抠图网络训练方法，其特征在于，包括：

获取带标注和无标注的样本图像，所述样本图像包括目标物，所述标注是所述样本图像的目标物抠图结果；

利用多重知识蒸馏的方式，基于所述样本图像对教师网络和学生网络进行训练，得到轻量化的用于目标物抠图的学生网络。

2.如权利要求1所述的抠图网络训练方法，其特征在于，所述利用多重知识蒸馏的方式，基于所述样本图像对教师网络和学生网络进行训练，得到轻量化的用于目标物抠图的学生网络，包括：

先采用离线蒸馏后采用自监督蒸馏的方式，基于所述样本图像对教师网络和学生网络进行训练，得到轻量化的用于目标物抠图的学生网络。

3.如权利要求2所述的抠图网络训练方法，其特征在于，所述先采用离线蒸馏后采用自监督蒸馏的方式，基于所述样本图像对教师网络和学生网络进行训练，得到轻量化的用于目标物抠图的学生网络，包括：

先采用离线蒸馏的方式，使用带标注的样本图像对教师网络和学生网络进行训练，得到经训练的学生网络；再采用自监督蒸馏的方式，将教师网络替换为离线蒸馏方式训练得到的学生网络，对学生网络的参数进行随机初始化，基于带标注和不带标注的样本图像对教师网络和学生网络进行训练，得到用于目标物抠图的学生网络。

4.如权利要求1所述的抠图网络训练方法，其特征在于，所述利用多重知识蒸馏的方式，基于所述样本图像对教师网络和学生网络进行训练，得到轻量化的用于目标物抠图的学生网络，包括：

先采用半监督蒸馏后采用自监督蒸馏的方式，基于所述样本图像对教师网络和学生网络进行训练，得到轻量化的用于目标物抠图的学生网络。

5.如权利要求4所述的抠图网络训练方法，其特征在于，所述先采用半监督蒸馏后采用自监督蒸馏的方式，基于所述样本图像对教师网络和学生网络进行训练，得到轻量化的用于目标物抠图的学生网络，包括：

先采用半监督蒸馏的方式，使用不带标注的样本图像对教师网络和学生网络进行训练，得到经训练的学生网络；再采用自监督蒸馏的方式，将教师网络替换为半监督蒸馏方式训练得到的学生网络，对学生网络的参数进行随机初始化，基于带标注和不带标注的样本图像对教师网络和学生网络进行训练，得到用于目标物抠图的学生网络。

6.如权利要求1所述的抠图网络训练方法，其特征在于，所述利用多重知识蒸馏的方式，基于所述样本图像对教师网络和学生网络进行训练，得到轻量化的用于目标物抠图的学生网络，包括：

先采用离线蒸馏、然后采用半监督蒸馏、最后采用自监督蒸馏的方式，基于所述样本图像对教师网络和学生网络进行训练，得到轻量化的用于目标物抠图的学生网络。

7.如权利要求6所述的抠图网络训练方法，其特征在于，所述先采用离线蒸馏、然后采用半监督蒸馏、最后采用自监督蒸馏的方式，基于所述样本图像对教师网络和学生网络进行训练，得到轻量化的用于目标物抠图的学生网络，包括：

先采用离线蒸馏的方式，使用带标注的样本图像对教师网络和学生网络进行训练，得到经训练的学生网络；然后采用半监督蒸馏的方式，保持教师网络不变，将学生网络的参数更新为离线蒸馏方式学习得到的学生网络的参数，使用不带标注的样本图像对教师网络和学生网络进行训练，得到经训练的学生网络；最后采用自监督蒸馏的方式，将教师网络替换为半监督蒸馏方式训练得到的学生网络，对学生网络的参数进行随机初始化，基于带标注和不带标注的样本图像对教师网络和学生网络进行训练，得到用于目标物抠图的学生网络。

8.一种抠图方法，其特征在于，包括：

采集包括目标物的当前图像；

将所述当前图像输入如权利要求1至7任一项所述的抠图网络训练方法得到的用于目标物抠图的学生网络，输出所述当前图像对应的抠图结果。