4 lines
200 B
Python
4 lines
200 B
Python
|
import onnx
|
||
|
|
model = onnx.load("/home/hx/yolo/onnx_val/yolo11n-obb.onnx")
|
||
|
|
for i, output in enumerate(model.graph.output):
|
||
|
|
print(f"Output {i}: {output.name}, shape: {output.type.tensor_type.shape}")
|