图像分割模型转换过程中报错

请问我在使用mobileunet训练模型时,转换模型过程中遇到如下问题,但是模型转换也完成了,在部署到板子上进行使用时运行结果与预期不符。我需要怎么做才能解决这个问题?输入图像大小为1*512*512*3。

模型训练时Reshape操作如下:

模型的一些信息如下:

感谢您使用地平线芯片算法工具链,最近我们在收集大家的满意度反馈,欢迎您填写问卷,详细情况可见:https://developer.horizon.ai/forumDetail/146177053698464782

你好,第一个图中的log提示不影响模型转换哈。

关于转换后模型上板精度有问题,建议您先参考一下这篇文章来解决:https://developer.horizon.ai/forumDetail/71036815603174578

您好我在docker中转换完成后进行推理报错:模型输入大小1*512*512,得到的时1204*2048,与模型输入要求的512*512不一致,脚本中没看出来需要修改哪里分辨率设置。

你好,请问你使用的OE版本是多少呀?

可以配置input_shape参数:

现在使用的是1.10.7,yaml文件中input_shape已经修改为与模型一致的大小了。

1.10.7是很早之前的版本了,X3派的板子,建议切换成OE2.5.2哈,获取链接:https://developer.horizon.ai/forumDetail/136488103547258769

使用最新版之后,如果还有这个问题,建议将onnx模型/config/校准数据压缩一下,通过百度网盘链接提供一下,我来复现一下你的问题哈

链接:百度网盘 请输入提取码

提取码:j10w

你好,我这边在OE2.5.2没有复现你的问题好哈:

提示一下:板端nv12输入时,它已经不再具有layout的概念了,需要准备nv12对应格式的数据

还有就是,文件中提供的校准数据准备的尺寸不太对

我的校准数据就是训练模型使用的数据512*512的,不对的话我需要修改成多大的呢?板端nv12输入我是调用例子中的bgr2nv12_opencv(image)这个函数,应该没有问题吧?

校准数据准备这块,确实是准备成512x512哈,是不是float32或者uint8在使用的时候不太对呢,可以看一下这篇文章来比对哈:https://developer.horizon.ai/forumDetail/118363912788935513,还有就是,建议您更新一下OE包版本哈

关于板端nv12的输入,建议参考OE包中的转换示例,看看它的nv12数据是如何准备的