提交 1147b6cd 编写于 作者: Y Yuantao Feng 提交者: GitHub

update ONNX opset version of PP-ResNet50 to 11 (#37)

上级 ba91a3b9
......@@ -16,5 +16,5 @@ Benchmark:
Model:
name: "PPResNet"
modelPath: "models/image_classification_ppresnet/image_classification_ppresnet50_2021oct.onnx"
labelPath: "models/image_classification_ppresnet/imagenet_labels.txt"
\ No newline at end of file
modelPath: "models/image_classification_ppresnet/image_classification_ppresnet50_2022jan.onnx"
labelPath: "models/image_classification_ppresnet/imagenet_labels.txt"
......@@ -21,7 +21,7 @@ def str2bool(v):
parser = argparse.ArgumentParser(description='Deep Residual Learning for Image Recognition (https://arxiv.org/abs/1512.03385, https://github.com/PaddlePaddle/PaddleHub)')
parser.add_argument('--input', '-i', type=str, help='Path to the input image.')
parser.add_argument('--model', '-m', type=str, default='image_classification_ppresnet50_2021oct.onnx', help='Path to the model.')
parser.add_argument('--model', '-m', type=str, default='image_classification_ppresnet50_2022jan.onnx', help='Path to the model.')
parser.add_argument('--label', '-l', type=str, default='./imagenet_labels.txt', help='Path to the dataset labels.')
args = parser.parse_args()
......@@ -39,4 +39,4 @@ if __name__ == '__main__':
result = model.infer(image)
# Print result
print('label: {}'.format(result))
\ No newline at end of file
print('label: {}'.format(result))
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册