Toybrick

用RK1808S0读取pytorch模型,输入图像为单通道图像,报错

bobby_jiang

高级会员

积分
540
楼主
发表于 2021-5-19 09:42:35 | 显示全部楼层
本帖最后由 bobby_jiang 于 2021-5-19 09:43 编辑

升级到rknn toolkit v1.6.0
1. 使用CPU版本即可
2. 单通道图像,在config接口设置mean_values=[[0]], std_values=[[255]]
3. 这么设没有错。input_size_list=[[C, H, W]],其中C通道数,H高,W宽。
怎么使用rknn和rknn toolkit请多看看文档。
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表