增加部署代码
This commit is contained in:
4
onnx_val/valonnx.py
Normal file
4
onnx_val/valonnx.py
Normal file
@ -0,0 +1,4 @@
|
||||
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}")
|
||||
Reference in New Issue
Block a user