Pineberry commited on
Commit
24b9d6a
·
verified ·
1 Parent(s): 546b351

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +27 -10
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
- * Operators: 44 (10 types)
 
 
27
  - Conv2D: 4
28
- - Producer: 21
29
- - ReLU: 9
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