Alexander Suslov
commited on
Commit
·
64482e8
1
Parent(s):
3a2a474
added tf_model
Browse files- README.md +2 -2
- mobilenet_v2_imagenette.onnx +2 -2
- openvino_model.tgz +2 -2
- pytorch_model.bin +2 -2
- tf_model.h5 +3 -0
README.md
CHANGED
@@ -3,7 +3,7 @@ license: apache-2.0
|
|
3 |
---
|
4 |
MobileNet V2 model from Torchvision fine-tuned for Imagenette dataset. Checkpoint trained for 100 epoches using https://github.com/alexsu52/mobilenet_v2_imagenette.
|
5 |
|
6 |
-
Top-1 accuracy is
|
7 |
|
8 |
The main intent is to use it in samples and demos for model optimization. Here is the advantages:
|
9 |
- Imagenette can automatically downloaded and is quite small.
|
@@ -22,4 +22,4 @@ model = models.mobilenet_v2(num_classes=IMAGENETTE_CLASSES)
|
|
22 |
|
23 |
checkpoint = torch.hub.load_state_dict_from_url(CHECKPOINT_URL, progress=False)
|
24 |
model.load_state_dict(checkpoint['state_dict'])
|
25 |
-
```
|
|
|
3 |
---
|
4 |
MobileNet V2 model from Torchvision fine-tuned for Imagenette dataset. Checkpoint trained for 100 epoches using https://github.com/alexsu52/mobilenet_v2_imagenette.
|
5 |
|
6 |
+
Top-1 accuracy is 98.64%.
|
7 |
|
8 |
The main intent is to use it in samples and demos for model optimization. Here is the advantages:
|
9 |
- Imagenette can automatically downloaded and is quite small.
|
|
|
22 |
|
23 |
checkpoint = torch.hub.load_state_dict_from_url(CHECKPOINT_URL, progress=False)
|
24 |
model.load_state_dict(checkpoint['state_dict'])
|
25 |
+
```
|
mobilenet_v2_imagenette.onnx
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c6eab06589151102f6a0270bf860d08098a028f9849b9c628aa3d0c67f20235f
|
3 |
+
size 9456587
|
openvino_model.tgz
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:23ca9dfc28c2cda58bf8219fcdd5204feb7b33777f71ac87c42c0dc639bf16a0
|
3 |
+
size 8258642
|
pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:32971bd064c48216991208b4dbabd0b2065ab003d2a0cec1f1a4cf3cb3d25ad2
|
3 |
+
size 18172593
|
tf_model.h5
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c21120928145ffebbdc7629daa5ea04026ab52d7236008bc0ba908ad1a4ebe97
|
3 |
+
size 9476336
|