heal
horizon
occupancy
lane-detection
zacxr commited on
Commit
03ad201
·
verified ·
1 Parent(s): e2960c1

Add README.md

Browse files
Files changed (1) hide show
  1. README.md +77 -0
README.md ADDED
@@ -0,0 +1,77 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ tags:
4
+ - heal
5
+ - horizon
6
+ - occupancy
7
+ - lane-detection
8
+ ---
9
+
10
+ # SparseMultitaskOE+MapTR+FlashOcc+HENet Multitask
11
+
12
+ SparseMultitaskOE uses HENet as backbone to extract multi-view features, converts to BEV features, then SparseBEVOEHead drives three task heads: detection head (SparseBEVOE detection, 10-class 3D boxes), map head (MapTR-style vectorized map, 3 element classes), occupancy head (FlashOcc-style semantic occupancy, 18 classes). This task uses three-stage Float training: stage1 jointly trains det+map+occ (backbone), stage2 freezes backbone and trains map+occ heads only (`lr=1e-5`), merge stage combines stage1 (det) and stage2 (map/occ) weights into a complete model. `use_lidar_gt=True`, `defer_vectormap=True` (map generated online).
13
+
14
+ ---
15
+
16
+ ## Deployment Metrics
17
+
18
+ ### Model Parameters
19
+
20
+ | Model | Model Input | Backbone | Neck | Model Output |
21
+ |---|---|---|---|---|
22
+ | SparseMultitaskOE | 6-camera multi-view images `(B,6,3,256,704)` + lidar point cloud `(B,N,5)` | HENet-tiny | MMFPN | det detection boxes `(B,N,cls+reg)`; map vectorized map `(B,L,P,2)`; occ occupancy grid `(B,C,H,W)` |
23
+
24
+ ### Accuracy Metrics
25
+
26
+ | March | Metric | float | calibration | qat | hbm |
27
+ | --- | --- | --- | --- | --- | --- |
28
+ | J6M | NDS | 0.5434 | 0.5325 | — | 0.5281 |
29
+ | | mAP | — | — | — | — |
30
+ | | chamfer mAP (MAP) | 0.592 | 0.5838 | — | 0.5833 |
31
+ | | Occ mIoU | 0.3197 | 0.3277 | — | 0.3273 |
32
+
33
+ > Data measured with `march = March.NASH_M` (J6M) configuration; this task has no QAT stage (qat column is `—`).
34
+ >
35
+ > HEAL version: heal 0.0.2 / hbdk4-compiler 4.11.11 / horizon_plugin_pytorch 3.3.10.
36
+
37
+ ### Performance Metrics
38
+
39
+ > **Performance test methodology**: FPS for J6M/J6P is single-core eight-thread; J6B is single-core dual-thread; Latency is single-core single-thread; Memory is peak DDR usage.
40
+
41
+ | March | latency (ms) | fps | Memory Usage |
42
+ |---|---|---|---|
43
+ | J6M | 27.65 | 36.66 | 183.60 |
44
+ | J6P | 17.75 | 188.57 | 213.50 |
45
+ | J6B | 201.07 | 7.88 | 122.00 |
46
+
47
+ ---
48
+
49
+ ## Model Overview
50
+
51
+ ### Core Design
52
+
53
+ SparseMultitaskOE uses HENet as backbone to extract multi-view features, converts to BEV features, then SparseBEVOEHead drives three task heads: detection head (SparseBEVOE detection, 10-class 3D boxes), map head (MapTR-style vectorized map, 3 element classes), occupancy head (FlashOcc-style semantic occupancy, 18 classes). This task uses three-stage Float training: stage1 jointly trains det+map+occ (backbone), stage2 freezes backbone and trains map+occ heads only (`lr=1e-5`), merge stage combines stage1 (det) and stage2 (map/occ) weights into a complete model. `use_lidar_gt=True`, `defer_vectormap=True` (map generated online).
54
+
55
+ - **Task type**: Multitask fusion (3D object detection + vectorized map construction + occupancy grid prediction).
56
+ - **backbone**: HENet-tiny (pretrained).
57
+ - **neck**: MMFPN.
58
+ - **Detection head**: `SparseBEVOEHead` (`SparseBEVOEEncoder` + `SparseBEVOERefinementModule`, 10-class 3D detection boxes, `num_classes=10`).
59
+ - **Map head**: `SparseMapPerceptionDecoder` (`SparseMapHead`, 3-class vectorized map elements, `map_classes=[divider,ped_crossing,boundary]`).
60
+ - **Occupancy head**: Semantic occupancy prediction, 18 classes (`num_classes_occ=18`).
61
+ - **BEV range**: `point_cloud_range=[-51.2,-51.2,-5.0,51.2,51.2,3.0]` (det/occ), `map_point_cloud_range=[-15.0,-30.0,-10.0,15.0,30.0,10.0]` (map), `occ_bev_size=(40,40,0.625)`.
62
+ - **Model input**: 6-camera multi-view images (B,6,3,256,704) + lidar point cloud (B,N,D).
63
+ - **Model output**: det 3D detection boxes + map vectorized map elements + occ occupancy grid semantics.
64
+
65
+ **Deployment notes**: HBIR export enables `enable_vpu=True`; compilation uses `input_source=compile_cfg["input_source"]`.
66
+
67
+ ### Official Repo and Paper
68
+
69
+ Official repo: det SparseBEV/map MapTR/occ FlashOcc
70
+ https://github.com/MCG-NJU/SparseBEV, https://github.com/hustvl/MapTR, https://github.com/Yzichen/FlashOCC
71
+ Paper: https://arxiv.org/abs/2308.09244, https://arxiv.org/abs/2208.14437, https://arxiv.org/abs/2311.12058
72
+
73
+ Note: Camera backbone HENet is HEAL in-house; upstream papers for det/map/occ are SparseBEV/MapTR/FlashOcc respectively.
74
+
75
+ ### Reference
76
+
77
+ For more J6 chip deployment details, see https://developer.horizon.auto/blog/13254