摘要
针对传统鱼类识别方法存在特征提取复杂、算法可移植性差等不足,该研究提出了一种基于改进ResNet50模型的淡水鱼种类识别方法。研究以鳙鱼、鳊鱼、鲤鱼、鲫鱼、草鱼、白鲢6种大宗淡水鱼为对象,通过搭建淡水鱼图像采集系统获取具有单一背景的淡水鱼图像,同时通过互联网搜索具有干扰背景的淡水鱼图像,共同构建淡水鱼图像数据集;再对淡水鱼图像进行预处理,以增加样本多样性;构建改进ResNet50模型,增加全连接层Fc1以及Dropout,引入迁移学习机制训练模型,同时选择CELU作为激活函数提高神经网络表达能力,通过Adam优化算法更新梯度,并嵌入余弦退火方法衰减学习率。为验证改进ResNet50模型的准确率等性能,对6种淡水鱼进行种类识别,结果表明:在单次验证方法下,选用包含单一背景图像和干扰背景图像构成的淡水鱼图像数据集训练模型,识别准确率为96.94%,比经典模型提高1.22%,单张淡水鱼图像样本的平均检测时间为0.2345 s;在四折交叉验证下,选用具有单一背景的图像数据集,模型的识别准确率为100%,选用包含单一背景图像和干扰背景图像的淡水鱼图像数据集,模型的识别准确率为96.20%,说明模型具有较好的泛化性能和鲁棒性。针对混淆矩阵的可视化结果表明:改进的ResNet50模型具有通用的结构和训练方式,对不同背景下的淡水鱼进行种类识别具有较高的准确率,可为淡水鱼种类识别提供技术借鉴。
Species identification of freshwater fish has a wide range of applications in most fields,such as breeding,fishing,and processing.However,most traditional algorithms of fish identification cannot meet the ever-increasingly high requirements in recent years,such as simple feature extraction,high accuracy,and portability.In this study,a new identification algorithm was proposed for the freshwater fish species using an improved ResNet50 model.Six types of freshwater fish were taken as the research objects,including the bighead,bream,carp,crucian,grass carp,and silver carp.An image acquisition system was established for the freshwater fish images with a single background.As such,an image dataset of freshwater fish was constructed to joint those images with interference background on the Internet.A Pytorch framework was then selected to perform image preprocessing of freshwater fish for the sample diversity.An improved ResNet50 model was thus built to identify the freshwater fish species.Firstly,the fully connected layer Fc1 and Dropout were added,while the migration learning was introduced to train the model.Secondly,CELU was selected as the activation function to improve the expression of the neural network.Finally,Adam optimization was used to update the gradient.A cosine annealing was also embedded to attenuate the learning rate.In addition,the hyperparameters of the model were optimized in the multiple model training.Correspondingly,six kinds of freshwater fish were identified to verify the accuracy and performance of the improved ResNet50 model.A single validation test under a four-fold cross-validation model was carried out to train and evaluate the model.The confusion matrix was used to visualize the recognition of each type of fish.The results showed that:the image dataset of freshwater fish consisting of a single and interference background images was selected to train the model under the single validation,where the accuracy rate was 96.94%,1.22%higher than before.The average detection time was 0.2345s for a
作者
万鹏
赵竣威
朱明
谭鹤群
邓志勇
黄毓毅
吴文锦
丁安子
Wan Peng;Zhao Junwei;Zhu Ming;Tan Hequn;Deng Zhiyong;Huang Yuyi;Wu Wenjin;Ding Anzi(College of Engineering,Huazhong Agricultural University,Wuhan 430070,China;Key Laboratory of Agricultural Equipment in Mid-lower Yangtze River,Ministry of Agriculture and Rural Affairs,Wuhan 430070,China;Research Institute of Agricultural Products Processing and Nuclear-agricultural Technology,Hubei Academy of Agricultural Sciences,Wuhan 430070,China)
出处
《农业工程学报》
EI
CAS
CSCD
北大核心
2021年第12期159-168,共10页
Transactions of the Chinese Society of Agricultural Engineering
基金
国家重点研发计划项目(2018YFD0700903-2)
湖北省农业科技创新中心2020年重大科技研发专项(2020-620-000-002-03)
湖北省农业科技创新中心创新团队项目(2016620000001044)
中央高校基本科研业务费专项基金资助项目(107-11041910103)。
关键词
图像识别
水产养殖
淡水鱼
种类识别
深度学习
改进ResNet50模型
超参数优化
可视化
image recognition
aquaculture
freshwater fish
species recognition
deep learning
improved ResNet50 model
hyperparameter optimization
visualization
