| { |
| "input": "./Restormer_real_denoising_224x224_sim.onnx", |
| "output_dir": "./Restormer", |
| "output_name": "Restormer_real_denoising_224x224_sim.axmodel", |
| "model_type": "ONNX", |
| "target_hardware": "AX650", |
| "npu_mode": "NPU3", |
| "quant": { |
| "input_configs": [ |
| { |
| "tensor_name": "input", |
| "calibration_dataset": "./SIDD.tar.gz", |
| "calibration_size": 32, |
| "calibration_mean": [0.0, 0.0, 0.0], |
| "calibration_std": [255.0, 255.0, 255.0] |
| } |
| ], |
| "calibration_method": "MinMax", |
| "precision_analysis": true, |
| "transformer_opt_level":1, |
| "precision_analysis_method": "EndToEnd", |
| "precision_analysis_mode": "NPUBackend", |
| "device": "cuda:1", |
| "layer_configs": [ |
| { |
| "start_tensor_names": ["DEFAULT"], |
| "end_tensor_names": ["DEFAULT"], |
| "data_type": "U16" |
| } |
| ] |
| }, |
| "input_processors": [ |
| { |
| "tensor_name": "input", |
| "tensor_format": "RGB", |
| "tensor_layout": "NCHW", |
| "src_format": "BGR", |
| "src_dtype": "U8", |
| "src_layout": "NCHW", |
| "csc_mode": "NoCSC" |
| } |
| ], |
| "compiler": { |
| "check": 2 |
| } |
| } |
|
|
|
|
