完善资料让更多小伙伴认识你,还能领取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报错的原因?怎么解决?
975浏览 2评论
527浏览 1评论
648浏览 1评论
654浏览 1评论
使用sophon-demo_v0.1.8_dbb4632_20231116下面的YOLOv8中的yolov8_bmcv历程出现段错误的原因?
865浏览 1评论
/9
小黑屋| 手机版| Archiver| 电子发烧友 ( 湘ICP备2023018690号 )
GMT+8, 2025-12-10 07:33 , Processed in 0.410270 second(s), Total 39, Slave 32 queries .
Powered by 电子发烧友网
© 2015 bbs.elecfans.com
关注我们的微信
下载发烧友APP
电子发烧友观察
版权所有 © 湖南华秋数字科技有限公司
电子发烧友 (电路图) 湘公网安备 43011202000918 号 电信与信息服务业务经营许可证:合字B2-20210191

淘帖
641
