Skip to content

Commit

Permalink
update ONNX opset version of PP-ResNet50 to 11 (opencv#37)
Browse files Browse the repository at this point in the history
  • Loading branch information
fengyuentau authored Jan 7, 2022
1 parent ba91a3b commit 1147b6c
Show file tree
Hide file tree
Showing 4 changed files with 7 additions and 7 deletions.
4 changes: 2 additions & 2 deletions benchmark/config/image_classification_ppresnet.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -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"
modelPath: "models/image_classification_ppresnet/image_classification_ppresnet50_2022jan.onnx"
labelPath: "models/image_classification_ppresnet/imagenet_labels.txt"
4 changes: 2 additions & 2 deletions models/image_classification_ppresnet/demo.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()

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

# Print result
print('label: {}'.format(result))
print('label: {}'.format(result))

This file was deleted.

Git LFS file not shown

0 comments on commit 1147b6c

Please sign in to comment.