Detection Method for Individual Pig Based on Improved YOLOv4 Convolutional Neural Network
Shudong Li,Xi Kang,Yankun Feng,Gang Liu
DOI: https://doi.org/10.1145/3478905.3478951
2021-01-01
Abstract:Aiming at the problem of low non-invasive intelligent identification of pigs in healthy breeding, a pig individual identification method based on the improved YOLOv4 convolutional neural network is proposed. By preprocessing the images of 30 live pigs in the collected video, a method based on the whole live pig as the feature extraction sample is proposed, instead of using the pig face as the feature extraction sample, to make a data set to obtain the effective features of the whole body. Considering the living habits of live pigs and the living environment of live pigs, the improved YOLOv4 model is proposed, which adapted certain data enhancement and attention mechanisms, and conducts multi-scale training and detection on the data set to achieve accurate detection and identification of individual live pigs; obtains the best parameters according to the training log, and carries out verification and comparison experiments for individual pig identification. The result shows that when the threshold is 0.5, the average accuracy (mAP) on the test set reaches 98.12%. The recall rate reached 95%, the F1-score was 96%, the average recognition time Mean FPS was 34.3ms, and the average crossover ratio (IoU) was 83.91%. The improved model comparison between YOLOv3 and Fast R-CNN has improved the recognition accuracy and speed of individual pigs; at the same time, compared with the traditional pig face data set, the pig body data set has significantly improved the recognition accuracy. Based on the whole live pig as a feature extraction sample combined with the improved YOLOv4 model, the live pig can be recognized quickly and effectively in various postures and complex backgrounds. As a multi-angle and multi-target recognition method, it can quickly and accurately identify individual pigs in a large-scale group healthy breeding environment as well as realization of individualized healthy breeding and efficient management for pig.
What problem does this paper attempt to address?