YOLO-animal: an animal detection network based on improved YOLOv5 and attention mechanism

Si Li

doi:10.1117/12.3015658

29 January 2024 YOLO-animal: an animal detection network based on improved YOLOv5 and attention mechanism

Si Li

Author Affiliations +

Proceedings Volume 12984, Fourth International Conference on Computer Vision and Information Technology (CVIT 2023); 1298402 (2024) https://doi.org/10.1117/12.3015658
Event: 2023 4th International Conference on Computer Vision and Information Technology (CVIT 2023), 2023, Beijing, China

Abstract

Animal detection and recognition is a crucial task in computer vision. YOLOv5 has been widely used for animal identification in the past few years. However, it is still a challenging task due to the diverse array of animal types found in complex environments. In this paper, we introduce a new attention mechanism based on the CBAM attention mechanism to enhance the performance of the network model. Specifically, the attention mechanism enhances the interplay between globally pooled channel information, thereby bolstering the ability to detect and recognize animals with similar features in complex backgrounds. Experimental results on the Oxford-IIIT Pet validation dataset demonstrate the effectiveness of the proposed model's robustness and its ability to perform effectively in real-world scenarios.

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Si Li "YOLO-animal: an animal detection network based on improved YOLOv5 and attention mechanism", Proc. SPIE 12984, Fourth International Conference on Computer Vision and Information Technology (CVIT 2023), 1298402 (29 January 2024); https://doi.org/10.1117/12.3015658

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

;

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE