NavidGh commited on
Commit
7a9dcff
·
verified ·
1 Parent(s): 41a5e3e

Add model card

Browse files
Files changed (1) hide show
  1. README.md +71 -0
README.md ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: cc-by-nc-4.0
3
+ tags:
4
+ - object-detection
5
+ - multi-object-tracking
6
+ - thermal
7
+ - wildlife
8
+ - yolo
9
+ - rf-detr
10
+ library_name: pytorch
11
+ ---
12
+
13
+ # BAMBI Wildlife MOT — trained weights
14
+
15
+ Trained model checkpoints for the **BAMBI Wildlife MOT** project: detection and
16
+ multi-object tracking of wildlife (Wild boar, Red deer, Roe deer) in aerial
17
+ **thermal** imagery from the [BAMBI dataset](https://www.bambi.eco/).
18
+
19
+ Code, documentation, and reproduction instructions:
20
+ **https://github.com/Navid-alt/BAMBI_MOT**
21
+
22
+ These weights are git-ignored in the source repo because of their size and are
23
+ hosted here instead. File paths below mirror where each script writes/expects the
24
+ checkpoint in the repo checkout.
25
+
26
+ ## Files
27
+
28
+ ### Detectors
29
+
30
+ | Model | File | Place at (in repo checkout) |
31
+ |---|---|---|
32
+ | YOLO26-s (640) | `yolo26-s/runs/train_yolo26-s/weights/best.pt` | `detection_models/yolo26-s/runs/train_yolo26-s/weights/best.pt` |
33
+ | YOLO26-l (1024) | `yolo26-l/runs/train_yolo26-l_1024/weights/best.pt` | `detection_models/yolo26-l/runs/train_yolo26-l_1024/weights/best.pt` |
34
+ | YOLO26-l (default res) | `yolo26-l/runs/train_yolo26-l/weights/best.pt` | `detection_models/yolo26-l/runs/train_yolo26-l/weights/best.pt` |
35
+ | RF-DETR-l — EMA (recommended) | `rf-detr-l/output/checkpoint_best_ema.pth` | `detection_models/rf-detr-l/output/checkpoint_best_ema.pth` |
36
+ | RF-DETR-l — regular | `rf-detr-l/output/checkpoint_best_regular.pth` | `detection_models/rf-detr-l/output/checkpoint_best_regular.pth` |
37
+
38
+ `*/weights/last.pt` (YOLO) and `rf-detr-l/output/{checkpoint_39,last}.ckpt`
39
+ (full training state incl. optimizer, ~527 MB each) are included for full
40
+ reproducibility but are **not needed for inference**.
41
+
42
+ ### Learned tracker (TrackAssociator, MOTRv2-style)
43
+
44
+ | File | Notes |
45
+ |---|---|
46
+ | `transformer_tracking/output/associator_last.pth` | Final associator — use this for inference |
47
+ | `transformer_tracking/output/associator_epoch{0..9}.pth` | Per-epoch checkpoints |
48
+
49
+ The associator runs on top of the **frozen** RF-DETR-l detector
50
+ (`checkpoint_best_ema.pth` above).
51
+
52
+ ## Classes
53
+
54
+ `0: Wild boar` · `1: Red deer` · `2: Roe deer`
55
+
56
+ ## Usage
57
+
58
+ Download a file and drop it at the path listed above; the detection/tracking
59
+ scripts pick it up automatically. Example:
60
+
61
+ ```python
62
+ from huggingface_hub import hf_hub_download
63
+
64
+ ckpt = hf_hub_download(
65
+ repo_id="NavidGh/BambiMot",
66
+ filename="rf-detr-l/output/checkpoint_best_ema.pth",
67
+ )
68
+ ```
69
+
70
+ See the GitHub repo's `detection_models/readme.md` and
71
+ `transformer_tracking/README.md` for training and evaluation commands.