Toybrick

标题: rknn 模型转换，是否支持YCbCr 格式的三通道直接输入 [打印本页]

作者: ivanzheng 时间: 2020-4-8 11:43
标题: rknn 模型转换，是否支持YCbCr 格式的三通道直接输入
你好，我想把FSRCNN 模型（超分辨率）跑在 rk3399pro开发板上，模型输入为 YCbCr 的 Y通道，我们的rknn 模型转换工具是否支持，感谢

作者: jefferyzhang 时间: 2020-4-8 11:45
本帖最后由 jefferyzhang 于 2020-4-8 11:54 编辑

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

作者: ivanzheng 时间: 2020-4-8 11:54

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

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

作者: jefferyzhang 时间: 2020-4-8 12:00
本帖最后由 jefferyzhang 于 2020-4-8 14:34 编辑

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

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

欢迎光临 Toybrick (https://t.rock-chips.com/)