晟腾A200DK开发板移植Yolov8(3)

移植分为pt->onnx->om三步,我们今天先来讲解第二步onnx->om

1)设置环境变量(请确认install_path路径是否正确);

 bash . /usr/local/Ascend/ascend-toolkit/set_env.sh 

上述示例为Ascend-cann-toolkit开发套件包默认安装路径,请根据实际安装路径修改。

2)执行以下命令转换yolov8n模型:

atc \
--output_type=FP32 \
--input_shape="images:1,3,640,640"  \
--input_format=NCHW \
--output="yolov8n" \
--soc_version=Ascend310B1 \
--insert_op_conf=insert_op.cfg \
--framework=5 \
--model="yolov8n.onnx" \

atc参数含义:

atc: 命令行工具,用于将模型转换为适用于芯片处理器的模型格式。

--input_shape="images:1,3,640,640":每次推理一张图像,3通道,图像大小为640*640。input_shape需要与image_list的量化校准文件数据匹配。

--input_format=NCHW: 指定模型输入的数据格式为NCHW或NHWC

N:每次推理输入图像的数目。

C:Channels,图像通道,例如彩色RGB图像的Channels

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值