移植分为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