Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- do{
- var options = InterpreterOptions()
- options.isErrorLoggingEnabled = true
- try self.interpreter = Interpreter(modelPath: Bundle.main.path(forResource: modelFileName, ofType: "tflite")!, options: options)
- // allocate input tensors at the given shapes
- try self.interpreter.allocateTensors()
- }catch {
- print("initialize TF interpreter failed: (error)")
- return nil
- }
- let data = Data(copyingBufferOf: Array<Float>(repeating: 1, count: 128*10))
- try interpreter.copy(data, toInputAt: 0)
- try interpreter.invoke()
- let outputTensor = try interpreter.output(at: 0)
- let outputResult = [Float](unsafeData: outputTensor.data)!
- print("interpreter result on test data", outputResult)
- precondition(outputResult == [0.2032134])
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement