摘要
本发明公开了一种基于Transformer的鱼群摄食决策方法,包括:获取鱼群运动视频,采用YOLOv5模型对每帧图像进行检测获得每条鱼的检测框及位置信息;计算每帧图像的鱼群质心并归一化:构建Delaunay三角网;根据二维高斯正态分布计算权重因子;获得鱼群质心坐标;计算相邻两帧图像的鱼群质心之差获得特征向量;对Transformer模型进行处理并将预设长度的特征向量输入模型进行训练;将待识别的预设长度的特征向量输入训练好的Transformer模型获得鱼群饥饿值,比较鱼群饥饿值和预设阈值,判断是否投食。该方法可避免多余饵料污染或鱼群营养不良,计算量低,准确率高,可实现养殖实时监测,泛化能力好。