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

权利要求书

1.一种禽蛋大小头检测方法，其特征在于，包括：

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

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

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

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

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

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

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

2.根据权利要求1所述的方法，其特征在于，所述通过预先训练的语义分割模型对所述禽蛋侧面图像进行语义分割，得到所述禽蛋侧面图像对应的第一掩模图像，包括：

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

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

3.根据权利要求1所述的方法，其特征在于，所述通过轮廓检测算法从所述禽蛋掩模图像检测禽蛋边缘像素点，包括：

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

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

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

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

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

4.根据权利要求1所述的方法，其特征在于，所述确定所述禽蛋侧面轮廓沿短轴方向的中线，包括：

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

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

获取经过所述中点且垂直于所述第一连线的中线。

5.根据权利要求1所述的方法，其特征在于，所述通过比较所述第一侧面轮廓和第二侧面轮廓确定禽蛋大小方向，包括：

将所述第一侧面轮廓沿所述中线进行垂直翻转，得到第三侧面轮廓；

将所述第三侧面轮廓与所述第二侧面轮廓进行比较，确定所述禽蛋大小头方向；

或，

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

将所述第二侧面轮廓沿所述中线进行垂直翻转，得到第四侧面轮廓；

将所述第一侧面轮廓与所述第四侧面轮廓进行比较，确定所述禽蛋大小头方向。

6.根据权利要求1所述的方法，其特征在于，所述方法还包括：

根据所述禽蛋大小头方向生成分拣指令，所述分拣指令用于控制分拣设备对禽蛋执行相应的分拣操作；

将所述分拣指令发送至所述分拣设备。

7.一种禽蛋大小头检测装置，其特征在于，包括：

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

确定模块，用于确定所述禽蛋侧面轮廓沿短轴方向的中线；

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

比较模块，用于通过比较所述第一侧面轮廓和第二侧面轮廓确定禽蛋大小方向；

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

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

8.一种电子设备，其特征在于，包括：处理器、通信接口、存储器和通信总线，其中，处理器，通信接口，存储器通过通信总线完成相互间的通信；

所述存储器，用于存放计算机程序；

所述处理器，用于执行所述计算机程序时，实现权利要求1-6任一项所述的方法步骤。

9.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，该计算机程序被处理器执行时实现权利要求1-6任一项所述的方法步骤。