Automatic capture and interception algorithm for cattle face images based on improved YOLOv3 in small-sample scenarios
摘要
With the increasing demand for intelligent development in animal husbandry, traditional manual creation of cattle face datasets faces problems such as low efficiency and large errors. This paper proposes an automatic cattle face image capture and cropping algorithm based on improved YOLOv3 in small-sample scenarios, aiming to achieve automated construction of cattle face datasets. Due to the low-latency requirements for subsequent transplantation into embedded systems, the lightweight MobileNet is first used as the feature extraction network to improve detection speed, and the K-means + + algorithm is adopted to re-cluster anchor boxes, thereby optimizing target scale adaptability. Meanwhile, data augmentation and transfer learning strategies are introduced to address overfitting issues caused by small-sample training. Ablation experiments based on a dataset of 7286 images from 48 cattle have shown that the improved model achieves an accuracy of 99.86% and a detection time of only 39.24 min, with a 9% increase in accuracy and a 50% reduction in detection time compared to the original YOLOv3 model, and a detection efficiency 27 times higher than that of manual cropping. This algorithm provides an efficient data acquisition solution for rapid modeling of cattle face recognition systems and can be migrated to intelligent management scenarios of other livestock species.