project_foxhound/anima_module.yaml at main · RobotFlow-Labs/project_foxhound · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
name: foxhound
codename: FOXHOUND
wave: 10
theme: WARDOG
version: "0.1.0"

paper:
  title: "Learning Motion Blur Robust ViTs for UAV Tracking (BDTrack)"
  arxiv_id: "2407.05383"
  github: "https://github.com/wuyou3474/BDTrack"

status: build_complete
tier: T3
defense_score: 38

description: >
  BDTrack-based UAV tracker with blur-robust ViT backbone, dynamic early exit
  (DEEM), and YOLO26 proposal integration for real-time drone defense.

compute:
  backends: [cuda, mlx, cpu]
  default_backend: auto
  gpu_vram_mb: 4000  # estimated for DeiT-Tiny
  batch_size_cuda: 16
  batch_size_mlx: 4

entry_points:
  train: "python -m anima_foxhound.train"
  eval: "python -m anima_foxhound.eval"
  serve: "python -m anima_foxhound.serve"
  export: "python -m anima_foxhound.export"
  cli: "python -m anima_foxhound"

docker:
  train: docker/Dockerfile.cuda
  serve: docker/Dockerfile.serve
  mlx: docker/Dockerfile.mlx

ros2:
  node: ros2/foxhound_node.py
  launch: ros2/launch/foxhound.launch.py
  topics:
    subscribe:
      - /camera/image_raw
      - /foxhound/template_update
      - /detections/yolo26
    publish:
      - /foxhound/track
      - /foxhound/status

dependencies:
  models:
    - deit_tiny_patch16_224
  datasets:
    - UAV123
    - VisDrone2018
    - UAVDT

artifacts:
  checkpoints: /mnt/artifacts-datai/checkpoints/project_foxhound
  logs: /mnt/artifacts-datai/logs/project_foxhound
  exports: /mnt/artifacts-datai/exports/project_foxhound
  tensorboard: /mnt/artifacts-datai/tensorboard/project_foxhound