Update README.md
Browse files
README.md
CHANGED
|
@@ -9,6 +9,23 @@ tags:
|
|
| 9 |
# DS-CNN
|
| 10 |
|
| 11 |
This is an exported version of DS-CNN from Aidge.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
|
| 13 |
## Aidge support
|
| 14 |
|
|
@@ -16,24 +33,24 @@ This is an exported version of DS-CNN from Aidge.
|
|
| 16 |
|
| 17 |
| Feature | Tested in CI |
|
| 18 |
| :---------: | :----------: |
|
| 19 |
-
| ONNX import |
|
| 20 |
-
| Backend CPU | ❌
|
| 21 |
| Export CPP | ❌ |
|
| 22 |
|
| 23 |
|
| 24 |
## Model
|
| 25 |
|
| 26 |
-
*
|
|
|
|
|
|
|
| 27 |
- Conv2D: 4
|
| 28 |
-
-
|
| 29 |
-
-
|
| 30 |
-
- FC: 1
|
| 31 |
-
- Transpose: 1
|
| 32 |
- PaddedConvDepthWise2D: 4
|
|
|
|
|
|
|
|
|
|
| 33 |
- Softmax: 1
|
| 34 |
-
- Reshape: 1
|
| 35 |
-
- PaddedConv2D: 1
|
| 36 |
-
- AvgPooling2D: 1
|
| 37 |
|
| 38 |
## Google Speech Commands v2
|
| 39 |
|
|
|
|
| 9 |
# DS-CNN
|
| 10 |
|
| 11 |
This is an exported version of DS-CNN from Aidge.
|
| 12 |
+
Model from https://github.com/mlcommons/tiny/tree/master/benchmark/training/keyword_spotting
|
| 13 |
+
|
| 14 |
+
ONNX version exported from `.pb` model doing
|
| 15 |
+
|
| 16 |
+
```bash
|
| 17 |
+
# setup environment
|
| 18 |
+
python3.10 -m venv pytf
|
| 19 |
+
source pytf/bin/activate
|
| 20 |
+
|
| 21 |
+
# install latest officially compatible versions
|
| 22 |
+
pip install tensorflow==2.15.1 tf2onnx==1.16.1
|
| 23 |
+
|
| 24 |
+
# use most recent opset officially supported
|
| 25 |
+
python -m tf2onnx.convert \
|
| 26 |
+
--saved-model <path/to/dir> \
|
| 27 |
+
--output ds_cnn.onnx --opset 18
|
| 28 |
+
```
|
| 29 |
|
| 30 |
## Aidge support
|
| 31 |
|
|
|
|
| 33 |
|
| 34 |
| Feature | Tested in CI |
|
| 35 |
| :---------: | :----------: |
|
| 36 |
+
| ONNX import | ✔️ |
|
| 37 |
+
| Backend CPU | ❌ |
|
| 38 |
| Export CPP | ❌ |
|
| 39 |
|
| 40 |
|
| 41 |
## Model
|
| 42 |
|
| 43 |
+
* operators: 46 (10 types)
|
| 44 |
+
- Add: 1
|
| 45 |
+
- AvgPooling2D: 1
|
| 46 |
- Conv2D: 4
|
| 47 |
+
- MatMul: 1
|
| 48 |
+
- PaddedConv2D: 1
|
|
|
|
|
|
|
| 49 |
- PaddedConvDepthWise2D: 4
|
| 50 |
+
- Producer: 22
|
| 51 |
+
- ReLU: 9
|
| 52 |
+
- Reshape: 2
|
| 53 |
- Softmax: 1
|
|
|
|
|
|
|
|
|
|
| 54 |
|
| 55 |
## Google Speech Commands v2
|
| 56 |
|