tucan9389 / TFLiteSwift-Vision

Apache License 2.0

38 stars 1 forks source link

[PR] Simplify setup options #6

Closed tucan9389 closed 3 years ago

tucan9389 commented 3 years ago

Related Issue

5

PR Points

support uInt8 input type
remove input size argument and setup input size when loading tflite model and get the input shape
change normalization cases

AS-IS

let interpreterOptions = TFLiteVisionInterpreter.Options(
    modelName: "mobilenet_v2_1.0_224",
    inputWidth: 224, inputHeight: 224,
    normalization: .scaledNormalization
)

TO-BE

let interpreterOptions = TFLiteVisionInterpreter.Options(
    modelName: "mobilenet_v2_1.0_224",
    inputRankType: .bwhc,
    normalization: .scaled(from: 0.0, to: 1.0)
)