模型转换

导出模型（ONNX）

导出edsr onnx可以参考：https://github.com/sanghyun-son/EDSR-PyTorch/blob/master/src/main.py

在main.py加上如下代码，可以正常导出onnx：

model = model.to('cpu')
target_onnx_file = './edsr_baseline_x2_1.onnx'
dummy_input = torch.randn(1, 3, 1080, 1920)
idx_scale = 0
torch.onnx.export(model,
                 (dummy_input, idx_scale),
                  target_onnx_file,
                  export_params=True,
                  opset_version=11,
                  do_constant_folding=True,
                  dynamic_axes = {},
                  )
                                               
print(f"Export model onnx to {target_onnx_file} finished")

这里固定onnx输入尺寸为：1x3x1080x1920

动态onnx转静态

onnxsim edsr_baseline_x2_1.onnx  edsr_baseline_x2_1_sim.onnx --overwrite-input-shape=1,1,1080,1920

转换模型（ONNX -> Axera）

使用模型转换工具 Pulsar2 将 ONNX 模型转换成适用于 Axera 的 NPU 运行的模型文件格式 .axmodel，通常情况下需要经过以下两个步骤：

生成适用于该模型的 PTQ 量化校准数据集
使用 Pulsar2 build 命令集进行模型转换（PTQ 量化、编译），更详细的使用说明请参考 AXera Pulsar2 工具链指导手册

量化数据集

准备量化图片若张，打包成Image.zip

模型转换

修改配置文件

检查config.json 中 calibration_dataset 字段，将该字段配置的路径改为上一步下载的量化数据集存放路径

Pulsar2 build

参考命令如下：

pulsar2 build --input edsr_baseline_x2_1.onnx --config ./build_config_edsr.json --output_dir ./output --output_name edsr_baseline_x2_1.axmodel  --target_hardware AX650 --compiler.check 0

也可将参数写进json中，直接执行：
pulsar2 build --config ./build_config_edsr.json