Pytorch-Wildlife: A Collaborative Deep Learning Framework for Conservation

Andres Hernandez,Zhongqi Miao,Luisa Vargas,Rahul Dodhia,Pablo Arbelaez,Juan M. Lavista Ferres
2024-07-02
Abstract:The alarming decline in global biodiversity, driven by various factors, underscores the urgent need for large-scale wildlife monitoring. In response, scientists have turned to automated deep learning methods for data processing in wildlife monitoring. However, applying these advanced methods in real-world scenarios is challenging due to their complexity and the need for specialized knowledge, primarily because of technical challenges and interdisciplinary barriers. To address these challenges, we introduce Pytorch-Wildlife, an open-source deep learning platform built on PyTorch. It is designed for creating, modifying, and sharing powerful AI models. This platform emphasizes usability and accessibility, making it accessible to individuals with limited or no technical background. It also offers a modular codebase to simplify feature expansion and further development. Pytorch-Wildlife offers an intuitive, user-friendly interface, accessible through local installation or Hugging Face, for animal detection and classification in images and videos. As two real-world applications, Pytorch-Wildlife has been utilized to train animal classification models for species recognition in the Amazon Rainforest and for invasive opossum recognition in the Galapagos Islands. The Opossum model achieves 98% accuracy, and the Amazon model has 92% recognition accuracy for 36 animals in 90% of the data. As Pytorch-Wildlife evolves, we aim to integrate more conservation tasks, addressing various environmental challenges. Pytorch-Wildlife is available at <a class="link-external link-https" href="https://github.com/microsoft/CameraTraps" rel="external noopener nofollow">this https URL</a>.
Computer Vision and Pattern Recognition,Machine Learning
What problem does this paper attempt to address?
The paper aims to address the challenges faced in wildlife monitoring against the backdrop of a sharp decline in global biodiversity. Specifically, the paper introduces Pytorch-Wildlife, an open-source deep learning platform built on PyTorch, designed specifically for conservation projects. The platform is designed with a focus on ease of use, scalability, and transparency, enabling researchers with limited or no technical background to easily use advanced AI models for animal detection and classification. The main contributions of Pytorch-Wildlife include: 1. **Ease of Use**: Simple installation via pip, compatible with multiple operating systems, and provides a user interface and detailed documentation support. 2. **Scalability**: Adopts a modular code structure, making it easy to add new features and customize data processing options. 3. **Transparency**: Fully open-source, allowing the community to contribute new models, and provides model architectures and pre-trained weights to ensure reproducibility. 4. **Practical Application**: Demonstrates its efficient performance in two case studies: species identification in the Amazon rainforest and opossum detection in the Galápagos Islands. Through these features, Pytorch-Wildlife aims to become a unified AI platform supporting a wide range of conservation tasks, thereby promoting ecological conservation efforts in different environments and research fields.