(model_url, name)
| 61 | |
| 62 | def get_network(name, dtype, input_shape=(1, 3, 224, 224)): |
| 63 | def download_model(model_url, name): |
| 64 | model_path = download_testdata(model_url, name + ".onnx", module="onnx") |
| 65 | onnx_model = onnx.load(model_path) |
| 66 | |
| 67 | shape_dict = {"x": input_shape} |
| 68 | mod = from_onnx(onnx_model, shape_dict) |
| 69 | return mod |
| 70 | |
| 71 | def create_model(name): |
| 72 | if "vgg11" == name: |
no test coverage detected
searching dependent graphs…