您好，欢迎访问中国水产科学研究院 机构知识库！
中国水产科学研究院机构知识库

全部

  • 全部
  • 标题
  • 学者
  • 机构
  • 关键词

YOLO-FD: An accurate fish disease detection method based on multi-task learning

文献类型： 外文期刊

作者： Li, Xuefei 1 ; Zhao, Shili 1 ; Chen, Chunlin 1 ; Cui, Hongwu 5 ; Li, Daoliang 1 ; Zhao, Ran 1 ;

作者机构： 1.China Agr Univ, Natl Innovat Ctr Digital Fishery, Beijing 100083, Peoples R China

2.China Agr Univ, Key Lab Smart Farming Aquat Anim & Livestock, Minist Agr & Rural Affairs, Beijing 100091, Peoples R China

3.China Agr Univ, Beijing Engn Technol Res Ctr Internet Things Agr, Beijing 100083, Peoples R China

4.China Agr Univ, Coll Informat & Elect Engn, Beijing 100083, Peoples R China

5.Chinese Acad Fishery Sci, Yellow Sea Fisheries Res Inst, Key Lab Sustainable Dev Marine Fisheries, Minist Agr & Rural Affairs, Qingdao 266071, Peoples R China

关键词： Fish disease detection; Object detection; Semantic segmentation; Multi-task learning

期刊名称：EXPERT SYSTEMS WITH APPLICATIONS （ 影响因子：7.5； 五年影响因子：7.8 ）

ISSN： 0957-4174

年卷期： 2024 年 258 卷

页码：

收录情况： SCI

摘要： Fish diseases often exhibit high risks of contagion, resulting in substantial economic losses. Accurate assessment of fish disease severity during diagnosis using deep learning poses a considerable challenge. Currently, deep learning models mainly focus on single tasks in fish disease detection, such as classification, object detection and segmentation. However, the accurate assessment of fish disease severity requires the integration of multiple dimensions of information, which is beyond the capabilities of traditional single-task methods. Therefore, this paper proposes YOLO-FD, a multi-task learning network specifically designed for simultaneous detection and segmentation. YOLO-FD extends the YOLOv8 backbone by integrating a novel semantic segmentation branch dedicated to precisely segmenting infected areas in diseased fish, while retaining the original object detection branch for identifying infected fish. Weight uncertainty and PCGrad are employed to balance the weights of different losses and to optimize conflicting gradients during the training process. With only a negligible increase in network parameters, YOLO-FD, tested on our constructed Nocardiosis fish dataset, achieves a detection accuracy of 94.2% mAP50 and gets mIOU of 79.4%, showcasing a 0.5% improvement over the baseline YOLOv8 and surpassing the state-of-the-art semantic segmentation network Deeplabv3plus by 4%. Notably, compared to the adapted multi-task network YOLOP, YOLO-FD demonstrates substantial improvements, displaying a 13.7% increase in mAP50-95 and a 15.1% boost in mIOU. On the VOC2012 segmentation dataset, the proposed method exhibits a 3.2% increase in mAP50 and a 2.2% rise in mAP50-95 compared to the baseline. Furthermore, results of the ablation experiment validate the effectiveness and generalization of the proposed multi-task learning approach. Source code is available at https://github.com/feifei-Lee/YOLO-FD.

  • 相关文献
作者其他论文 更多>>
置顶
购物车
反馈

© 京ICP备09089781号-29 主办单位：中国水产科学研究院

学术资源： 中国农业科学院 农业专业知识服务系统 农科联盟资源共建共享平台 中国农业科技文献与信息服务平台 中国农业期刊集成服务平台