Skip to content
/ DIMBA Public

DIMBA: Discretely Masked Black-Box Attack in Single Object Tracking - Machine Learning Journal (2022)

Notifications You must be signed in to change notification settings

TrustAI/DIMBA

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

94 Commits
 
 
 
 

Repository files navigation

DIMBA: Discretely Masked Black-Box Attack in Single Object Tracking

The adversarial attack can force a CNN-based model to produce an incorrect output by craftily manipulating human-imperceptible input. Exploring such perturbations can help us gain a deeper understanding of the vulnerability of neural networks, and provide robustness to deep learning against miscellaneous adversaries. Despite extensive studies focusing on the robustness of image, audio, and NLP, works on adversarial examples of visual object tracking—especially in a black-box manner—are quite lacking. In this paper, we propose a novel adversarial attack method to generate noises for single object tracking under black-box settings, where perturbations are merely added on initialized frames of tracking sequences, which is difficult to be noticed from the perspective of a whole video clip. Specifically, we divide our algorithm into three components and exploit reinforcement learning for localizing important frame patches precisely while reducing unnecessary computational queries overhead. Compared to existing techniques, our method requires less time to perturb videos, but to manipulate competitive or even better adversarial performance. We test our algorithm in both long-term and short-term datasets, including OTB100, VOT2018, UAV123, and LaSOT. Extensive experiments demonstrate the effectiveness of our method on three mainstream types of trackers: discrimination, Siamese-based, and reinforcement learning-based trackers. We release our attack tool, DIMBA, via GitHub https://github.com/TrustAI/DIMBA for use by the community.

Checkpoints

-SiamMask(ResNet50) https://drive.google.com/file/d/1p4edJaJoYsNwipbisryYoJsefDioyu4a/view?usp=share_link
-SiamRPN(ALexNet) https://drive.google.com/file/d/1g_3mvGaY0ew4rnRXUuXzOZ8SLTv8H3B3/view?usp=share_link
-SiamRPN(MobileNetv2) https://drive.google.com/file/d/1dcIgj0B0FjwV0hL4I3h3juL-EPIcFy7a/view?usp=share_link
-SiamRPN(ResNet50) https://drive.google.com/file/d/1ZgYmPoTWoQwfLxs3VDJ42mmDdCtYNSk0/view?usp=share_link

Installation

Requirements

-Conda with Python 3.7
-NVIDIA GPU
-PyTorch 1.11.0
-yacs
-pyyaml
-matplotlib
-tqdm

Instrusctions

Create Environment and Activate

conda create --name ae python=3.7
conda activate ae

Install numpy/pytorch/opencv

conda install numpy
conda install pytorch=0.4.1 torchvision cuda90 -c pytorch
pip install opencv-python

Download Testing dataset

Download test dataset(e.g. OTB100) and put them into testing_dataset directory.

OTB100 Link

Note: This work is accepted by Machine Learning Journal. Pls find the paper here: DIMBA: Discretely Masked Black-Box Attack in Single Object Tracking

-- Xiangyu Yin & Wenjie Ruan

About

DIMBA: Discretely Masked Black-Box Attack in Single Object Tracking - Machine Learning Journal (2022)

Resources

Stars

Watchers

Forks