tensorRT_Pro icon indicating copy to clipboard operation
tensorRT_Pro copied to clipboard

Published 20 hours ago • verified publisher icon shouxieai

导出yolov5s的onnx模型后 执行make yolo -j32报错

Open ZZY-666 opened this issue 3 years ago • 1 comments

您好 这是什么原因呢？ image

ZZY-666 avatar Mar 07 '22 23:03 ZZY-666

你需要把torch.onnx.export导出时的dynamic_axes，去掉宽度和高度，只保留batch

shouxieai avatar Mar 08 '22 13:03 shouxieai