算能科技
直播中

张超

7年用户 1295经验值
私信 关注
[问答]

yolov5模型onnx转bmodel无法识别出结果如何解决?

问题描述:
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

      

更多回帖

发帖
×
20
完善资料,
赚取积分