Infrared-dim-small-target-tracking

anti-UAV

Install the environment

Use the Anaconda (CUDA 10.2)

conda create -n ostrack python=3.8
conda activate ostrack
bash install.sh

Set project paths

Run the following command to set paths for this project

python tracking/create_default_local_file.py --workspace_dir . --data_dir ./data --save_dir ./output

Data Preparation

Create a folder video .Put the tracking datasets in ./video

test

Download the model weights from Google Drive

  • video-test
python tracking/video_demo.py ostrack vitb_384_mae_ce_32x4_ep300 ./video/name.mp4

display

ours
n3.mp4
n4.mp4
n6.mp4
n11.mp4
n14.mp4
OSTrack
n14_ostrack.mp4
SiamRPN
n14_siamrpn.mp4