[发明专利]一种南美白对虾个体标记识别方法、装置、计算机设备及存储介质

权利要求书

1.一种南美白对虾个体标记识别方法，其特征在于，包括步骤：

S1、采集进行颜色标记后的南美白对虾个体的俯视图图像样本；

S2、基于精简的YOLO v3进行俯视图的目标检测，获取虾尾区域的包围盒信息，将虾尾的图像裁剪出来作为下一步骤的输入；

S3、基于HR Net网络推理得到虾尾颜色点热图，并将虾尾颜色点热图输入到Softmax中得到颜色点类别信息，确定南美白对虾个体的个体标签。

2.根据权利要求1所述的南美白对虾个体标记识别方法，其特征在于，所述步骤S1包括：对南美白对虾个体进行颜色标记时，从红、绿、蓝、黄、橙、紫、白七种颜色中随机选择可重复的4个颜色注射在虾尾两节4个位置处。

3.根据权利要求1或2所述的南美白对虾个体标记识别方法，其特征在于，所述步骤S1包括：在装满3/4水的水盒中放入南美白对虾，将水盒放入图像样本采集设备，在所述图像样本采集设备内布置工业相机，启动工业相机，采集到南美白对虾的俯视图图像样本。

4.根据权利要求3所述的南美白对虾个体标记识别方法，其特征在于，所述步骤S1包括：采集1000张南美白对虾在无约束水中的俯视图图像样本。

5.根据权利要求1所述的南美白对虾个体标记识别方法，其特征在于，所述步骤S2包括：使用Darknet19代替YOLOv3中的Darknet53来精简网络结构，同时，将3个尺度的输出精简到2个尺度的输出，每个尺度上选择两个不同大小的Anchors；网络以416x416x3的图片作为输入，两个尺度的输出分支大小分别为13x13x18和26x26x18用来进行多尺度融合；最后使用非极大抑制算法筛选出置信度和IOU最大的边框即为虾尾区域。

6.根据权利要求1所述的南美白对虾个体标记识别方法，其特征在于，所述步骤S3包括：将截取的虾尾图像输入到骨干网络HR Net中，推理得到4x6的虾尾颜色点热图，并分别将4组虾尾颜色点热图输入到Softmax中得到颜色点类别信息。

7.一种南美白对虾个体标记识别装置，其特征在于，包括：

数据采集单元，采集南美白对虾个体的俯视图图像样本；

虾尾检测模块，精简的YOLO v3进行俯视图的目标检测，获取虾尾区域的包围盒信息，将虾尾的图像裁剪出来作为下一步骤的输入；

颜色点检测模块，基于HR Net网络推理得到虾尾颜色点热图，并将虾尾颜色点热图输入到Softmax中得到颜色点类别信息，确定南美白对虾个体的个体标签。

8.一种计算机设备，其特征在于，所述设备包括存储器及处理器，所述存储器上存储有计算机程序，所述处理器执行所述计算机程序时实现如权利要求1至6中任一项所述的方法。

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