File size: 604 Bytes
477da44 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
models:
name: resnet_18 # 模型输出名称
platform: pytorch # 原始模型使用的框架
model_file_path: ./resnet18_i8.pt # 原模型路径
subgraphs: # 描述输入输出shape等信息
input_size_list:
- 1, 3, 224, 224
quantize: False # QAT已经量化过，不需要再量化
configs:
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转换
|