import numpy as np
import os
import onnx
import onnxruntime as ort
rawFile = np.fromfile("f_y_x_y.bgrp",dtype=np.float32)
img_input = rawFile.reshape(1,3,144,1024)
sess = ort.InferenceSession("float_model.onnx")
input_name = sess.get_inputs()[0].name
result = sess.run([],{input_name: img_input})
outputs_name = [x.name for x in sess.get_outputs()]
print (outputs_name)
print(type(result))
result[0].tofile(outputs_name[0]+".bin")
result[1].tofile(outputs_name[1]+".bin")
result[2].tofile(outputs_name[2]+".bin")
result[3].tofile(outputs_name[3]+".bin")
result[4].tofile(outputs_name[4]+".bin")
result[5].tofile(outputs_name[5]+".bin")
result[6].tofile(outputs_name[6]+".bin")