feat: initial HSAP platform
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>
This commit is contained in:
8
algorithms/lane_ufld/code/UFLD/demo_onnx.py
Executable file
8
algorithms/lane_ufld/code/UFLD/demo_onnx.py
Executable file
@@ -0,0 +1,8 @@
|
||||
import onnx
|
||||
# 加载模型
|
||||
model = onnx.load('./model/tusimple_18.onnx')
|
||||
# 检查模型格式是否完整及正确
|
||||
onnx.checker.check_model(model)
|
||||
# 获取输出层,包含层名称、维度信息
|
||||
output = model.graph.output
|
||||
print(output)
|
||||
Reference in New Issue
Block a user