Commit 707d86da authored by Ted Themistokleous's avatar Ted Themistokleous
Browse files

Replace cat photo with picture of my own bird, update script

Using photo of my own pet bird to beat copyright and not get sued.
parent fd88122c
......@@ -57,7 +57,7 @@ torch.onnx.export(
# Pre-processing for ResNet-50 Inferencing, from https://pytorch.org/hub/pytorch_vision_resnet/
resnet50.eval()
filename = 'cat.jpg' # change to your filename
filename = 'bird.jpg' # change to your filename
input_image = Image.open(filename)
preprocess = T.Compose([
......@@ -102,6 +102,6 @@ def run_sample(session, image_file, categories, inputs):
return ort_outputs
ort_output = run_sample(session_fp32, 'cat.jpg', categories, input_batch)
ort_output = run_sample(session_fp32, 'bird.jpg', categories, input_batch)
print("resnet50, time = {} ms".format(
format(sum(latency) * 1000 / len(latency), '.2f')))
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment