FBAGSTM commited on
Commit
a3aa9ee
·
verified ·
1 Parent(s): 294ccaa

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +67 -6
README.md CHANGED
@@ -1,6 +1,67 @@
1
- ---
2
- license: other
3
- license_name: apache2.0
4
- license_link: >-
5
- https://github.st.com/AIS/stm32ai-modelzoo/raw/master/pose_estimation/LICENSE.md
6
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ license_link: >-
4
+ https://github.st.com/AIS/stm32ai-modelzoo/raw/master/pose_estimation/LICENSE.md
5
+ ---
6
+ # Head landmarks quantized
7
+
8
+ ## **Use case** : `Pose estimation`
9
+
10
+ # Model description
11
+
12
+
13
+ Head landmarks is a single pose estimation model targeted for real-time processing implemented in ONNX.
14
+
15
+ The model is quantized in int8 format using onnx quantizer.
16
+
17
+ ## Network information
18
+
19
+
20
+ | Network information | Value |
21
+ |-------------------------|-----------------|
22
+ | Framework | ONNX Runtime |
23
+ | Quantization | int8 |
24
+ | Provenance | https://github.com/PINTO0309/PINTO_model_zoo/tree/main/032_FaceMesh
25
+ | Paper | https://developers.google.com/ml-kit/vision/face-mesh-detection |
26
+
27
+
28
+ ## Networks inputs / outputs
29
+
30
+ With an image resolution of NxM with K keypoints to detect :
31
+
32
+ | Input Shape | Description |
33
+ | ----- | ----------- |
34
+ | (1, N, M, 3) | Single NxM RGB image with UINT8 values between 0 and 255 |
35
+
36
+ | Output Shape | Description |
37
+ | ----- | ----------- |
38
+ | (1, 1, 1, Kx2) | FLOAT values Where Kx2 are the (x,y) values of each keypoints |
39
+
40
+ ## Recommended Platforms
41
+
42
+ | Platform | Supported | Recommended |
43
+ |----------|-----------|-------------|
44
+ | STM32L0 | [] | [] |
45
+ | STM32L4 | [] | [] |
46
+ | STM32U5 | [] | [] |
47
+ | STM32H7 | [] | [] |
48
+ | STM32MP1 | [x] | [] |
49
+ | STM32MP2 | [x] | [x] |
50
+ | STM32N6 | [x] | [x] |
51
+
52
+ # Performances
53
+
54
+ ## Metrics
55
+
56
+ Measures are done with default STM32Cube.AI configuration with enabled input / output allocated option.
57
+
58
+ ### Reference **NPU** memory footprint
59
+ |Model | Format | Resolution | Series | Internal RAM (KiB) | External RAM (KiB) | Weights Flash (KiB) | STM32Cube.AI version | STEdgeAI Core version |
60
+ |----------|--------|-------------|------------------|------------------|---------------------|-------|----------------------|-------------------------|
61
+ | [head_landmarks](https://github.com/STMicroelectronics/stm32ai-modelzoo/tree/main/pose_estimation/hand_landmarks/Public_pretrainedmodel_custom_dataset/custom_dataset_heads_468kpts/face_landmarks_v1_192_int8_pc.onnx) | Int8 | 224x224x3 | STM32N6 | 1739.5 | 0.0 | 3246.47 | 10.2.0 | 2.2.0 |
62
+
63
+ ### Reference **NPU** inference time
64
+ | Model | Format | Resolution | Board | Execution Engine | Inference time (ms) | Inf / sec | STM32Cube.AI version | STEdgeAI Core version |
65
+ |--------|--------|-------------|------------------|------------------|---------------------|-------|----------------------|-------------------------|
66
+ | [head_landmarks](https://github.com/STMicroelectronics/stm32ai-modelzoo/tree/main/pose_estimation/hand_landmarks/Public_pretrainedmodel_custom_dataset/custom_dataset_heads_468kpts/face_landmarks_v1_192_int8_pc.onnx) | Int8 | 224x224x3 | STM32N6570-DK | NPU/MCU | 20.52 | 48.73 | 10.2.0 | 2.2.0 |
67
+