问题描述:
1. yolov5模型pt转bmodel可以识别出结果。(转化成功,结果正确)
2. yolov5模型pt转onnx转bmodel可以无法识别出结果。(转化成功,结果没有)
配置:
1. 推理硬件:质算盒SE5,芯片BM1684。
2. SDK: v2.7.0
代码:
1. 模型来源yolov5官方:
https://github.com/ultralytics/yolov5
2. pt转onnx代码:
dummy_input = input
dummy_output = model(dummy_input)
torch.onnx.export(model,
dummy_input,
"test_cnn.onnx",
opset_version=12, # opset version must be 12
example_outputs = dummy_output,
)
3. onnx转化bmodel代码:
python3 -m bmnetp --model models/yolov5s.torchscript_bs1.pt --shapes [1,3,640,640] --net_name yolov5 --opt 2 --dyn False --outdir ./ --target BM1684 --cmp False --mode compile --v 3