Code Monkey home page Code Monkey logo

nolanzzz / mtmct Goto Github PK

View Code? Open in Web Editor NEW
75.0 3.0 14.0 357.9 MB

Design and Implementation of a Multi-Target Multi-Camera Tracking Solution

Home Page: https://wiser.cas.mcmaster.ca/?docs=interactive-perception-on-edge-devices#6-toc-title

License: MIT License

Python 71.17% Jupyter Notebook 21.77% Dockerfile 0.09% Makefile 0.02% Batchfile 0.02% C++ 2.50% Cuda 3.51% Shell 0.49% MATLAB 0.05% C 0.21% Cython 0.19%
python tracking resnet detection reidentification pytorch cuda opencv research-project machine-learning deep-learning

mtmct's Introduction

English | 简体中文

MTMCT

This project demonstrates the design and implementation of a Multi-Target Multi-Camera Tracking (MTMCT) solution.

Pipeline of our solution:

Tracking performance

Results and comparisons with FairMOT and wda_tracker trained and tested on a 6x2-minute MTA dataset

Method Single-Camera Multi-Camera
MOTA IDF1 IDs MT ML MOTA IDF1 IDs MT ML
WDA 58.2 37.3 534.2 16.8% 17.2 46.6 19.8 563.8 6.5% 7.0%
FairMOT 64.1 48.0 588.2 34.7% 7.8% N/A N/A N/A N/A N/A
Ours 70.8 47.8 470.2 40.5% 5.6% 65.6 31.5 494.5 31.2% 1.1%

Demo GIFs can be seen here

Full-length demo videos can be found at: https://youtu.be/lS9YvbrhOdo

Installation

conda create -n mtmct python=3.7.7 -y
conda activate mtmct
pip install -r requirements.txt

Install dependencies for FairMOT:

cd trackers/fair
conda install pytorch==1.7.0 torchvision==0.8.0 cudatoolkit=10.2 -c pytorch
pip install cython
pip install -r requirements.txt
cd DCNv2
./make.sh
conda install -c conda-forge ffmpeg

Download data

Go to https://github.com/schuar-iosb/mta-dataset to download the MTA data. Or use other datasets that match the same format.

Configurations

Modify config files under tracker_configs and clustering_configs for customization. Create a work_dirs and see more instructions at FairMOT and wda_tracker.

E.g. in configs/tracker_configs/fair_high_30e set the data -> source -> base_folder to your dataset location.

Tracking

Run single and the multi-camera tracking with one script:

sh start.sh fair_high_30e

Modify config files under tracker_configs and clustering_configs for customization. More instructions can be found at FairMOT and wda_tracker.

Acknowledgement

A large part of the code is borrowed from FairMOT and wda_tracker. The dataset used is MTA

Copyright

Ruizhe Zhang is the author of this repository and the corresponding report, the copyright belongs to Wireless System Research Group (WiSeR), McMaster University.

mtmct's People

Contributors

nolanzzz avatar zhanr110 avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar

mtmct's Issues

running "multicam_trackwise_evaluation.py" gives me bellow error. Kindly help me the fix it

Traceback (most recent call last):
File "multicam_trackwise_evaluation.py", line 383, in
result = Multicam_trackwise_evaluation(dataset_folder="/home/mca/Downloads/wda_tracker-master/MTA_ext_short/test"
File "multicam_trackwise_evaluation.py", line 195, in evaluate
track_eval_res_df = self.get_track_eval_res_df(summary)
File "multicam_trackwise_evaluation.py", line 275, in get_track_eval_res_df
idx_hids = id_global_assignment["idx_hids"]
KeyError: 'idx_hids'

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.