我是yolov5s的模型,输入维度是1*3*640*640,输出维度是1*25200*85,
使用下述转换的模型。
推理时会报错,input size是1228800和op_size的4915232没对齐

然后我就手动分配input buffer的内存大小为4915232,然后就会报is_input:0 blob size:1 user !=op_size

目前试用了各类atc转换代码,都是这两个错误,不知道怎么解决。
onnx模型和 soc_version=Ascend310B4 条件下转出的om文件在下述链接。
链接:https://pan.baidu.com/s/13WQHuZMDjFdCzuv2zL9PNw?pwd=udo2
提取码:udo2
--来自百度网盘超级会员V5的分享
我是yolov5s的模型,输入维度是1*3*640*640,输出维度是1*25200*85,
使用下述转换的模型。
推理时会报错,input size是1228800和op_size的4915232没对齐
然后我就手动分配input buffer的内存大小为4915232,然后就会报is_input:0 blob size:1 user !=op_size
目前试用了各类atc转换代码,都是这两个错误,不知道怎么解决。
onnx模型和 soc_version=Ascend310B4 条件下转出的om文件在下述链接。
链接:https://pan.baidu.com/s/13WQHuZMDjFdCzuv2zL9PNw?pwd=udo2
提取码:udo2
--来自百度网盘超级会员V5的分享