Skip to content

Commit 1147b6c

Browse files
authored
update ONNX opset version of PP-ResNet50 to 11 (#37)
1 parent ba91a3b commit 1147b6c

File tree

4 files changed

+7
-7
lines changed

4 files changed

+7
-7
lines changed

benchmark/config/image_classification_ppresnet.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,5 +16,5 @@ Benchmark:
1616

1717
Model:
1818
name: "PPResNet"
19-
modelPath: "models/image_classification_ppresnet/image_classification_ppresnet50_2021oct.onnx"
20-
labelPath: "models/image_classification_ppresnet/imagenet_labels.txt"
19+
modelPath: "models/image_classification_ppresnet/image_classification_ppresnet50_2022jan.onnx"
20+
labelPath: "models/image_classification_ppresnet/imagenet_labels.txt"

models/image_classification_ppresnet/demo.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ def str2bool(v):
2121

2222
parser = argparse.ArgumentParser(description='Deep Residual Learning for Image Recognition (https://arxiv.org/abs/1512.03385, https://github.com/PaddlePaddle/PaddleHub)')
2323
parser.add_argument('--input', '-i', type=str, help='Path to the input image.')
24-
parser.add_argument('--model', '-m', type=str, default='image_classification_ppresnet50_2021oct.onnx', help='Path to the model.')
24+
parser.add_argument('--model', '-m', type=str, default='image_classification_ppresnet50_2022jan.onnx', help='Path to the model.')
2525
parser.add_argument('--label', '-l', type=str, default='./imagenet_labels.txt', help='Path to the dataset labels.')
2626
args = parser.parse_args()
2727

@@ -39,4 +39,4 @@ def str2bool(v):
3939
result = model.infer(image)
4040

4141
# Print result
42-
print('label: {}'.format(result))
42+
print('label: {}'.format(result))

models/image_classification_ppresnet/image_classification_ppresnet50_2021oct.onnx

Lines changed: 0 additions & 3 deletions
This file was deleted.
Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
version https://git-lfs.github.com/spec/v1
2+
oid sha256:ad5486b0de6c2171ea4d28c734c2fb7c5f64fcdbd97180a0ef515cf4b766a405
3+
size 102567035

0 commit comments

Comments
 (0)