MIndyolo用310推理时出现错误

问题描述

型号：香橙派aipro  芯片310b4  cann7.0.0  mindspore 2.2.14  mindspore lite 2.2.14 python3.9.2

根据readme进行环境配置

现在进行完用cpu来模型转换（因为用Acend只支持910），成功转换为yolov5n.mindir

在运行推理时出现错误：

python deploy/predict.py --model_type Lite --model_path ./yolov5n.mindir --config ./configs/yolov5/yolov5n.yaml --image_path ./coco/image/val2017/image.jpg

解答方案

deploy/predict.py代码中默认为Ascend推理，所以你CPU导出的mindir用Ascend推理就会报错。

parser.add_argument("--device_target", type=str, default="Ascend", help="device target, Ascend/GPU/CPU")

用如下操作正常。

python ./deploy/export.py --config ./configs/yolov5/yolov5n.yaml --weight yolov5n_300e_mAP273-9b16bd7b.ckpt --per_batch_size 1 --file_format MINDIR --device_target CPU

python deploy/predict.py --model_type MindIR --model_path ./yolov5n.mindir --config ./configs/yolov5/yolov5n.yaml --device_target CPU --image_path .xxxx.jpg