Multi-target detection based on feature pyramid attention and deep convolution network for pigs

被引:0
|
作者
Yan H. [1 ]
Liu Z. [1 ]
Cui Q. [2 ]
Hu Z. [1 ]
机构
[1] College of Information Science and Engineering, Shanxi Agricultural University, Taigu
[2] College of Engineering, Shanxi Agricultural University, Taigu
关键词
Algorithm; Feature pyramid attention(FPA); Image processing; Target detection; Tiny-YOLO;
D O I
10.11975/j.issn.1002-6819.2020.11.022
中图分类号
学科分类号
摘要
In the pig breeding environment, pig adhesion and debris shielding made it difficult to detect multiple targets of pig individuals. In this paper, pigs in the pen were used as the research object, and the video frame was used as the data source to propose a model FPA-Tiny-YOLO that combines Feature Pyramid Attention (FPA) and Tiny-YOLO. In this model, attention information was integrated into feature extraction, semantic content of different regions was aggregated hierarchically, and global context information was mined. Image processing (include randomly change the brightness, adding Gaussian noise and flipping 180°) was performed on video clips of 45 pigs in 8 bars with age of 20 to 105 days, and 4 102 labeled pictures were obtained. Four kinds of deep FPA modules were constructed and added to YOLOV3 and Tiny-YOLO models. Experiments show that adding a variety of feature pyramid attention information could improve the accuracy of Tiny-YOLO and YOLOV3 models to some extent. Compared with the YOLOV3 models, before adding the FPA module, the Tiny-YOLO model had higher detection accuracy, and the detection real-time performance was better than that of the YOLOV3 model with the same module added. After adding the FPA module, the detection performance of the Tiny-YOLO model was better than that of the YOLOV3 models, the mAP, Precision ratio, Recall rate and F1 value of the Tiny-YOLO model with the FPA-3 module increased by 8.4, 1.04, 7.93, and 5.09 percentage points compared to the YOLOV3 model with the same FPA module. The Recall rate, F1 value and mAP of Tiny-YOLO model with FPA-3 module for multi-target detection of group pigs on the test set reached 86.09%, 91.47% and 85.85% respectively, which improved by 3.75, 2.59 and 4.11 percentage points respectively compared with Tiny-YOLO model without FPA module, The the Recall rate, F1 value and mAP decrease as the score threshold increases when the score and IOU value fixed, but the Precision ratio gradually decreases, whicb indicating that the depth of the FPA module had no regular effects on the detecting effect. The test set pictures were divided into four scenes according to whether adhesion or shielding to explore the robustness of the Tiny-YOLO series models. The experiments showed that compared with Tiny-YOLO model, the Recall rate, F1 value and mAP of the Tiny-YOLO model adding FPA-3 module were improved by 6.73, 4.34 and 7.33 percentage points respectively, the Tiny-YOLO model adding FPA module could extract more complete edges of pigs and had higher prediction reliability and had better detection effects for the pigs far away from the camera and with occlusion, the feature pyramid attention information was beneficial to precisely and effectively conduct multi-target detection of live pigs in different scenes. The research results can provide a reference for the subsequent mobile application of pig identification and behavior analysis. © 2020, Editorial Department of the Transactions of the Chinese Society of Agricultural Engineering. All right reserved.
引用
收藏
页码:193 / 202
页数:9
相关论文
共 42 条
  • [1] Zou Yuanbing, Sun Longqing, Li Yue, Et al., Video monitoring and analysis system for pig breeding based on distributed flow computing, Transactions of the Chinese Society for Agricultural Machinery, 48, S1, pp. 365-373, (2017)
  • [2] Marchant J A, Schofield C P, White R P., Pig growth and conformation monitoring using image analysis, Animal Science, 68, 1, pp. 141-150, (2016)
  • [3] Krizhevsky A, Sutskever I, Hinton G E., Imagenet classification with deep convolutional neural networks, Advances in Neural Information Processing Systems, pp. 1097-1105, (2012)
  • [4] He Kaiming, Zhang Xiangyu, Ren Shaoqin, Et al., Deep residual learning for image recognition, Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition(CVPR), pp. 770-778, (2016)
  • [5] Psota E, Mittek M, Perez L, Et al., Multi-pig part detection and association with a fully-convolutional network, Sensors, 19, 4, pp. 852-876, (2019)
  • [6] Gao Yun, Guo Jiliang, Li Xuan, Et al., Instance-level segmentation method for group pig images based on deep learning, Transactions of the Chinese Society for Agricultural Machinery, 50, 4, pp. 186-194, (2019)
  • [7] Hu Zhiwei, Yang Hua, Lou Tiantian, Et al., Extraction of pig contour based on fully convolutional networks, Journal of South China Agricultural University, 39, 6, pp. 111-119, (2018)
  • [8] Yang Aqing, Xue Yueju, Huang Huasheng, Et al., Lactating sow image segmentation based on fully convolutional networks, Transactions of the Chinese Society of Agricultural Engineering (Transactions of the CSAE), 33, 23, pp. 219-225, (2017)
  • [9] Gao Yun, Chen Bin, Liao Huimin, Et al., Recognition method for aggressive behavior of group pigs based on deep learning, Transactions of the Chinese Society of Agricultural Engineering (Transactions of the CSAE), 35, 23, pp. 192-200, (2019)
  • [10] Li Dan, Zhang Kaifeng, Li Hangjian, Et al., Multi-target pigs detection algorithm based on improved CNN, Transactions of the Chinese Society for Agricultural Machinery, 50, S1, pp. 261-266, (2019)