Commit c8d3fbd
committed
feat(models): add the StreamPETR detector
The query-based head with temporal memory (pre/post memory update, top-k
propagation, denoising queries, geometry-aware position embeddings) and the
model wrapper with the three-module ONNX export ABI.
Signed-off-by: vividf <yihsiang.fang@tier4.jp>1 parent a005ccd commit c8d3fbd
4 files changed
Lines changed: 1877 additions & 0 deletions
File tree
- autoware_ml
- models/detection3d
- heads
- tests/models
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
65 | 65 | | |
66 | 66 | | |
67 | 67 | | |
| 68 | + | |
68 | 69 | | |
69 | 70 | | |
70 | 71 | | |
| |||
152 | 153 | | |
153 | 154 | | |
154 | 155 | | |
| 156 | + | |
155 | 157 | | |
156 | 158 | | |
| 159 | + | |
157 | 160 | | |
158 | 161 | | |
159 | 162 | | |
| |||
0 commit comments