Release reported GaussianMOT checkpoint (epoch 23)
Browse files- README.md +49 -0
- gaussianmot_epoch23.ckpt +3 -0
README.md
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
tags:
|
| 4 |
+
- 3d-object-detection
|
| 5 |
+
- multi-object-tracking
|
| 6 |
+
- gaussian-splatting
|
| 7 |
+
- camera-radar-fusion
|
| 8 |
+
- nuscenes
|
| 9 |
+
datasets:
|
| 10 |
+
- nuscenes
|
| 11 |
+
---
|
| 12 |
+
|
| 13 |
+
# GaussianMOT
|
| 14 |
+
|
| 15 |
+
Joint 3D object detection and tracking on a camera–radar Gaussian BEV.
|
| 16 |
+
Code: <https://github.com/KimJunHan/GaussianMOT>
|
| 17 |
+
|
| 18 |
+
This repository holds the checkpoint reported in the paper (weights only,
|
| 19 |
+
end of the 24th training epoch).
|
| 20 |
+
|
| 21 |
+
| File | Size | SHA-256 |
|
| 22 |
+
|---|---|---|
|
| 23 |
+
| `gaussianmot_epoch23.ckpt` | 368 MB | `d3e3a5e55265b878420dbbcf1d6868e47350e8c968fb5061e2aad3b88af5dd23` |
|
| 24 |
+
|
| 25 |
+
## Results (nuScenes val, inference seed 0)
|
| 26 |
+
|
| 27 |
+
| Detection | | Tracking | |
|
| 28 |
+
|---|---|---|---|
|
| 29 |
+
| mAP | 35.0 | AMOTA | 0.348 |
|
| 30 |
+
| NDS | 42.5 | AMOTP | 1.31 |
|
| 31 |
+
| mATE | 0.539 | Recall | 0.411 |
|
| 32 |
+
| mASE | 0.307 | ID-S | 897 |
|
| 33 |
+
| mAOE | 0.681 | | |
|
| 34 |
+
| mAVE | 0.511 | | |
|
| 35 |
+
| mAAE | 0.465 | | |
|
| 36 |
+
|
| 37 |
+
6 cameras + 5 radars, 448×800 input, 6.5 FPS on an RTX 3090 (batch size 1).
|
| 38 |
+
|
| 39 |
+
## Usage
|
| 40 |
+
|
| 41 |
+
```bash
|
| 42 |
+
git clone https://github.com/KimJunHan/GaussianMOT && cd GaussianMOT
|
| 43 |
+
hf download kimjunhan/GaussianMOT gaussianmot_epoch23.ckpt --local-dir ckpt
|
| 44 |
+
python tools/evaluate.py checkpoint_path=ckpt/gaussianmot_epoch23.ckpt device=cuda
|
| 45 |
+
```
|
| 46 |
+
|
| 47 |
+
The file contains `state_dict` only (no optimizer state); it loads with
|
| 48 |
+
`tools/evaluate.py` as is. Densification draws random shifts at inference, so
|
| 49 |
+
results depend on `eval_seed` (default 0) in the last digit.
|
gaussianmot_epoch23.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d3e3a5e55265b878420dbbcf1d6868e47350e8c968fb5061e2aad3b88af5dd23
|
| 3 |
+
size 367668342
|