Yuantao Feng
commited on
Commit
·
2074f99
1
Parent(s):
aebf101
update ONNX opset version of PP-ResNet50 to 11 (#37)
Browse files
benchmark/config/image_classification_ppresnet.yaml
CHANGED
@@ -16,5 +16,5 @@ Benchmark:
|
|
16 |
|
17 |
Model:
|
18 |
name: "PPResNet"
|
19 |
-
modelPath: "models/image_classification_ppresnet/
|
20 |
-
labelPath: "models/image_classification_ppresnet/imagenet_labels.txt"
|
|
|
16 |
|
17 |
Model:
|
18 |
name: "PPResNet"
|
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
CHANGED
@@ -21,7 +21,7 @@ def str2bool(v):
|
|
21 |
|
22 |
parser = argparse.ArgumentParser(description='Deep Residual Learning for Image Recognition (https://arxiv.org/abs/1512.03385, https://github.com/PaddlePaddle/PaddleHub)')
|
23 |
parser.add_argument('--input', '-i', type=str, help='Path to the input image.')
|
24 |
-
parser.add_argument('--model', '-m', type=str, default='
|
25 |
parser.add_argument('--label', '-l', type=str, default='./imagenet_labels.txt', help='Path to the dataset labels.')
|
26 |
args = parser.parse_args()
|
27 |
|
@@ -39,4 +39,4 @@ if __name__ == '__main__':
|
|
39 |
result = model.infer(image)
|
40 |
|
41 |
# Print result
|
42 |
-
print('label: {}'.format(result))
|
|
|
21 |
|
22 |
parser = argparse.ArgumentParser(description='Deep Residual Learning for Image Recognition (https://arxiv.org/abs/1512.03385, https://github.com/PaddlePaddle/PaddleHub)')
|
23 |
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_2022jan.onnx', help='Path to the model.')
|
25 |
parser.add_argument('--label', '-l', type=str, default='./imagenet_labels.txt', help='Path to the dataset labels.')
|
26 |
args = parser.parse_args()
|
27 |
|
|
|
39 |
result = model.infer(image)
|
40 |
|
41 |
# Print result
|
42 |
+
print('label: {}'.format(result))
|