Skip to content

Commit a019325

Browse files
committed
add resnet_v2_152
Signed-off-by: xavier dupré <[email protected]>
1 parent 7deccdb commit a019325

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed
Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
# SPDX-License-Identifier: Apache-2.0
2+
import os
3+
import numpy
4+
from _tools import generate_random_images, measure_time, download_model, convert_model, benchmark
5+
6+
url = "https://tfhub.dev/google/imagenet/resnet_v2_152/classification/5?tf-hub-format=compressed"
7+
dest = "tf-resnet_v2_152"
8+
name = "resnet_v2_152"
9+
opset = 13
10+
onnx_name = os.path.join(dest, "%s-%d.onnx" % (name, opset))
11+
12+
imgs = generate_random_images(shape=(1, 224, 224, 3))
13+
14+
benchmark(url, dest, onnx_name, opset, imgs)

0 commit comments

Comments
 (0)