RTMPose / config.json
fangmingguo's picture
Upload 13 files
993d81c verified
{
"model_type": "ONNX",
"npu_mode": "NPU3",
"input_shapes": "input:1x3x256x192",
"quant": {
"input_configs": [
{
"tensor_name": "DEFAULT",
"calibration_dataset": "./rtmpose_cali.tar",
"calibration_size": 100,
"calibration_mean": [
123.675,
116.28,
103.53
],
"calibration_std": [
58.395,
57.12,
57.375
]
}
],
"calibration_method": "MSE",
"layer_configs": [
{
"layer_names": [
"Conv_0",
"Conv_3",
"Conv_6",
"Conv_9",
"Conv_12",
"Conv_15",
"Conv_18",
"Conv_21",
"Conv_24",
"Conv_28",
"Conv_31",
"Conv_34",
"Conv_46",
"Conv_55",
"Conv_58",
"Conv_65",
"Conv_68",
"Conv_75",
"Conv_78",
"Conv_85",
"Conv_88",
"Conv_103",
"Conv_112",
"Conv_115",
"Conv_122",
"Conv_125",
"Conv_132",
"Conv_135",
"Conv_142",
"Conv_145",
"Conv_160",
"Conv_179",
"Conv_182",
"Conv_188",
"Conv_191",
"Conv_205",
"Conv_40",
"Conv_43",
"Conv_49",
"Conv_52",
"Conv_71",
"Conv_81",
"Conv_91",
"Conv_97",
"Conv_100",
"Conv_106",
"Conv_109",
"Conv_118",
"Conv_128",
"Conv_138",
"Conv_148",
"Conv_154",
"Conv_157",
"Conv_163",
"Conv_170",
"Conv_173",
"Conv_176",
"Conv_185",
"Conv_194",
"Conv_199",
"Conv_202",
"GlobalAveragePool_39",
"GlobalAveragePool_153",
"GlobalAveragePool_198",
"Concat_152",
"Concat_169",
"op_1:onnx.Silu",
"op_2:onnx.Silu",
"op_5:onnx.Silu",
"op_6:onnx.Silu",
"op_7:onnx.Silu",
"op_8:onnx.Silu",
"op_9:onnx.Silu",
"op_10:onnx.Silu",
"op_11:onnx.Silu",
"op_12:onnx.Silu",
"op_13:onnx.Silu",
"op_15:onnx.Silu",
"op_16:onnx.Silu",
"op_17:onnx.Silu",
"op_18:onnx.Silu",
"op_19:onnx.Silu",
"op_20:onnx.Silu",
"op_21:onnx.Silu",
"op_23:onnx.Silu",
"op_25:onnx.Silu",
"op_26:onnx.Silu",
"op_27:onnx.Silu",
"op_29:onnx.Silu",
"op_30:onnx.Silu",
"op_34:onnx.Silu",
"op_49:onnx.Silu",
"op_50:onnx.Silu",
"op_52:onnx.Silu",
"op_54:onnx.Silu",
"op_56:onnx.Silu",
"op_57:onnx.Silu",
"op_58:onnx.Silu",
"Add_64",
"Add_74",
"Add_84",
"Add_94",
"Add_121",
"Add_131",
"Add_141",
"Add_151",
"MaxPool_166",
"MaxPool_167",
"MaxPool_168",
"Reshape_212",
"Split_233",
"Unsqueeze_234",
"Split_237",
"Squeeze_238",
"Squeeze_239",
"Transpose_240"
],
"data_type": "U16"
},
{
"layer_names": [
"Conv_61",
"op_32:onnx.Silu",
"op_43:onnx.Silu",
"op_53:onnx.Silu",
"op_4:onnx.Silu",
"op_14:onnx.Silu",
"op_22:onnx.Silu",
"op_33:onnx.Silu",
"op_35:onnx.Silu",
"op_37:onnx.Silu",
"op_38:onnx.Silu",
"op_39:onnx.Silu",
"op_40:onnx.Silu",
"hs_replace_1_Mul",
"hs_replace_1_Add",
"hs_replace_1_Clip",
"Mul_42",
"Concat_95",
"GlobalAveragePool_96",
"hs_replace_2_Mul",
"hs_replace_2_Add",
"hs_replace_2_Clip",
"Mul_99",
"hs_replace_3_Mul",
"hs_replace_3_Add",
"hs_replace_3_Clip",
"Mul_156",
"hs_replace_4_Mul",
"hs_replace_4_Add",
"hs_replace_4_Clip",
"Mul_201",
"op_66:onnx.RMSNormalization",
"op_61:onnx.FullyConnected",
"Mul_249",
"op_67:onnx.RMSNormalization",
"op_62:onnx.FullyConnected",
"op_36:onnx.Silu",
"Mul_235",
"Add_236",
"MatMul_241",
"op_68:onnx.Mul",
"Relu_244",
"Mul_245",
"MatMul_246",
"Mul_247",
"op_63:onnx.FullyConnected",
"Add_250",
"op_64:onnx.FullyConnected",
"op_65:onnx.FullyConnected",
"op_48:onnx.Silu",
"op_41:onnx.Silu",
"op_42:onnx.Silu",
"op_44:onnx.Silu",
"op_45:onnx.Silu",
"op_46:onnx.Silu",
"op_47:onnx.Silu",
"Add_27",
"Add_37",
"op_24:onnx.Silu",
"Concat_38",
"op_28:onnx.Silu",
"Concat_197",
"op_3:onnx.Silu",
"op_55:onnx.Silu",
"op_31:onnx.Silu",
"op_51:onnx.Silu"
],
"data_type": "FP32",
"weight_data_type": "FP32",
"output_data_type": "FP32"
}
],
"precision_analysis": true,
"precision_analysis_method": "EndToEnd"
},
"input_processors": [
{
"tensor_name": "DEFAULT",
"tensor_format": "RGB",
"src_format": "BGR",
"src_dtype": "U8",
"src_layout": "NHWC"
}
],
"compiler": {
"check": 0
}
}