×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

【MagicMind—yolov7模型移植问题】 xw2024-02-01 09:35:49 回复 查看 技术答疑 使用求助
【MagicMind—yolov7模型移植问题】
分享到:

【寒武纪硬件产品型号】:MLU370-S4

【MagicMind版本号】

magicmind_1.3.0-1_ubuntu20.04

【出错信息】必填*:yolov7模型移植成功后目标检测多框

【操作步骤】选填:

python gen_model.py --precision force_float16 \

--input_dims 1 3 640 640 \

--dynamic_shape true \

--magicmind_model ./model \

--pytorch_pt ./yolov7_traced_model.pt \

--input_layout NHWC \

--computation_preference fast   \

--type64to32_conversion "true"  \

--conv_scale_fold "true"   \

--dim_range_min 1 3 640 640 \

--dim_range_max 64 3 640 640 \

--means 0 0 0 \

--vars  65025 65025 65025

最终执行这个脚本进行转换
25.jpg.jpg


版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭