package main

import (
    "fmt"
    "log"

    "<a style='color:#f60; text-decoration:underline;' href="https://www.php.cn/zt/15841.html" target="_blank">git</a>hub.com/tensorflow/tensorflow/tensorflow/go"
    "github.com/tensorflow/tensorflow/tensorflow/go/core/interpreter"
)

func main() {
    // 加载 MobileNet 模型
    model, err := interpreter.New(
        "path/to/mobilenet_v2_1.0_224quant.tflite",
        interpreter.Options{},
    )
    if err != nil {
        log.Fatal(err)
    }
    defer model.Close()

    // 设置模型输入
    inputs := [][]float32{}} // 数组表示输入图像
    model.SetTensor(0, inputs)

    // 运行模型
    if err = model.Invoke(); err != nil {
        log.Fatal(err)
    }

    // 获取模型输出
    outputs := model.Tensor(0)
    results, err := outputs.Float32s()
    if err != nil {
        log.Fatal(err)
    }

    // 打印结果
    for i, result := range results {
        fmt.Printf("预测结果 %d: %f\n", i, result)
    }
}