1import onnxruntime as ort
2import numpy as np
3
4sess = ort.InferenceSession("data/onnx/face_parsing.farl.celebm.main_ema_181500_jit.onnx", providers=["CPUExecutionProvider"])
5input_name = sess.get_inputs()[0].name
6
7dummy = np.zeros((1, 3, 448, 448), dtype=np.float32)
8logits, aux = sess.run(None, {input_name: dummy})
9print(logits.shape, aux.shape)