完善资料让更多小伙伴认识你,还能领取20积分哦, 立即完善>
问题描述:
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 |
|
相关推荐
|
|
只有小组成员才能发言,加入小组>>
H264硬编码CVI_VENC_StartRecvFrame报错的原因?怎么解决?
363浏览 2评论
使用bm-smi查看信息时发现没有程序运行但是设备内存被占用是怎么回事?
211浏览 1评论
232浏览 1评论
SE5盒子如何修改IP?执行bm_set_ip提示没有该命令?
215浏览 1评论
258浏览 1评论
小黑屋| 手机版| Archiver| 电子发烧友 ( 湘ICP备2023018690号 )
GMT+8, 2024-8-20 03:02 , Processed in 0.639314 second(s), Total 42, Slave 36 queries .
Powered by 电子发烧友网
© 2015 www.ws-dc.com
关注我们的微信
下载发烧友APP
电子发烧友观察
版权所有 © 湖南华秋数字科技有限公司
电子发烧友 (电路图) 湘公网安备 43011202000918 号 电信与信息服务业务经营许可证:合字B2-20210191 工商网监 湘ICP备2023018690号