一种鱼类识别方法

本发明公开了一种鱼类识别方法，应用于图像识别技术领域，包括：对鱼类识别样本数据进行数据预处理，并对预处理后的数据进行数据划分，得到训练集以及测试集；基于训练集，训练基于改进YOLOV5的鱼类识别模型；其中，YOLOV5的改进包括：使用Swi n Transformer作为骨干网络引入YOLOV5中、使用CARAFE上采样算法代替线性插值上采样、使用BIFPN加权特征融合代替普通拼接操作以及使用新的颈部网络替换YOLOV5中的颈部网络；利用Rep‑G I oU Loss损失函数计算改进YOLOV5的损失，输入测试集至训练收敛后的鱼类识别模型，得到鱼类识别结果。本发明有效提高了鱼类识别精度。