Skip to content

A pipeline for processing wildlife crossing datasets using Kalman filter-based tracking, segmentation masks, and feature embeddings. Includes data loading, mask prediction, feature extraction, and trajectory prediction.

License

Notifications You must be signed in to change notification settings

jnbreid/animal_tracking

Repository files navigation

Animal Tracker

Overview

This project provides a pipeline for tracking and segmenting animals in video footage.

It uses pre-trained models and basic models to facilitate the automatic analysis of animal videos. This pipeline is specifically designed for tracking in camera trap videos. The pipeline requires minimal training and does not require data annotated with segmentation masks for training. This enables automated analysis of camera trap video in domains where little training data is available.

Example videos

Example video 1 Example video 2
Example 1 Example 2

The animals are annotated by applying the pipeline. The raw videos used as input for the pipeline can be found here.

The videos are part of the GMOT-40 Benchmark. The video data is licensed under a Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License. Further information about the data can be found here under point 7.

Demo

The notebook demo_inference.ipynb can be used to track animals in two example videos.

Dependencies

To install all required packages run

pip install -r requirements.txt

License

This project is licensed under a GPL-3.0 license.

About

A pipeline for processing wildlife crossing datasets using Kalman filter-based tracking, segmentation masks, and feature embeddings. Includes data loading, mask prediction, feature extraction, and trajectory prediction.

Topics

Resources

License

Stars

Watchers

Forks