外文摘要：With the advancement of machine vision technology, pig face recognition has garnered significant attention as a key component in the establishment of precision breeding models. In order to explore non-contact individual pig recognition, this study proposes a lightweight pig face feature learning method based on attention mechanism and two-stage transfer learning. Using a combined approach of online and offline data augmentation, both the self-collected dataset from Shanxi Agricultural University's grazing station and public datasets underwent enhancements in terms of quantity and quality. YOLOv8 was employed for feature extraction and fusion of pig face images. The Coordinate Attention (CA) module was integrated into the YOLOv8 model to enhance the extraction of critical pig face features. Fine-tuning of the feature network was conducted to establish a pig face feature learning model based on two-stage transfer learning. The YOLOv8 model achieved a mean average precision (mAP) of 97.73% for pig face feature learning, surpassing lightweight models such as EfficientDet, SDD, YOLOv5, YOLOv7-tiny, and swin_transformer by 0.32, 1.23, 1.56, 0.43 and 0.14 percentage points, respectively. The YOLOv8-CA model's mAP reached 98.03%, a 0.3 percentage point improvement from before its addition. Furthermore, the mAP of the two-stage transfer learning-based pig face feature learning model was 95.73%, exceeding the backbone network and pre-trained weight models by 10.92 and 3.13 percentage points, respectively. The lightweight pig face feature learning method, based on attention mechanism and two-stage transfer learning, effectively captures unique pig features. This approach serves as a valuable reference for achieving non-contact individual pig recognition in precision breeding.
外文关键词：deep learning;Transfer learning;lightweight;attention mechanisms;pig face feature learning
作者：Li, Yaoyu;Li, Fuzhong;Zhang, Wuping;Yin, Zhe;Peng, Mingkang;Guo, Zhaodong;Zhao, Yue;Guo, Xiaohong
作者单位：Shanxi Agr Univ
期刊名称：AGRICULTURE-BASEL
期刊影响因子：0.0
出版年份：2024
出版刊次：14（1）
原文传递申请：江苏省科技资源（工程技术文献）统筹服务平台