python 语法
模型转换流程:
1.初始化:
rknn = RKNN(verbose=True, verbose_file='./mobilenet_build.log')
verbose 决定是否在屏幕上显示详细日志信息。
如果设置了 verbose_file 参数并且 verbose 为 True,日志信息还将写入到指定的文
件中
2.模型 Config 配置:
rknn.config(
mean_values=[[103.94, 116.78, 123.68]],
std_values=[[58.82, 58.82, 58.82]],
quant_img_RGB2BGR=False,
target_platform='rk3566')
mean_values 和 std_values 用于设置输入的均值和归一化值。这些值在量化过程中使用,
且 C API 推理阶段图片不需再做均值和归一化值减小布署耗时。
quant_img_RGB2BGR 用于控制量化时加载量化校正图像时是否需要先进行 RGB 到
BGR 的转换,默认值为 False。该配置只在量化数据集时生效,实际部署模型时,模
型推理阶段不会生效,需要用户在输入前处理里预先处理好。注:
quant_img_RGB2BGR = True 时 模 型 的 推 理 顺 序 为 先 做 RGB2BGR 转 换 再 做
mean_values 和 std_values 操作,详细注意事项请见 10.3 章节。