golang tensorflow client类型问题。。。

legendlzy · 2018-02-09 12:15:01 · 998 次点击 · 大约8小时之前 开始浏览    置顶
这是一个创建于 2018-02-09 12:15:01 的主题,其中的信息可能已经有所发展或是发生改变。

    f, err := ioutil.ReadFile("2.npy")
    if err != nil {
        fmt.Println(err)
        return
    }

    tensor, err := tf.NewTensor(f)
    if err != nil {
        fmt.Println("Cannot read image file")
        return
    }

    tensorByte, ok := tensor.Value().([]byte)
    if !ok {
        fmt.Println("Cannot type assert tensor value to ")
        return
    }

    request := &pb.PredictRequest{
        ModelSpec: &pb.ModelSpec{
            Name:          "wav",
            SignatureName: "voice_check",
            Version: &google_protobuf.Int64Value{
                Value: int64(1),
            },
        },
        Inputs: map[string]*tf_core_framework.TensorProto{
            "voice": &tf_core_framework.TensorProto{
                Dtype: tf_core_framework.DataType_DT_FLOAT,
                TensorShape: &tf_core_framework.TensorShapeProto{
                    Dim: []*tf_core_framework.TensorShapeProto_Dim{
                        &tf_core_framework.TensorShapeProto_Dim{
                            Size: 1,
                        },
                        &tf_core_framework.TensorShapeProto_Dim{
                            Size: 1,
                        },
                        &tf_core_framework.TensorShapeProto_Dim{
                            Size: 96,
                        },
                        &tf_core_framework.TensorShapeProto_Dim{
                            Size: 89,
                        },
                    },
                },
                //TensorContent: f,
                StringVal: [][]byte{tensorByte},
            },
        },
    }

我使用python起了一个tensorflow的grpc服务端,想使用golang调用,但是需要传的类型是[][][][]float32,查了很多资料,没有找到相关,有大佬帮我解答一下吗。。。


有疑问加站长微信联系(非本文作者)

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

998 次点击  
加入收藏 微博
1 回复  |  直到 2018-02-10 10:05:31
legendlzy
legendlzy · #1 · 7年之前

自己帮顶。。。有大牛知道吗。。。

添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传