Chengfang Lu
7c43b44c57
Huaxu Sentinel Active Safety Platform with embedded algorithm code, Docker Compose setup, and vendored dataset scaffolds for clone-and-run. Co-authored-by: Cursor <cursoragent@cursor.com>
8 lines
246 B
Python
Executable File
8 lines
246 B
Python
Executable File
import onnx
|
|
# 加载模型
|
|
model = onnx.load('./model/tusimple_18.onnx')
|
|
# 检查模型格式是否完整及正确
|
|
onnx.checker.check_model(model)
|
|
# 获取输出层，包含层名称、维度信息
|
|
output = model.graph.output
|
|
print(output) |