liuguoyou / MFT

MFT: Long-Term Tracking of Every Pixel -- code for the WACV 2024 paper

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

MFT: Long-Term Tracking of Every Pixel

Project Page Official implementation of the MFT tracker from the paper:

Michal Neoral, Jonáš Šerých, Jiří Matas: "MFT: Long-Term Tracking of Every Pixel", WACV 2024

Please cite our paper, if you use any of this.

@inproceedings{neoral2024mft,
               title={{MFT}: Long-Term Tracking of Every Pixel},
               author={Neoral, Michal and {\v{S}}er{\`y}ch, Jon{\'a}{\v{s}} and Matas, Ji{\v{r}}{\'\i}},
               journal={arXiv preprint arXiv:2305.12998},
               year={2023},
}

Install

Create and activate a new virtualenv:

# we have tested with python 3.7.4
python -m venv venv
source venv/bin/activate

Then install all the dependencies:

pip install torch numpy einops tqdm opencv-python scipy Pillow==9 matplotlib ipdb

Run the demo

Simply running:

python demo.py

should produce a demo_out directory with two visualizations.

License

The demo video in demo_in was extracted from youtube.

This work is licensed under the Attribution-NonCommercial-ShareAlike 4.0 International license. The MFT/RAFT directory contains a modified version of RAFT, which is licensed under BSD-3-Clause license. Our modifications (OcclusionAndUncertaintyBlock and its integration in raft.py) are licensed again under the Attribution-NonCommercial-ShareAlike 4.0 International.

Acknowledgments

This work was supported by Toyota Motor Europe, by the Grant Agency of the Czech Technical University in Prague, grant No. SGS23/173/OHK3/3T/13, and by the Research Center for Informatics project CZ.02.1.01/0.0/0.0/16_019/0000765 funded by OP VVV.

About

MFT: Long-Term Tracking of Every Pixel -- code for the WACV 2024 paper

License:Other


Languages

Language:Python 96.7%Language:Cuda 2.9%Language:C++ 0.4%Language:Shell 0.0%