[发明专利]冰激凌图像训练方法、检测方法及处理设备

权利要求书

1.一种冰激凌图像训练方法，其特征在于，包括以下步骤：

S100：获取真实场景的多张冰激凌样本图像并对冰激凌进行标注，得到多张带有原始mask标签的冰激凌标注图像；

S200：选择所述冰激凌标注图像，根据所述原始mask标签抠出所有的冰激凌块图；

S300：将所述冰激凌块图制作为含有四通道的冰激凌贴图；

S400：对每一张所述冰激凌标注图像，随机选取多张所述冰激凌贴图依次贴附于所述冰激凌标注图像上；

S500：制作一张记录所述冰激凌标注图像与所述冰激凌贴图之间贴附情况的白板图；

S600：生成每一个所述冰激凌贴图贴上所述冰激凌标注图像之后的原始mask图，并将所述原始mask图与所述白板图进行与运算，得到最终mask图，在所述白板图上将所述最终mask图中所述冰激凌贴图的对应位置置为0；

S700：全部的所述冰激凌贴图贴附完成后，依次将所有的所述原始mask标签与S600中最后得到的白板图进行与运算，得到每个所述原始mask标签在贴附所述冰激凌块图后的实际mask标签；

S800：重复执行所述S100~S700步骤，通过mask RCNN实例分割模型进行训练，直至训练的所述冰激凌样本图像数量达到设定值。

2.根据权利要求1所述的冰激凌图像训练方法，其特征在于，所述S100步骤中，所述冰激凌样本图像的数量为1000；所述S400步骤中，所述冰激凌贴图的数量为100。

3.根据权利要求1所述的冰激凌图像训练方法，其特征在于，所述S200步骤中，所述冰激凌块图为单个冰激凌图像所对应的块图。

4.根据权利要求1所述的冰激凌图像训练方法，其特征在于，所述S300步骤中，所述四通道分别为红色通道、绿色通道、蓝色通道和透明通道。

5.根据权利要求1所述的冰激凌图像训练方法，其特征在于，所述S400步骤中，所述冰激凌贴图对所述冰激凌样本图像中的冰激凌进行部分遮挡。

6.根据权利要求1所述的冰激凌图像训练方法，其特征在于，所述S700、S800之间还包括步骤S710：

S710：对每个识别出的冰激凌，如果所述实际mask标签之间有相交部位，则将所述相交部位的位置进行提取，计算所述相交部位中每个像素的概率，将所述相交部位的像素位置分配给概率最高的所述实际mask标签。

7.一种冰激凌图像检测方法，其特征在于，通过权利要求1-6中任一所述冰激凌图像训练方法对冰激凌图像进行检测，得到所述冰激凌图像中被遮挡冰激凌的所述实际mask标签。

8.根据权利要求7所述的冰激凌图像检测方法，其特征在于，所述检测方法还能够对所述冰激凌图像中的表层未被遮挡冰激凌进行标注，得到表层mask标签。

9.根据权利要求8所述的冰激凌图像检测方法，其特征在于，通过所述实际mask标签、表层mask标签的数量能够对所述冰激凌图像中的冰激凌进行计数。

10.一种处理设备，其特征在于，包括：

一个或多个处理器；

存储器，用于存储一个或多个计算机程序，一个或多个所述处理器用于执行所述存储器存储的一个或多个计算机程序，以使一个或多个所述处理器执行如权利要求7-9中任一所述的冰激凌图像检测方法。