初始化算子预加载文件示例
初始化算子预加载文件需与MxInitFromConfig接口配合使用。
{
"Operations": [
{
"name": "Multiply",
"preload_list": [
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "float;float",
"output_shape": "1,3,16,16",
"output_type": "float"
},
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "float16;float16",
"output_shape": "1,3,16,16",
"output_type": "float16"
},
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "uint8;uint8",
"output_shape": "1,3,16,16",
"output_type": "uint8"
},
{
"input_shape": "1, 3, 16, 16; 1, 3, 16, 16",
"input_type": "float;uint8",
"output_shape": "1, 3, 16, 16",
"output_type": "float",
"attr_name": "scale",
"attr_type": "double",
"attr_val": "1.5f"
}
]
},
{
"name": "Divide",
"preload_list": [
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "float;float",
"output_shape": "1,3,16,16",
"output_type": "float"
},
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "float16;float16",
"output_shape": "1,3,16,16",
"output_type": "float16"
},
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "uint8;uint8",
"output_shape": "1,3,16,16",
"output_type": "uint8"
}
]
},
{
"name": "Tile",
"preload_list": [
{
"input_shape": "1,16,16,1",
"input_type": "float",
"output_shape": "1,16,16,3",
"output_type": "float"
}
]
},
{
"name": "Abs",
"preload_list": [
{
"input_shape": "1,3,16,16",
"input_type": "float",
"output_shape": "1,3,16,16",
"output_type": "float"
},
{
"input_shape": "1,3,16,16",
"input_type": "float16",
"output_shape": "1,3,16,16",
"output_type": "float16"
},
{
"input_shape": "1,3,16,16",
"input_type": "uint8",
"output_shape": "1,3,16,16",
"output_type": "uint8"
}
]
},
{
"name": "AbsDiff",
"preload_list": [
{
"input_shape": "480, 640;480, 640",
"input_type": "float;float",
"output_shape": "480, 640",
"output_type": "float"
}
]
},
{
"name": "Log",
"preload_list": [
{
"input_shape": "480, 640",
"input_type": "float",
"output_shape": "480, 640",
"output_type": "float"
}
]
},
{
"name": "Pow",
"preload_list": [
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "float;float",
"output_shape": "1,3,16,16",
"output_type": "float"
}
]
},
{
"name": "Sqrt",
"preload_list": [
{
"input_shape": "1,3,16,16",
"input_type": "float",
"output_shape": "1,3,16,16",
"output_type": "float"
}
]
},
{
"name": "Sum",
"preload_list": [
{
"input_shape": "1,2,3,4",
"input_type": "uint8",
"output_shape": "1,4",
"output_type": "uint8"
},
{
"input_shape": "2,3,4",
"input_type": "float",
"output_shape": "4",
"output_type": "float"
}
]
},
{
"name": "Hstack",
"preload_list": [
{
"input_shape": "10, 10;10, 10",
"input_type": "uint8;uint8",
"output_shape": "10, 20",
"output_type": "uint8"
}
]
},
{
"name": "Vstack",
"preload_list": [
{
"input_shape": "10, 10;10, 10",
"input_type": "uint8;uint8",
"output_shape": "20, 10",
"output_type": "uint8"
}
]
},
{
"name": "ScaleAdd",
"preload_list": [
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "float;float",
"output_shape": "1,3,16,16",
"output_type": "float",
"attr_name": "scale",
"attr_type": "float",
"attr_val": "2.0"
}
]
},
{
"name": "Min",
"preload_list": [
{
"input_shape": "480, 640;480, 640",
"input_type": "float;float",
"output_shape": "480, 640",
"output_type": "float"
}
]
},
{
"name": "Max",
"preload_list": [
{
"input_shape": "480, 640;480, 640",
"input_type": "float;float",
"output_shape": "480, 640",
"output_type": "float"
}
]
},
{
"name": "Sort",
"preload_list": [
{
"input_shape": "2,3",
"input_type": "float",
"output_shape": "2,3",
"output_type": "float",
"attr_name": "axis;descending",
"attr_type": "int;bool",
"attr_val": "0;true"
},
{
"input_shape": "2,3",
"input_type": "float16",
"output_shape": "2,3",
"output_type": "float16",
"attr_name": "axis;descending",
"attr_type": "int;bool",
"attr_val": "0;true"
},
{
"input_shape": "2,3",
"input_type": "uint8",
"output_shape": "2,3",
"output_type": "uint8",
"attr_name": "axis;descending",
"attr_type": "int;bool",
"attr_val": "0;true"
}
]
},
{
"name": "SortIdx",
"preload_list": [
{
"input_shape": "2,3",
"input_type": "float",
"output_shape": "2,3",
"output_type": "int32",
"attr_name": "axis;descending",
"attr_type": "int;bool",
"attr_val": "0;true"
},
{
"input_shape": "2,3",
"input_type": "float16",
"output_shape": "2,3",
"output_type": "int32",
"attr_name": "axis;descending",
"attr_type": "int;bool",
"attr_val": "0;true"
},
{
"input_shape": "2,3",
"input_type": "uint8",
"output_shape": "2,3",
"output_type": "int32",
"attr_name": "axis;descending",
"attr_type": "int;bool",
"attr_val": "0;true"
}
]
},
{
"name": "Split",
"preload_list": [
{
"input_shape": "16,16,3",
"input_type": "float",
"output_shape": "16,16,1;16,16,1;16,16,1",
"output_type": "float;float;float"
}
]
},
{
"name": "Merge",
"preload_list": [
{
"input_shape": "16,16,1;16,16,2",
"input_type": "float;float",
"output_shape": "16,16,3",
"output_type": "float"
}
]
},
{
"name": "Transpose",
"preload_list": [
{
"input_shape": "2,3,2",
"input_type": "uint8",
"output_shape": "2,3,2",
"output_type": "uint8"
}
]
},
{
"name": "Add",
"preload_list": [
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "float;float",
"output_shape": "1,3,16,16",
"output_type": "float"
},
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "float16;float16",
"output_shape": "1,3,16,16",
"output_type": "float16"
},
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "uint8;uint8",
"output_shape": "1,3,16,16",
"output_type": "uint8"
}
]
},
{
"name": "BitwiseAnd",
"preload_list": [
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "uint8;uint8",
"output_shape": "1,3,16,16",
"output_type": "uint8"
}
]
},
{
"name": "Reduce",
"preload_list": [
{
"input_shape": "1, 640, 480, 1",
"input_type": "uint8",
"output_shape": "1, 480, 1",
"output_type": "uint8"
}
]
},
{
"name": "BitwiseXor",
"preload_list": [
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "uint8;uint8",
"output_shape": "1,3,16,16",
"output_type": "uint8"
}
]
},
{
"name": "BitwiseOr",
"preload_list": [
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "uint8;uint8",
"output_shape": "1,3,16,16",
"output_type": "uint8"
}
]
},
{
"name": "Clip",
"preload_list": [
{
"input_shape": "1,16,16,3",
"input_type": "float",
"output_shape": "1,16,16,3",
"output_type": "float"
},
{
"input_shape": "1,16,16,3",
"input_type": "float16",
"output_shape": "1,16,16,3",
"output_type": "float16"
},
{
"input_shape": "1,16,16,3",
"input_type": "uint8",
"output_shape": "1,16,16,3",
"output_type": "uint8"
}
]
},
{
"name": "ConvertTo",
"preload_list": [
{
"input_shape": "3, 3, 1",
"input_type": "uint8",
"output_shape": "3, 3, 1",
"output_type": "uint32"
}
]
},
{
"name": "Exp",
"preload_list": [
{
"input_shape": "2,3,2",
"input_type": "float",
"output_shape": "2,3,2",
"output_type": "float"
},
{
"input_shape": "2,3,2",
"input_type": "float16",
"output_shape": "2,3,2",
"output_type": "float16"
}
]
},
{
"name": "Subtract",
"preload_list": [
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "float;float",
"output_shape": "1,3,16,16",
"output_type": "float"
},
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "float16;float16",
"output_shape": "1,3,16,16",
"output_type": "float16"
},
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "uint8;uint8",
"output_shape": "1,3,16,16",
"output_type": "uint8"
}
]
},
{
"name": "Sqr",
"preload_list": [
{
"input_shape": "2,16,16,4",
"input_type": "uint8",
"output_shape": "2,16,16,4",
"output_type": "uint8"
}
]
},
{
"name": "Compare",
"preload_list": [
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "uint8;uint8",
"output_shape": "1,3,16,16",
"output_type": "uint8",
"attr_name": "operation",
"attr_type": "string",
"attr_val": "eq"
},
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "float16;float16",
"output_shape": "1,3,16,16",
"output_type": "float16",
"attr_name": "operation",
"attr_type": "string",
"attr_val": "eq"
},
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "float;float",
"output_shape": "1,3,16,16",
"output_type": "float",
"attr_name": "operation",
"attr_type": "string",
"attr_val": "eq"
}
]
},
{
"name": "AddWeighted",
"preload_list": [
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "uint8;uint8",
"output_shape": "1,3,16,16",
"output_type": "uint8",
"attr_name": "alpha;beta;gamma",
"attr_type": "float;float;float",
"attr_val": "1.2;1.0;1.1"
},
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "float16;float16",
"output_shape": "1,3,16,16",
"output_type": "float16",
"attr_name": "alpha;beta;gamma",
"attr_type": "float;float;float",
"attr_val": "1.2;1.0;1.1"
},
{
"input_shape": "1,3,16,16;1,3,16,16",
"input_type": "float;float",
"output_shape": "1,3,16,16",
"output_type": "float",
"attr_name": "alpha;beta;gamma",
"attr_type": "float;float;float",
"attr_val": "1.2;1.0;1.1"
}
]
},
{
"name": "ThresholdBinary",
"preload_list": [
{
"input_shape": "1,3,16,16",
"input_type": "uint8",
"output_shape": "1,3,16,16",
"output_type": "uint8",
"attr_name": "thresh;maxVal",
"attr_type": "float;float",
"attr_val": "20.0;30.0"
},
{
"input_shape": "1,3,16,16",
"input_type": "float16",
"output_shape": "1,3,16,16",
"output_type": "float16",
"attr_name": "thresh;maxVal",
"attr_type": "float;float",
"attr_val": "20.0;30.0"
},
{
"input_shape": "1,3,16,16",
"input_type": "float",
"output_shape": "1,3,16,16",
"output_type": "float",
"attr_name": "thresh;maxVal",
"attr_type": "float;float",
"attr_val": "20.0;30.0"
}
]
}
]}
父主题： 文件示例