c# - 如何在 C# 中创建 TensorProto？

Question

这是我创建的 C# 客户端的一个片段，用于查询我使用本教程设置的 tensorflow 服务器:https://tensorflow.github.io/serving/serving_inception.html

        var channel = new Channel("TFServer:9000", ChannelCredentials.Insecure);

        var request = new PredictRequest();

        request.ModelSpec = new ModelSpec();
        request.ModelSpec.Name = "inception";

        var imgBuffer = File.ReadAllBytes(@"sample.jpg");

        ByteString jpeg = ByteString.CopyFrom(imgBuffer, 0, imgBuffer.Length);

        var jpgeproto = new TensorProto();
        jpgeproto.StringVal.Add(jpeg);
        jpgeproto.Dtype = DataType.DtStringRef;


        request.Inputs.Add("images", jpgeproto); // new TensorProto{TensorContent = jpeg});

        PredictionClient client = new PredictionClient(channel);

我发现大多数类需要使用protoc从proto文件生成

我唯一找不到的是如何构建 TensorProto。我不断收到的错误是:附加信息:Status(StatusCode=InvalidArgument, Detail="张量解析错误:图像")

有一个示例客户端 ( https://github.com/tensorflow/serving/blob/master/tensorflow_serving/example/inception_client.py )，但我的 Python 技能不足以理解最后一点。

Answer 1

我还用另一种语言(Java)实现了该客户端。

尝试改变

jpgeproto.Dtype = DataType.DtStringRef; 

到

jpgeproto.Dtype = DataType.DtString;

您可能还需要向张量原型(prototype)添加具有尺寸的张量形状。这是我在 Java 中的工作解决方案，在 C# 中应该类似:

TensorShapeProto.Dim dim = TensorShapeProto.Dim.newBuilder().setSize(1).build();
TensorShapeProto shape = TensorShapeProto.newBuilder().addDim(dim).build();
TensorProto proto = TensorProto.newBuilder()
                .addStringVal(ByteString.copyFrom(imageBytes))
                .setTensorShape(shape)
                .setDtype(DataType.DT_STRING)
                .build();
ModelSpec spec = ModelSpec.newBuilder().setName("inception").build();
PredictRequest r = PredictRequest.newBuilder()
                .setModelSpec(spec)
                .putInputs("images", proto).build();
PredictResponse response = blockingStub.predict(r);