befozg/PortraitTransfer

1import onnx2from onnxruntime import InferenceSession3import numpy as np4import torch5 6 7def execute_onnx_model(x, onnx_model) -> None:8    sess = InferenceSession(onnx_model.SerializeToString(), providers=[9                            'AzureExecutionProvider', 'CPUExecutionProvider'])10    out = sess.run(None, {'input.1': x.numpy().astype(np.float32)})[0]11 12    return out13