PriMAT: A robust multi-animal tracking model for primates in the wild
Richard Vogg,Matthias Nuske,Marissa A Weis,Timo Luddecke,Elif Karakoc,Zurna Ahmed,Sofia Pereira,Suchinda Malaivijitnond,Suthirote Meesawat,Florentin Worgotter,Peter M Kappeler,Alexander Gail,Julia Ostner,Oliver Schulke,Claudia Fichtel,Alexander S Ecker
DOI: https://doi.org/10.1101/2024.08.21.607881
2024-08-23
Abstract:Detection and tracking of animals is an important first step for automated behavioral studies on videos. Nowadays, animal tracking is mostly done using deep learning frameworks based on keypoints, which show remarkable results in lab settings with fixed cameras, backgrounds and lighting. In contrast, multi-animal tracking in the wild is currently underexplored as it presents numerous challenges such as high background variability, lighting conditions, complex motion, and occlusion. We propose a multi-animal tracking model, PriMAT, for non-human primates in the wild. The model learns to detect and track primates and other objects of interest from labeled videos or single images using bounding boxes instead of keypoints. Using bounding boxes facilitates data annotation significantly and allows for more robust results. Our one-stage model is conceptually simple but highly flexible, and we add a classification branch that allows us to train individual identification. To evaluate the performance of our model, we apply it in two case studies with Assamese macaques and redfronted lemurs in the wild. We show that with only a few hundred frames labeled with bounding boxes, we can achieve robust tracking results. Combining these results with the classification branch, our model shows an accuracy of 83% when predicting lemur identities. Our approach presents a promising solution for accurately tracking and identifying animals in the wild using deep learning based bounding box tracking, offering researchers a tool to study animal behavior in their natural habitats. Our code, models, training images, and evaluation video sequences are publicly available, facilitating its use for animal behavior analyses and future research in this field.
Animal Behavior and Cognition