Toybrick

onnx转rknn 非yolov5模型的mean_values和std_values如何设置

hihihi1

注册会员

积分
67
楼主
发表于 2022-3-8 15:52:59    查看: 2626|回复: 1 | [复制链接]    打印 | 只看该作者
rknn-toolkit-v1.7.1 → examples → onnx → resnet50v2 下的test.py中
  1. # pre-process config
  2. print('--> Config model')
  3. rknn.config(mean_values=[[123.675, 116.28, 103.53]], std_values=[[58.82, 58.82, 58.82]], reorder_channel='0 1 2')
  4. print('done')
复制代码
问题1:
    mean_values=[[123.675, 116.28, 103.53]], std_values=[[58.82, 58.82, 58.82]]   请问mean_values和std_values的值是如何得出的?
问题2:
    我用旷视的YOLOX生成了onnx,转rknn的时候,不知道mean_values和std_values如何设置? 或者说例子中的mean_values和std_values其他模型也可以通用?
回复

使用道具 举报

jefferyzhang

版主

积分
12939
沙发
发表于 2022-3-8 16:08:12 | 只看该作者
这都是训练时候针对训练集的归一化参数,请详细阅读论文
回复

使用道具 举报

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

本版积分规则

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


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