IDEAS home Printed from https://ideas.repec.org/a/gam/jagris/v13y2023i10p1938-d1253476.html
   My bibliography  Save this article

Real-Time Cattle Pose Estimation Based on Improved RTMPose

Author

Listed:
  • Xiaowu Li

    (Faculty of Information Engineering and Automation, Kunming University of Science and Technology, Kunming 650500, China)

  • Kun Sun

    (Faculty of Information Engineering and Automation, Kunming University of Science and Technology, Kunming 650500, China)

  • Hongbo Fan

    (Faculty of Modern Agricultural Engineering, Kunming University of Science and Technology, Kunming 650300, China)

  • Zihan He

    (Faculty of Information Engineering and Automation, Kunming University of Science and Technology, Kunming 650500, China)

Abstract

Accurate cattle pose estimation is essential for Precision Livestock Farming (PLF). Computer vision-based, non-contact cattle pose estimation technology can be applied for behaviour recognition and lameness detection. Existing methods still face challenges in achieving fast cattle pose estimation in complex scenarios. In this work, we introduce the FasterNest Block and Depth Block to enhance the performance of cattle pose estimation based on the RTMPose model. First, the accuracy of cattle pose estimation relies on the capture of high-level image features. The FasterNest Block, with its three-branch structure, effectively utilizes high-level feature map information, significantly improving accuracy without a significant decrease in inference speed. Second, large kernel convolutions can increase the computation cost of the model. Therefore, the Depth Block adopts a method based on depthwise separable convolutions to replace large kernel convolutions. This addresses the insensitivity to semantic information while reducing the model’s parameter. Additionally, the SimAM module enhances the model’s spatial learning capabilities without introducing extra parameters. We conducted tests on various datasets, including our collected complex scene dataset (cattle dataset) and the AP-10K public dataset. The results demonstrate that our model achieves the best average accuracy with the lowest model parameters and computational requirements, achieving 82.9% on the cattle test set and 72.0% on the AP-10K test set. Furthermore, in conjunction with the object detection model RTMDet-m, our model reaches a remarkable inference speed of 39FPS on an NVIDIA GTX 2080Ti GPU using the PyTorch framework, making it the fastest among all models. This work provides adequate technical support for fast and accurate cattle pose estimation in complex farm environments.

Suggested Citation

  • Xiaowu Li & Kun Sun & Hongbo Fan & Zihan He, 2023. "Real-Time Cattle Pose Estimation Based on Improved RTMPose," Agriculture, MDPI, vol. 13(10), pages 1-18, October.
  • Handle: RePEc:gam:jagris:v:13:y:2023:i:10:p:1938-:d:1253476
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2077-0472/13/10/1938/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2077-0472/13/10/1938/
    Download Restriction: no
    ---><---

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jagris:v:13:y:2023:i:10:p:1938-:d:1253476. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.