PLM-YOLOv5: Improved YOLOv5 for Pig Detection in Livestock Monitoring
With the rapid development of smart agriculture, the demand for pig monitoring has been increasing, particularly in disease prevention and behavioral analysis. Real-time monitoring of pig behavior is essential for improving farming efficiency and ensuring animal welfare. However, challenges such as pigs’ varying body sizes, movement, and complex environments make accurate detection and tracking difficult. To address these issues, this paper proposes an improved YOLOv5 model, named PLM-YOLOv5, specifically designed for pig detection in livestock monitoring. First, we add an additional detection head to enhance the model’s capability to detect smaller pigs effectively. Next, we integrate the Convolutional Block Attention Module (CBAM) to enhance feature extraction, particularly in handling complex backgrounds. Finally, we utilize a self-collected dataset of over 1,400 images for model training and validation, ensuring the model adapts to real-world livestock environments. Experimental results show that PLM-YOLOv5 achieves significant improvements in pig detection tasks, with the mean Average Precision (mAP) increasing by approximately 9% compared to the original YOLOv5. Additionally, training and validation loss curves demonstrate the model's stability and convergence. These results confirm the effectiveness and reliability of PLM-YOLOv5 for real-world applications, offering a robust solution for smart agriculture.
Authors
- Li Fang (ORCID: https://orcid.org/0009-0009-5342-7036)
- Xiangyang Luo (ORCID: https://orcid.org/0000-0003-3225-4649)
- Ni Peng
Institutions
- Guilin University of Electronic Technology (CN)
Publication Details
- Published
- 2024-11-29
- DOI
- https://doi.org/10.1109/irac63143.2024.10871878
- Citations
- 1
- Primary Topic
- Industrial Vision Systems and Defect Detection
- Type
- article
- Field-Weighted Citation Impact
- 0.35
Funders
- National Natural Science Foundation of China
- Research and Development