[发明专利]一种图像筛选方法、装置、电子设备及存储介质

说明书

本申请实施例提供的一种图像筛选方法、装置、电子设备及存储介质，采用预训练的图像分割模型对包含待分析对象的待处理图像进行图像语义分割，得到待分析对象对应的掩膜图像，对掩膜图像中的待分析对象进行位置矫正处理，得到标准掩膜图像，对标准掩膜图像进行特征提取，根据提取的特征利用预训练的姿态确定模型确定标准掩膜图像中待分析对象的姿态是否满足预设的姿态要求标准，将姿态满足姿态要求的待分析对象所属的标准掩膜图像作为用于后续生物分析的最终掩膜图像。通过上述方式对待处理图像中包含的待分析对象进行姿态筛选，使得最终掩膜图像中的待分析对象的姿势满足预设姿态要求，保证了后续对待分析对象进行生物分析时分析结果的准确性。

技术领域

本申请涉及人工智能领域，尤其涉及一种图像筛选方法、装置、电子设备及存储介质。

背景技术

渔业是中国农业的重要组成部分，而且不同于其他主要渔业国，中国的水产品主要来自于水产养殖业，并且是世界上唯一养殖水产品总量超过捕捞总量的主要渔业国，2018年，中国养殖水产品总量4991万吨，占到中国水产品总量的77％以上，占世界养殖水产品总量的60％以上。

目前在水产养殖过程中，都是凭借经验进行投料喂食，没有鱼的生长趋势的记录，如果能够记录鱼的大致生长趋势，那么对于投料时间以及用量会有指导作用，能够在一定程度上提高肉料比，并节省饲料。

目前可以通过在水下安装摄像机来采集鱼的图像，通过对采集的图像进行生物分析来确定鱼的生物信息(例如体尺信息等)，但是由于鱼在水中通常为运动状态，在运动过程中鱼的尾巴会摆动，这就有可能使得采集到的图像中鱼的姿势不是舒展的，进而使得生物分析得到的鱼的生物信息不准确，例如分析得到的体长比实际体长短，从而影响到对鱼生长趋势的确定。

发明内容

为了解决上述技术问题或者至少部分地解决上述技术问题，本申请提供了一种。

第一方面，本申请提供了一种图像筛选方法，包括：

采用预训练的图像分割模型对包含待分析对象的待处理图像进行图像语义分割，得到掩膜图像；

对所述掩膜图像中的所述待分析对象进行位置矫正处理，得到标准掩膜图像；

对所述标准掩膜图像进行特征提取，得到所述标准掩膜图像的特征；

将所述标准掩膜图像的特征输入预训练的姿态确定模型，确定所述标准掩膜图像中包含的待分析对象的姿态是否满足预设的姿态要求；

确定姿态满足预设姿态要求的待分析对象所属的标准掩膜图像作为最终掩膜图像，以通过所述最终掩膜图像对所述待分析对象进行生物分析。

在一种可能的实现方式中，所述预训练的姿态确定模型为利用预设的模型超参数和多张包含所述待分析对象的样本图像的特征对预设的XGBoost模型进行训练后得到的模型，其中所述样本图像中包含的所述待分析对象的姿态已知是否满足预设的姿态要求。

在一种可能的实现方式中，对所述掩膜图像中的所述待分析对象进行位置矫正处理之前，所述方法还包括：

检测所述掩膜图像中包含的待分析对象的数量；

若所述数量大于1，则将所述掩膜图像拆分为对应数量张只包含单个待分析对象的掩膜图像；

对只包含单个待分析对象的掩膜图像，执行对所述掩膜图像中的所述待分析对象进行位置矫正处理的步骤。

在一种可能的实现方式中，对所述掩膜图像中的所述待分析对象进行位置矫正处理之前，所述方法还包括：

确定包含的待分析对象不完整的待剔除掩膜图像；

将所述待剔除掩膜图像从所述掩膜图像中删除；