Toybrick

标题: 用RK1808S0读取pytorch模型，输入图像为单通道图像，报错 [打印本页]

作者: 小二黑 时间: 2021-5-18 17:59
标题: 用RK1808S0读取pytorch模型，输入图像为单通道图像，报错
运行环境：RK1808S0，win10，pytorch=1.1.0，rknn=1.4，
错误：rknn.load_pytorch()加载模型的时候报错：TypeError: int() argument must be a string, a bytes-like object or a number, not 'NoneType'
主要问题：
1）.pt文件是由.pth文件转换来的，.pth->.pt是不是需要在GPU环境下？还是在CPU环境下？
2）模型推理使用的是单通道图像，使用rknn.config()函数怎样设置参数呀？channel_mean_value和reorder_channel改怎样设置？目前channel_mean_value=‘0 255’
3）rknn.load_pytorch()函数中的input_size_list改怎样设置？目前设置为：input_size_list = [[1, 240, 320]]

作者: bobby_jiang 时间: 2021-5-19 09:42
本帖最后由 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请多看看文档。

作者: 小二黑 时间: 2021-5-19 10:38

bobby_jiang 发表于 2021-5-19 09:42
升级到rknn toolkit v1.6.0
1. 使用CPU版本即可
2. 单通道图像，在config接口设置mean_values=[[0]], std_v ...

我昨天这样设置了，还是不行，依然是TypeError: int() argument must be a string, a bytes-like object or a number, not 'NoneType'

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