zacxr commited on
Commit
8e8d0f7
·
verified ·
1 Parent(s): 71b0141

Add README.md

Browse files
Files changed (1) hide show
  1. README.md +66 -0
README.md ADDED
@@ -0,0 +1,66 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ tags:
4
+ - heal
5
+ - horizon
6
+ ---
7
+
8
+ # MixVarGENet
9
+
10
+ MixVarGENet builds a lightweight backbone with mixed-variant blocks (mixvarge); the first two stages use f2/f4 base blocks, the last two stages use f2_gb16 blocks with groups, downsampling stage by stage to stride 32.
11
+
12
+ ---
13
+
14
+ ## Deployment Metrics
15
+
16
+ ### Model Parameters
17
+
18
+ | Model | Model Input | Backbone | Neck | Model Output |
19
+ |---|---|---|---|---|
20
+ | MixVarGENet | `1x3x224x224` | MixVarGENet | 5 stage mixvarge blocks | classification logits `(B,1000)` |
21
+
22
+ ### Accuracy Metrics
23
+
24
+ | March | Metric | float | calibration | qat | hbm |
25
+ | --- | --- | --- | --- | --- | --- |
26
+ | J6M | Accuracy | 0.716 | 0.7116 | — | 0.7116 |
27
+
28
+ > Data tested with `march = March.NASH_M` (J6M); this task has no QAT stage (`—` in the qat column).
29
+ >
30
+ > HEAL versions: heal 0.0.2 / hbdk4-compiler 4.11.11 / horizon_plugin_pytorch 3.3.10.
31
+
32
+ ### Performance Metrics
33
+
34
+ > **Performance test methodology**: FPS is measured with 8 threads on a single core; Latency is measured with single core, single thread; Memory is peak DDR usage.
35
+
36
+ | March | latency (ms) | fps | Memory Usage |
37
+ |---|---|---|---|
38
+ | J6M | 0.36 | 5464.88 | 6.40 |
39
+ | J6P | 0.33 | 9374.11 | 6.40 |
40
+ | J6B | - | - | - |
41
+
42
+ J6B performance is not available for this model.
43
+
44
+ ---
45
+
46
+ ## Model Overview
47
+
48
+ ### Core Design
49
+
50
+ MixVarGENet builds a lightweight backbone with mixed-variant blocks (mixvarge); the first two stages use f2/f4 base blocks, the last two stages use f2_gb16 blocks with groups, downsampling stage by stage to stride 32.
51
+
52
+ - **Task type**: Image Classification.
53
+ - **backbone**: MixVarGENet (`type=MixVarGENet`, 5 stages: stride 2/4/8/16/32, `num_classes=1000`), builds a lightweight backbone with mixed-variant blocks (mixvarge), downsampling stage by stage to stride 32.
54
+ - **neck**: 5 stages of mixvarge blocks: `mixvarge_f2`, `mixvarge_f4`, `mixvarge_f2_gb16` (last two stages use group blocks, gb16), channels 32→32→64→96→160.
55
+ - **classification head**: MixVarGENet built-in fully connected classification head, directly outputs 1000-class logits.
56
+ - **Loss function**: `CEWithLabelSmooth` (cross-entropy with label smoothing).
57
+ - **Model input**: single RGB image at resolution `224 × 224` (`1x3x224x224`).
58
+ - **Model output**: 1000-class prediction logits; argmax gives the predicted class.
59
+
60
+ ### Official Repo and Paper
61
+
62
+ MixVarGENet is an efficient network architecture proposed by Horizon Robotics, optimized for the Journey series chips.
63
+
64
+ ### Reference
65
+
66
+ For more J6 chip deployment details, see https://developer.horizon.auto/blog/10387