[发明专利]一种禽蛋大小头检测方法、装置、电子设备及存储介质

说明书

本申请涉及一种禽蛋大小头检测方法、装置、电子设备及存储介质，该方法包括：对禽蛋侧面图像进行图像分割，得到禽蛋侧面轮廓；确定所述禽蛋侧面轮廓沿短轴方向的中线；根据所述中线将所述禽蛋侧面轮廓划分为第一侧面轮廓和第二侧面轮廓；通过比较所述第一侧面轮廓和第二侧面轮廓确定禽蛋大小方向。该技术方案可以准确确定大小头方向，为降低禽蛋运输过程中的破损率、提高禽蛋孵化时的孵化率及出雏率提供了保证。并且，仅需训练一个图像分割模型即可，模型训练所需样本图像数量较少且不涉及分类错误问题。另外，模型泛化性较好。

技术领域

本申请涉及人工智能技术领域，尤其涉及一种禽蛋大小头检测方法、装置、电子设备及存储介质。

背景技术

禽蛋大小头定向排列的主要目的是使所有禽蛋都大头向上放置在蛋盒或蛋盘中，防止蛋黄粘结在蛋壳上，延长保存期。在孵化场景，胚胎需要气室朝上才能正常的发育，而气室绝大多数情况位于在禽蛋的大头方向，因此，禽蛋的大小头摆放方向会直接影响出雏率，对生产效益产生直接影响。另外，在运输过程中大头朝上会减少禽蛋的破损率。

现有技术中，通过计算机视觉对禽类种蛋的判别方法主要是使用卷积神经网络进行大小头分类。由于没有明确的特征指定去学习，卷积神经网络分类在这个大小头分类任务上像是一个黑盒模型，分类错误后无法弥补。

发明内容

为了解决上述技术问题或者至少部分地解决上述技术问题，本申请实施例提供了一种禽蛋大小头检方法、装置、电子设备及存储介质。

第一方面，本申请实施例提供了一种禽蛋大小头检测方法，包括：

对禽蛋侧面图像进行图像分割，得到禽蛋侧面轮廓；

确定所述禽蛋侧面轮廓沿短轴方向的中线；

根据所述中线将所述禽蛋侧面轮廓划分为第一侧面轮廓和第二侧面轮廓；

通过比较所述第一侧面轮廓和第二侧面轮廓确定禽蛋大小方向。

可选的，所述对禽蛋侧面图像进行图像分割，得到禽蛋侧面轮廓，包括：

通过预先训练的语义分割模型对所述禽蛋侧面图像进行语义分割，得到所述禽蛋侧面图像对应的第一掩模图像；

通过轮廓检测算法从所述第一掩模图像检测禽蛋边缘像素点，得到包括所述禽蛋边缘像素点坐标的所述禽蛋侧面轮廓。

可选的，所述通过预先训练的语义分割模型对所述禽蛋侧面图像进行语义分割，得到所述禽蛋侧面图像对应的第一掩模图像，包括：

获取语义分割后所述禽蛋侧面图像对应的类别标签；

当根据所述类别标签确定存在遮挡禽蛋的遮挡物时，将所述遮挡物与禽蛋对应的类别标签合并，得到无遮挡物的第一掩膜图像。

可选的，所述通过轮廓检测算法从所述禽蛋掩模图像检测禽蛋边缘像素点，包括：

获取所述第一掩膜图像中前景图像的前景像素点坐标；

对所述前景像素点坐标进行主成分分析，得到主成分向量；

根据所述主成分向量确定将所述前景图像转正的旋转矩阵；

根据所述旋转矩阵对所述前景图像进行旋转，得到第二掩膜图像；

通过所述轮廓检测算法从所述第二掩膜图像检测得到所述禽蛋边缘像素点坐标。

可选的，所述确定所述禽蛋侧面轮廓沿短轴方向的中线，包括：

确定所述禽蛋侧面轮廓沿长轴方向的第一顶点和第二顶点；

根据所述第一顶点和第二顶点的第一连线得到所述禽蛋侧面轮廓沿长轴方向的中点；