rknn 模型转换，是否支持YCbCr 格式的三通道直接输入

[复制链接] · 发表于 2020-4-8 11:43:18

你好，我想把FSRCNN 模型（超分辨率）跑在 rk3399pro开发板上，模型输入为 YCbCr 的 Y通道，我们的rknn 模型转换工具是否支持，感谢

只看该作者 · 发表于 2020-4-8 11:45:32

本帖最后由 jefferyzhang 于 2020-4-8 11:54 编辑

三通道/单通道输入都支持，是什么格式rknn又不在乎，只是模型运算的输入和输出而已

只看该作者 · 发表于 2020-4-8 11:54:13

jefferyzhang 发表于 2020-4-8 11:45
三通道/单通道输入都支持，是什么格式我们又不会在乎。。。

了解，YCbCr 输入的话，量化也是支持的？是吧

只看该作者 · 发表于 2020-4-8 12:00:57

本帖最后由 jefferyzhang 于 2020-4-8 14:34 编辑

ivanzheng 发表于 2020-4-8 11:54
了解，YCbCr 输入的话，量化也是支持的？是吧

都是无关的。你只要能送和你模型输入一样的数据，就可以做。
目前量化接口默认读图片是rgb数据，要么可以兼容方法把ycbcr当rgb存，要么推荐用tensorflow的训练量化量化完再转rknn