基于CycleGAN和注意力增强迁移学习的小样本鱼类识别
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

青岛海洋科技中心山东省专项经费项目(2022QNLM030001-2)和中央级公益性科研院所基本科研业务费专项资金项目(2022XT06)


Recognition of Small Sample Cultured Fish Based on CycleGAN and Attention Enhanced Transfer Learning
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    围绕水产养殖水下目标精准识别的产业发展需求,针对小样本目标识别精度低、模型算法场景适应能力差等问题,提出一种基于改进循环对抗网络(Cycle constraint adversarial network, CycleGAN)样本扩增和注意力增强迁移学习的小样本养殖鱼类识别方法。利用水下采样装备收集实际养殖场景和可控养殖场景大黄鱼图像,并以可控场景图像作为辅助样本集。利用CycleGAN为基础框架实现辅助样本到实际养殖场景图像的迁移,并提出一种基于最大平均差异(Maximum mean discrepancy, MMD)的迁移模型损失函数优化方法。在迁移学习阶段使用ResNet50为基础框架,并引入SK-Net(Selective kernel network)注意力机制优化模型对不同感受野目标的感知能力,提升模型对无约束鱼类目标的识别精度。试验结果表明,本文方法有效提升了小样本鱼类目标的识别能力,鱼类识别召回率达到94.33%,平均精度均值达到96.67%,为鱼类行为跟踪和表型测量提供了有效的技术支撑。

    Abstract:

    Focusing on the industrial development needs of accurate underwater target recognition in aquaculture, and aiming at the problems of low target recognition accuracy of small samples and poor adaptability of model algorithm to scenarios, a small sample aquaculture fish recognition method based on improved cycle constraint adversarial network (CycleGAN) sample amplification and attention enhancement transfer learning was proposed. Firstly, the underwater sampling equipment was used to collect the images of the actual and controllable breeding scenes of Larimichthys crocea, and the controllable scene images were used as the auxiliary sample set. CycleGAN was used as the basic framework to realize the migration of auxiliary samples to the actual breeding scene images. In particular, an optimization method of the loss function of the migration model based on the maximum mean discrepancy (MMD) was proposed. Then in the transfer learning phase, ResNet50 was used as the basic framework, and SK-Net (selective kernel network) attention mechanism optimization model was introduced to improve the perception ability of different receptive field targets, so as to improve the recognition accuracy of the model for unconstrained fish targets. The experimental results showed that the method proposed effectively improved the recognition ability of fish small sample targets, with a recall rate of 94.33% of fish recognition, and an mAP of 96.67%, providing effective technical support for the next step of fish behavior tracking and phenotype measurement.

    参考文献
    相似文献
    引证文献
引用本文

刘世晶,刘阳春,钱程,郑浩君,周捷,张成林.基于CycleGAN和注意力增强迁移学习的小样本鱼类识别[J].农业机械学报,2023,54(s1):296-302. LIU Shijing, LIU Yangchun, QIAN Cheng, ZHENG Haojun, ZHOU Jie, ZHANG Chenglin. Recognition of Small Sample Cultured Fish Based on CycleGAN and Attention Enhanced Transfer Learning[J]. Transactions of the Chinese Society for Agricultural Machinery,2023,54(s1):296-302.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2023-06-30
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2023-12-10
  • 出版日期: