File size: 817 Bytes
477da44 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
models:
name: resnet_18 # 模型输出名称
platform: pytorch # 原始模型使用的框架
model_file_path: ./resnet18.pt # 原模型路径
subgraphs: # 描述输入输出shape等信息
input_size_list:
- 1, 3, 224, 224
quantize: true # 是否量化
dataset: ./dataset.txt # 量化dataset文件路径（相对yml路径）
configs:
quantized_dtype: asymmetric_quantized-8 # 量化类型
mean_values: [123.675, 116.28, 103.53] # rknn.config的mean_values参数
std_values: [58.395, 58.395, 58.395] # rknn.config的std_values参数
quant_img_RGB2BGR: false # 不进行RGB2BGR转换
quantized_algorithm: normal # 量化算法
quantized_method: channel # 量化方法
|