配置OSTrack到Anti-UA

前期准备

1. 在github上下载OSTrack源码，根据README配置环境。

注意：当安装pytorch1.13后，再利用install.sh安装其他库时，会报错

"libcublas.so.11: symbol cublasLtHSHMatmulAlgoInit"，

执行"pip uninstall nvidia_cublas_cu11"即可解决，详情见"https://blog.csdn.net/be_humble/article/details/129038771"

2. 测试OSTrack的训练和测试等，确保正常运行

配置Anti-UAV410数据集

测试

1. 打开./lib/test/evaluation.py，添加新数据集地址

settings.antiuav410_path = '/data3/publicData/Anti-UAV410/Anti-UAV'

2. 打开./lib/test/evaluation/datasets.py，仿照GOT10K添加路径

antiuav410_test=DatasetInfo(module=pt % "antiuav410", class_name="AntiUAV410Dataset", kwargs=dict(split='test')),

antiuav410_val=DatasetInfo(module=pt % "antiuav410", class_name="AntiUAV410Dataset", kwargs=dict(split='val')),

antiuav410_ltrval=DatasetInfo(module=pt % "antiuav410", class_name="AntiUAV410Dataset", kwargs=dict(split='ltrval')),

3. 在./lib/test/evalution/下新建antiuav410dataset.py，内容仿照got10kdataset.py

4. 执行"CUDA_VISIBLE_DEVICES=2,3 python tracking/test.py ostrack vitb_384_mae_ce_32x4_ep300 --dataset antiuav410_test --threads 16 --num_gpus 2"

结果保存在"/data3/QHL01/Anti-UAV/OSTrack/output/test/tracking_results/ostrack/vitb_384_mae_ce_32x4_ep300_antiuav410/"下

注意修改文件夹名称，避免结果覆盖

5. 在./tracking/下新建evalute_antiuav410.py，根据第二届Anti-UAV编写相应的评价指标函数。

注意：直接使用HB和CJJ论文的代码时，"pred_res[:, 2:] = pred_res[:, 2:] - pred_res[:, :2] + 1"要注释掉，因为OSTrack输出的是（x,y,w,h）左上角点坐标和宽高，GlobalTrack输出的是（x,y,x,y），左上和右下坐标。

6. 运行评价函数，结果应为0.5341，此为未经过fine-tune的结果

训练

1. 修改/lib/train/admin/local.py，添加新数据集路径

self.antiuav410_dir = '/data3/publicData/Anti-UAV410/Anti-UAV/train'

self.antiuav410_val_dir = '/data3/publicData/Anti-UAV410/Anti-UAV/val'

2. 修改/lib/train/dataset/__init__.py，添加import

from .antiuav410 import AntiUAV410

3. 在/lib/train/dataset/下添加antiuav410.py，仿照got10k.py

4. 修改/lib/train/base_functions.py，

添加"from lib.train.dataset import AntiUAV410"

在assert name in中添加"AntiUAV410"，"AntiUAV410_val"

添加

if name == "AntiUAV410":

datasets.append(AntiUAV410(settings.env.antiuav410_dir, image_loader=image_loader))

if name == "AntiUAV410_val":

datasets.append(AntiUAV410(settings.env.antiuav410_val_dir, image_loader=image_loader))

5. 在/experiments/ostrack/下新建config文件