YOLOV5模型转换后,推理结果出错
收藏回复举报
YOLOV5模型转换后,推理结果出错
t('forum.solved') 已解决
新人帖
发表于2024-04-10 17:41:14
0 查看
good.jpg

你好,我用yolov5s.pt模型对行人数据集进行训练(nc=1),测试结果正常;将其转换为om模型后;

软件包:Ascend-cann-nnrt_6.2.RC2_linux-aarch64.run

具体命令

atc --model=best_1000.onnx --framework=5 --output=best_new_1 --input_format=NCHW --log=error --soc_version=Ascend310B1 --input_shape="images:1,3,640,640"

在jupyter上面运行yolov5推理代码,运行结果如下:请问一下能看出这是什么问题吗?或者我还需要提供什么材料以供你们参考。

bad.jpg

我要发帖子