基于深度学习的河豚鱼个体识别方法
摘要:
基于深度学习的河豚鱼个体识别方法,涉及计算机视觉以及个体识别。采集河豚鱼图片,并进行预处理,包括河豚鱼图像掩膜标注以及数据增强;在预处理完的河豚鱼数据集上训练图像分割模型,得到训练好的分割模型;将用于河豚鱼特征提取的河豚鱼图像送入训练好的分割模型,并对分割得到的河豚鱼图像进行对齐;训练特征提取网络,将对齐后的河豚鱼图像划分成训练集和验证集,在训练集上训练特征提取网络,得可用于提取河豚鱼特征的特征提取模型;将用于测试的图片放入训练好的分割模型,分割结果对齐后送入训练好的特征提取模型得到输入图片的特征向量,计算该特征向量与数据集中的河豚鱼特征向量的距离,获得输入图片中河豚鱼个体信息。
公开/授权文献
0/0