[发明专利]网络模型的训练方法及装置、标注信息的确定方法及装置

权利要求书

1.一种网络模型的训练方法，其特征在于，包括：

确定初始网络模型；

确定待标注样本图像对应的第一标注信息；

基于所述第一标注信息和所述待标注样本图像确定所述待标注样本图像对应的概率图像；

基于待标注样本图像和所述待标注样本图像对应的概率图像训练所述初始网络模型，以生成图像理解模型，其中，所述图像理解模型用于基于待标注图像生成所述待标注图像对应的特征图和修复图像，所述概率图像用于表征所述待标注样本图像中的显示单元被擦除的概率；

其中，所述初始网络模型包括特征提取模块和图像修复模块，所述基于待标注样本图像和所述待标注样本图像对应的概率图像训练所述初始网络模型，包括：

基于所述待标注样本图像和所述概率图像生成所述待标注样本图像对应的擦除图像；

将所述擦除图像输入所述特征提取模块，以生成特征图；

将所述特征图输入所述图像修复模块，以生成修复图像；

基于所述待标注样本图像和所述修复图像训练所述初始网络模型。

2.根据权利要求1所述的方法，其特征在于，所述基于所述待标注样本图像和所述概率图像生成所述待标注样本图像对应的擦除图像，包括：

基于所述概率图像对所述待标注样本图像进行擦除操作，以生成第一阶段擦除图像，其中，所述第一阶段擦除图像包括第一擦除区域；

将所述第一阶段擦除图像确定为所述擦除图像。

3.根据权利要求2所述的方法，其特征在于，所述第一擦除区域的数量为多个，在所述将所述第一阶段擦除图像确定为所述擦除图像之前，还包括：

基于所述多个第一擦除区域各自对应的尺寸信息和预设尺寸条件生成第一擦除区域集合；

基于所述第一擦除区域集合中的第一擦除区域之间的距离信息，确定第二阶段擦除图像；

其中，所述将所述第一阶段擦除图像确定为所述擦除图像，包括：

将所述第二阶段擦除图像确定为所述擦除图像。

4.根据权利要求3所述的方法，其特征在于，所述基于所述第一擦除区域集合中的第一擦除区域之间的距离信息，确定第二阶段擦除图像，包括：

针对所述第一擦除区域集合中的任意两个第一擦除区域，计算所述两个第一擦除区域之间的欧式距离；

如果所述欧式距离符合预设连通条件，则将所述两个第一擦除区域进行连通处理；

基于经所述连通处理后的所述第一擦除区域集合生成所述第二阶段擦除图像。

5.根据权利要求1所述的方法，其特征在于，所述基于所述第一标注信息和所述待标注样本图像确定所述概率图像，包括：

基于所述第一标注信息确定所述待标注样本图像的第一区域和第二区域，其中，所述第一区域包括多个第一显示单元，所述第二区域包括多个第二显示单元；

针对所述多个第一显示单元，分别计算所述第一显示单元和与所述第一显示单元最接近的第二显示单元之间的第一欧式距离，以生成第一距离集合；

针对所述多个第二显示单元，分别计算所述第二显示单元和与所述第二显示单元最接近的第一显示单元之间的第二欧式距离，以生成第二距离集合；

基于所述第一距离集合和所述第二距离集合生成所述概率图像。

6.根据权利要求5所述的方法，其特征在于，所述第一距离集合包括多个第一欧式距离，所述第二距离集合包括多个第二欧式距离，所述基于所述第一距离集合和所述第二距离集合生成所述概率图像，包括：

基于所述多个第一欧式距离确定预设距离阈值；

基于所述多个第一欧式距离与所述预设距离阈值的关系，确定所述多个第一显示单元各自对应的擦除概率；

基于所述多个第二欧式距离与所述预设距离阈值的关系，确定所述多个第二显示单元各自对应的擦除概率；

基于所述多个第一显示单元各自对应的擦除概率和所述多个第二显示单元各自对应的擦除概率，生成所述概率图像。