Skip to content

Commit 698fc22

Browse files
author
Anna Grebneva
authored
Updated efficientnet-b0-pytorch model (#3336)
1 parent 5b110c2 commit 698fc22

File tree

4 files changed

+11
-11
lines changed

4 files changed

+11
-11
lines changed

models/public/efficientnet-b0-pytorch/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@ The model output for `efficientnet-b0-pytorch` is the typical object classifier
2525

2626
| Metric | Original model | Converted model |
2727
| ------ | -------------- | --------------- |
28-
| Top 1 | 76.91% | 76.91% |
29-
| Top 5 | 93.21% | 93.21% |
28+
| Top 1 | 77.70% | 77.70% |
29+
| Top 5 | 93.52% | 93.52% |
3030

3131
## Input
3232

models/public/efficientnet-b0-pytorch/accuracy-check.yml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -33,11 +33,11 @@ models:
3333
- name: accuracy@top1
3434
type: accuracy
3535
top_k: 1
36-
reference: 0.7691
36+
reference: 0.777
3737
- name: accuracy@top5
3838
type: accuracy
3939
top_k: 5
40-
reference: 0.9321
40+
reference: 0.9352
4141

4242
- name: efficientnet-b0-pytorch
4343

@@ -65,8 +65,8 @@ models:
6565
- name: accuracy@top1
6666
type: accuracy
6767
top_k: 1
68-
reference: 0.7691
68+
reference: 0.777
6969
- name: accuracy@top5
7070
type: accuracy
7171
top_k: 5
72-
reference: 0.9321
72+
reference: 0.9352

models/public/efficientnet-b0-pytorch/model.yml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -50,10 +50,10 @@ files:
5050
original_source: https://github.com/rwightman/gen-efficientnet-pytorch/raw/a36e2b2cd1bd122a508a6fffeaa7606890f8c882/gen_efficientnet/__init__.py
5151
source: https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2022.1/efficientnet-b0-pytorch/__init__.py
5252
- name: efficientnet-b0.pth
53-
size: 21376958
54-
checksum: f44c3ff7ceeeda4ccf95a6720483cf2bee89f2db59d47e8349d2a7d46024e006b76e92105b6723060b898505312e70ff
55-
original_source: https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_b0-d6904d92.pth
56-
source: https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2022.1/efficientnet-b0-pytorch/efficientnet_b0-d6904d92.pth
53+
size: 21376743
54+
checksum: f6926743bf2213489f95c3b4bc6edc358ed9623780306e329ac8184ae36114237d98aa772761c180608316731db8e586
55+
original_source: https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_b0_ra-3dd342df.pth
56+
source: https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2022.2/efficientnet-b0-pytorch/efficientnet_b0_ra-3dd342df.pth
5757
conversion_to_onnx_args:
5858
- --model-path=$dl_dir
5959
- --model-name=efficientnet_b0

models/public/index.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -332,7 +332,7 @@ You can download models and convert them into OpenVINO™ IR format (\*.xml + \*
332332
| CaffeNet | Caffe\* | [caffenet](./caffenet/README.md) | 56.714%/79.916% | 1.5 | 60.965 |
333333
| DenseNet 121 | Caffe\*<br>TensorFlow\*| [densenet-121](./densenet-121/README.md)<br>[densenet-121-tf](./densenet-121-tf/README.md)| 74.42%/92.136%<br>74.46%/92.13%| 5.723~5.7287 | 7.971 |
334334
| DLA 34 | PyTorch\* | [dla-34](./dla-34/README.md) | 74.64%/92.06% | 6.1368 | 15.7344 |
335-
| EfficientNet B0 | TensorFlow\*<br>PyTorch\* | [efficientnet-b0](./efficientnet-b0/README.md)<br>[efficientnet-b0-pytorch](./efficientnet-b0-pytorch/README.md) | 75.70%/92.76%<br>76.91%/93.21% | 0.819 | 5.268 |
335+
| EfficientNet B0 | TensorFlow\*<br>PyTorch\* | [efficientnet-b0](./efficientnet-b0/README.md)<br>[efficientnet-b0-pytorch](./efficientnet-b0-pytorch/README.md) | 75.70%/92.76%<br>77.70%/93.52% | 0.819 | 5.268 |
336336
| EfficientNet V2 B0 | PyTorch\* | [efficientnet-v2-b0](./efficientnet-v2-b0/README.md) | 78.36%/94.02% | 1.4641 | 7.1094 |
337337
| EfficientNet V2 Small | PyTorch\* | [efficientnet-v2-s](./efficientnet-v2-s/README.md) | 84.29%/97.26% | 16.9406 | 21.3816 |
338338
| HBONet 1.0 | PyTorch\* | [hbonet-1.0](./hbonet-1.0/README.md) | 73.1%/91.0% | 0.6208 | 4.5443 |

0 commit comments

Comments
 (0)