Repository files navigation

main

dev

Documentation:https://trex.run/docsYouTube:https://youtube.com/@TRexTracker

Welcome to the git repository ofTRex (https://trex.run) -- a software designed to track and identify individuals and other moving entities using computer vision and machine learning.

TRaditionally, using background-subtraction, TRex can track 256 fish faster than the video plays back, and for up to 100 individuals allows you to visually recognize individuals and automatically correct potential tracking errors using machine-learning based automatic classification.

Since version 2.0, TRex also supports machine-learning based detection models. Out of the box we support models compatible with theultralytics (https://github.com/ultralytics) package, like YOLO. All common tasks, such as bounding-boxes, oriented bounding-boxes, keypoints, and segmentation as well as offering simple SAHI-like features (https://github.com/obss/sahi). This replaces background subtraction in situations where it is not sufficient - you can bring your own.pt model, of course!

It allows you, from the same interface, to load up existing videos or record directly from your webcam. Camera support (and other things) can be extended for other APIs with a bit of C++ or Python knowledge, of course.

TRex supports all major platforms. Miniforge is the main package repository used for installing/building TRex. Simply follow the instructions here for installing miniforge:https://github.com/conda-forge/miniforge#download.You can then create a new virtual environment (namedtracking here) by running:

Windows, Linux (NVIDIA graphics card recommended), macOS (Intel / ARM64)

conda create -n tracking -c trexing trex

Within the conda environment, simply run:

trex

To convert a video to our custom pv format from the command-line:

trex -i /full/path/to/video.mp4 -o funny_name

This will output to/full/path/to/video.pv. If you want to change the output destination root, add-d /output/path.

Opening a video directly and adjustingparameters:

trex -i /full/path/to/video.mp4 -track_threshold 25 -track_max_individuals 10

If you don't want a graphical user interface and save/quit when tracking finishes, add-nowindow and-auto_quit:

trex -i /full/path/to/video.mp4 -nowindow -auto_quit

Readmore about parameters for TRexhere.

This project has been developed, is still being updated, byTristan Walter.If you want to contribute, please submit a pull request on GitHub and I will be happy to credit you here, for any substantial contributions!

If you have any issues running the software please consult the documentation first (especially the FAQ section)and if this does not solve your problem, please file an issue using theissue tracker here on github.If you experience problems withPyTorch, such as installing CUDA or cuDNN dependencies, then please direct issues to those development teams.

Significant contributions by:Angela Albi

Released under the GPLv3 License (seeLICENSE).

If you use this software in your work, please cite ouropen-access paper:

@article {walter2020trex,  article_type = {journal},  title = {TRex, a fast multi-animal tracking system with markerless identification, and 2D estimation of posture and visual fields},  author = {Walter, Tristan and Couzin, Iain D},  editor = {Lentink, David},  volume = 10,  year = 2021,  month = {feb},  pub_date = {2021-02-26},  pages = {e64000},  citation = {eLife 2021;10:e64000},  doi = {10.7554/eLife.64000},  url = {https://doi.org/10.7554/eLife.64000},  journal = {eLife},  issn = {2050-084X},  publisher = {eLife Sciences Publications, Ltd},}