rknn SSD demo修改camera输入图像的size报错

[复制链接] · 发表于 2021-7-1 16:34:14

本帖最后由 zehao 于 2021-7-1 17:05 编辑

尝试修改rknn ssd demo中java的代码。
将img size从300改成了416，为了适配自己的model，自己的model的Input size是416x416。

奇怪的是，为什么输出的图像是扭曲的呢。

修改的部分是将imagebufferqueue的INPUT_SIZE改为了416，在run_ssd()里面将读取到的texID转为inData存入input.buf。尝试将inData里面的数据保存到文件并且查看，发现图像扭曲。如果将参数改为300，保存到文件的图像是正常的。另外，尝试将INPUT_SIZE改为280（即小于300）,也会发生读取图像扭曲的情况。请问代码需要修改什么地方可以适配不同的INPUT_SIZE（既是image size也是model的input size）

只看该作者 · 发表于 2021-7-2 15:49:54

你得用resize把原始输入图像的尺寸从300，resize成416才行吧

只看该作者 · 发表于 2021-7-2 22:38:02

troy 发表于 2021-7-2 15:49
你得用resize把原始输入图像的尺寸从300，resize成416才行吧

直接修改INPUT_SIZE不可以吗。必须在cc code里面将输入的input image(*inData的buffer地址)resize成为416x416才可以吗？

我现在把camera的的图像改为了640x480可以正常写到buffer里面，拿到的图像也是正常的。然后在cc code里面将这个image resize成为416x416，直接硬转换的，也没有借用第三方图像处理的库。

rknn SSD demo修改camera输入图像的size报错

本帖子中包含更多资源