import onnxruntime as ort
import numpy as np

# ✅ 模型路徑（你指定的）
onnx_path = r"C:\Users\rd_de\kneron-mmsegmentation\work_dirs\kn_stdc1_in1k-pre_512x1024_80k_cityscapes\latest.onnx"

# 建立 ONNX session
session = ort.InferenceSession(onnx_path)

# 印出模型 input 相關資訊
input_name = session.get_inputs()[0].name
input_shape = session.get_inputs()[0].shape
print(f"✅ Input name: {input_name}")
print(f"✅ Input shape: {input_shape}")

# 建立假圖輸入 (float32, shape = [1, 3, 512, 1024])
dummy_input = np.random.rand(1, 3, 512, 1024).astype(np.float32)

# 執行推論
outputs = session.run(None, {input_name: dummy_input})

# 顯示模型輸出資訊
for i, output in enumerate(outputs):
    print(f"✅ Output {i}: shape = {output.shape}, dtype = {output.dtype}")