RF-DETR / onnx /config.json
LosReturn's picture
Upload folder using huggingface_hub
67c84d4 verified
Raw
History Blame Contribute Delete
1.17 kB
{
"input": "./onnx/sim.onnx",
"output_dir": "./out",
"output_name": "detr.axmodel",
"model_type": "ONNX",
"target_hardware": "AX650",
"npu_mode": "NPU3",
"quant": {
"input_configs": [
{
"tensor_name": "DEFAULT",
"calibration_dataset": "./asserts/imagenet-calib.tar",
"calibration_size": 128,
"calibration_mean": [123.675, 116.28, 103.53],
"calibration_std": [58.395, 57.12, 57.375],
"calibration_format": "Image"
}
],
"calibration_method": "MinMax",
"precision_analysis": true,
"precision_analysis_method": "EndToEnd",
"precision_analysis_mode": "NPUBackend",
"conv_bias_data_type": "FP32",
"layer_configs": [
{
"start_tensor_names": ["DEFAULT"],
"end_tensor_names": ["DEFAULT"],
"data_type": "U16"
}
]
},
"input_processors": [
{
"tensor_name": "DEFAULT",
"tensor_format": "RGB",
"tensor_layout": "NCHW",
"src_dtype": "U8",
"src_layout": "NHWC",
"mean": [123.675, 116.28, 103.53],
"std": [58.395, 57.12, 57.375]
}
],
"compiler": {
"enable_slice_mode": true
}
}