[发明专利]水产品检测方法、装置、电子设备和存储介质

权利要求书

1.一种水产品检测方法，其特征在于，包括：

获取待检测水产品图像；

将所述待检测水产品图像输入至水产品检测模型中，得到所述水产品检测模型输出的检测结果；

其中，所述检测结果包括处于目标状态的水产品的参数信息；所述水产品检测模型是根据包含目标状态的水产品的样本图像训练得到的。

2.根据权利要求1所述的水产品检测方法，其特征在于，所述方法还包括：

确定所述包含目标状态的水产品的样本图像；

对每个所述样本图像中处于所述目标状态的水产品采用标记框进行标记；

将标记后的样本图像输入至初始模型中进行训练，得到所述水产品检测模型。

3.根据权利要求2所述的水产品检测方法，其特征在于，所述将标记后的样本图像输入至初始模型中进行训练，得到所述水产品检测模型包括：

将标记后的样本图像输入至所述初始模型的轻量级卷积网络进行特征提取，得到目标数量个第一特征层；

根据目标数量个所述第一特征层确定每个所述样本图像的预测框；其中，所述预测框用于表征处于所述目标状态的水产品的预测位置；

根据每个所述样本图像的预测框和对应的标记框的对比调整所述初始模型的权重，直至所述初始模型收敛，得到所述水产品检测模型。

4.根据权利要求3所述的水产品检测方法，其特征在于，所述轻量级卷积网络中内置有注意力机制。

5.根据权利要求3所述的水产品检测方法，其特征在于，所述根据目标数量个所述第一特征层确定每个所述样本图像的预测框包括：

将目标数量个所述第一特征层传输至所述初始模型的加强特征提取网络进行池化和特征融合处理，得到目标数量个第二特征层；

根据目标数量个所述第二特征层确定每个所述样本图像的预测框。

6.根据权利要求5所述的水产品检测方法，其特征在于，所述根据目标数量个所述第二特征层确定每个所述样本图像的预测框包括：

将目标数量个所述第二特征层输入至所述初始模型的预测端，得到每个所述样本图像的边界框和所述边界框对应的置信度；

根据每个所述样本图像的边界框和对应的置信度确定所述预测框。

7.一种水产品检测装置，其特征在于，包括：

获取单元，用于获取待检测水产品图像；

检测单元，用于将所述待检测水产品图像输入至水产品检测模型中，得到所述水产品检测模型输出的检测结果；

其中，所述检测结果包括处于目标状态的水产品的参数信息；所述水产品检测模型是根据包含目标状态的水产品的样本图像训练得到的。

8.一种电子设备，包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现如权利要求1至6任一项所述的水产品检测方法的步骤。

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

10.一种计算机程序产品，包括计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1至6任一项所述的水产品检测方法的步骤。