FairMOT-attack
FairMOT-attack copied to clipboard
TraSw for FairMOT
- A Single-Target Attack example (Attack ID: 19; Screener ID: 24):
TraSw: Tracklet-Switch Adversarial Attacks against Multi-Object Tracking,
Delv Lin, Qi Chen, Chengyu Zhou, Kun He,
arXiv 2111.08954
Related Works
Abstract
Multi-Object Tracking (MOT) has achieved aggressive progress and derives many excellent deep learning models. However, the robustness of the trackers is rarely studied, and it is challenging to attack the MOT system since its mature association algorithms are designed to be robust against errors during the tracking. In this work, we analyze the vulnerability of popular pedestrian MOT trackers and propose a novel adversarial attack method called Tracklet-Switch (TraSw) against the complete tracking pipeline of MOT. TraSw can fool the advanced deep trackers (i.e., FairMOT and ByteTrack) to fail to track the targets in the subsequent frames by attacking very few frames. Experiments on the MOT-Challenge datasets (i.e., 2DMOT15, MOT17, and MOT20) show that TraSw can achieve an extraordinarily high success rate of over 95% by attacking only four frames on average. To our knowledge, this is the first work on the adversarial attack against pedestrian MOT trackers.
Attack Performance
Single-Target Attack Results on MOT challenge test set
Dataset | Suc. Rate | Avg. Frames | Total L2 Distance |
---|---|---|---|
2DMOT15 | 93.28% | 4.25 | 13.99 |
MOT17 | 91.03% | 4.74 | 14.40 |
MOT20 | 96.46% | 3.94 | 11.67 |
Installation
-
same as FairMOT
-
Clone this repo, and we'll call the directory that you cloned as ${FA_ROOT}
-
Install dependencies. We use python 3.7 and pytorch >= 1.2.0
-
conda create -n FA conda activate FA conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch cd ${FA_ROOT} pip install -r requirements.txt cd src/lib/models/networks/DCNv2 sh make.sh
-
We use DCNv2 in our backbone network and more details can be found in their repo.
-
In order to run the code for demos, you also need to install ffmpeg.
Data preparation
-
We only use the same test data as FairMOT.
-
2DMOT15, MOT17 and MOT20 can be downloaded from the official webpage of MOT-Challenge. After downloading, you should prepare the data in the following structure:
${DATA_DIR} ├── MOT15 │ └── images │ ├── test │ └── train ├── MOT17 │ └── images │ ├── test │ └── train └── MOT20 └── images ├── test └── train
Target Model
- We choose DLA-34: [Google] [Baidu, code: 88yn] trained by FairMOT as our primary target model.
Tracking without Attack
- tracking on original videos of 2DMOT15, MOT17, and MOT20
cd src
python track.py mot --test_mot15 True --load_model all_dla34.pth --conf_thres 0.3 --data_dir ${DATA_DIR} --output_dir ${OUTPUT_DIR}
python track.py mot --test_mot17 True --load_model all_dla34.pth --conf_thres 0.4 --data_dir ${DATA_DIR} --output_dir ${OUTPUT_DIR}
python track.py mot --test_mot20 True --load_model all_dla34.pth --conf_thres 0.3 --data_dir ${DATA_DIR} --output_dir ${OUTPUT_DIR}
Attack
- attack all attackable objects separately in videos in parallel (may require a lot of memory).
cd src
python track.py mot --test_mot15 True --load_model all_dla34.pth --conf_thres 0.3 --data_dir ${DATA_DIR} --output_dir ${OUTPUT_DIR} --attack single --attack_id -1
python track.py mot --test_mot17 True --load_model all_dla34.pth --conf_thres 0.4 --data_dir ${DATA_DIR} --output_dir ${OUTPUT_DIR} --attack single --attack_id -1
python track.py mot --test_mot20 True --load_model all_dla34.pth --conf_thres 0.3 --data_dir ${DATA_DIR} --output_dir ${OUTPUT_DIR} --attack single --attack_id -1
- attack a specific object in a specific video (require to set specific video in
src/track.py
).
cd src
python track.py mot --test_mot15 True --load_model all_dla34.pth --conf_thres 0.3 --data_dir ${DATA_DIR} --output_dir ${OUTPUT_DIR} --attack single --attack_id ${a specific id in origial tracklets}
python track.py mot --test_mot17 True --load_model all_dla34.pth --conf_thres 0.4 --data_dir ${DATA_DIR} --output_dir ${OUTPUT_DIR} --attack single --attack_id ${a specific id in origial tracklets}
python track.py mot --test_mot20 True --load_model all_dla34.pth --conf_thres 0.3 --data_dir ${DATA_DIR} --output_dir ${OUTPUT_DIR} --attack single --attack_id ${a specific id in origial tracklets}
Acknowledgement
This source code is based on FairMOT. Thanks for their wonderful works.
Citation
@misc{lin2021trasw,
title={TraSw: Tracklet-Switch Adversarial Attacks against Multi-Object Tracking},
author={Delv Lin and Qi Chen and Chengyu Zhou and Kun He},
year={2021},
eprint={2111.08954},
archivePrefix={arXiv},
primaryClass={cs.CV}
}