cmoineau commited on
Commit
e4820c4
·
verified ·
1 Parent(s): b079402

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +56 -2
README.md CHANGED
@@ -4,10 +4,64 @@ datasets:
4
  - uoft-cs/cifar100
5
  base_model:
6
  - microsoft/resnet-18
 
 
 
7
  ---
8
 
9
- # Resnet18 - Aidge
10
 
11
  This is an exported version of Resnet18 from Aidge.
12
 
13
- The original version trained on Imagenet and the finetuned version on CIFAR100.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  - uoft-cs/cifar100
5
  base_model:
6
  - microsoft/resnet-18
7
+ pipeline_tag: image-classification
8
+ tags:
9
+ - arxiv:1512.03385
10
  ---
11
 
12
+ # Resnet18
13
 
14
  This is an exported version of Resnet18 from Aidge.
15
 
16
+ The original version trained on Imagenet and the finetuned version on CIFAR100.
17
+
18
+ ## Aidge support
19
+
20
+ > Note: We tested this network for the following features. If you encounter any error please open an [issue](https://gitlab.eclipse.org/groups/eclipse/aidge/-/issues). Features not tested in CI may not be functional.
21
+
22
+ | Feature | Tested in CI |
23
+ | :---------: | :----------: |
24
+ | ONNX import | ✔ |
25
+ | Backend CPU | ✔ |
26
+ | Export CPP | ❌ |
27
+
28
+
29
+ ## Model
30
+
31
+ * Operators: 171 (11 types)
32
+ - Add: 8
33
+ - BatchNorm2D: 20
34
+ - Conv2D: 3
35
+ - FC: 1
36
+ - Flatten: 1
37
+ - GlobalAveragePooling: 1
38
+ - Identity: 3
39
+ - PaddedConv2D: 17
40
+ - PaddedMaxPooling2D: 1
41
+ - Producer: 99
42
+ - ReLU: 17
43
+
44
+ ## CIFAR100
45
+
46
+ * Opset: 18
47
+ * Source: PyTorch
48
+ * **Input**
49
+ * size: [N, 3, 224, 224]
50
+ * format: [N, C, H, W]
51
+ * preprocessing:
52
+ * ?
53
+ * **Output**
54
+ * size: [N, 100]
55
+
56
+
57
+ ## ImageNet1k
58
+
59
+ * Opset: 8
60
+ * Source: ?
61
+ * **Input**
62
+ * size: [N, 3, 224, 224]
63
+ * format: [N, C, H, W]
64
+ * preprocessing:
65
+ * ?
66
+ * **Output**
67
+ * size: [N, 1000]