shapes / shapes.mlmodelc /metadata.json
finnvoorhees's picture
Shapes: on-device shape recognition from a single stroke
59d10f0
Raw
History Blame Contribute Delete
2.94 kB
[
{
"metadataOutputVersion" : "3.0",
"storagePrecision" : "Mixed (Float16, Palettized (4 bits))",
"outputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 1 × 6)",
"shortDescription" : "",
"shape" : "[1, 6]",
"name" : "probs",
"type" : "MultiArray"
}
],
"modelParameters" : [
],
"specificationVersion" : 7,
"mlProgramOperationTypeHistogram" : {
"Transpose" : 11,
"Ios16.maxPool" : 4,
"Squeeze" : 1,
"Ios16.softmax" : 3,
"Ios16.linear" : 10,
"Ios16.add" : 7,
"Ios16.realDiv" : 1,
"Ios16.reduceSum" : 2,
"ExpandDims" : 4,
"Ios16.sub" : 1,
"Ios16.clip" : 1,
"Ios16.cast" : 3,
"Ios16.constexprLutToDense" : 12,
"Ios16.relu" : 4,
"Ios16.layerNorm" : 4,
"SliceByIndex" : 6,
"Ios16.conv" : 3,
"Ios16.matmul" : 4,
"Ios16.gelu" : 2,
"Ios16.reshape" : 8,
"Ios16.mul" : 5
},
"computePrecision" : "Mixed (Float16, Float32, Int32)",
"isUpdatable" : "0",
"stateSchema" : [
],
"availability" : {
"macOS" : "13.0",
"tvOS" : "16.0",
"visionOS" : "1.0",
"watchOS" : "9.0",
"iOS" : "16.0",
"macCatalyst" : "16.0"
},
"modelType" : {
"name" : "MLModelType_mlProgram"
},
"userDefinedMetadata" : {
"snap_gates" : "{\"line\": {\"conf\": 0.3, \"resid\": 0.02}, \"rectangle\": {\"conf\": 0.3, \"resid\": 0.1}, \"triangle\": {\"conf\": 0.3, \"resid\": 1.0}, \"ellipse\": {\"conf\": 0.7, \"resid\": 0.05}, \"star\": {\"conf\": 0.75, \"resid\": 0.08}}",
"com.github.apple.coremltools.source" : "torch==2.12.1",
"seq_len" : "256",
"classes" : "[\"line\", \"rectangle\", \"triangle\", \"ellipse\", \"star\", \"none\"]",
"com.github.apple.coremltools.conversion_date" : "2026-06-26",
"com.github.apple.coremltools.version" : "9.0",
"com.github.apple.coremltools.source_dialect" : "TorchScript",
"preprocess" : "{\"spacing\": 0.02, \"dist_mean\": 0.019319261383684345, \"dist_std\": 0.0019667792003264808, \"add_curvature\": false}"
},
"inputSchema" : [
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 1 × 256 × 3)",
"shortDescription" : "",
"shape" : "[1, 256, 3]",
"name" : "features",
"type" : "MultiArray"
},
{
"hasShapeFlexibility" : "0",
"isOptional" : "0",
"dataType" : "Float32",
"formattedType" : "MultiArray (Float32 1 × 256)",
"shortDescription" : "",
"shape" : "[1, 256]",
"name" : "mask",
"type" : "MultiArray"
}
],
"generatedClassName" : "shapes",
"method" : "predict"
}
]