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/image_classification_ppresnet50_2021oct.onnx"
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='image_classification_ppresnet50_2021oct.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,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))