Docker Tensorflow-Serving Predictions too large

    xiaoxiao2022-07-12  136

    最近在使用TensorFlow Serving的时候,如果图片较大,会出现如下的报错 “grpc_message”:“Received message larger than max (14830491 vs. 4194304)”

    经过查询,是因为grpc 默认只能传输4M的文件,而我们超过的这个阈值,所以要设置这个阈值

    MAX_MESSAGE_LENGTH = 429496729 #传输最大设置 class Grpc_Reco(): '通过GRPC的方式调用方向检测服务' def __init__(self): channel = grpc.insecure_channel("172.1.0.1:8000", options=[('grpc.max_send_message_length', MAX_MESSAGE_LENGTH), #加入options参数 ('grpc.max_receive_message_length', MAX_MESSAGE_LENGTH)]) # print(channel) self.stub = prediction_service_pb2_grpc.PredictionServiceStub(channel) self.request = predict_pb2.PredictRequest() self.request.model_spec.name = "mymodel" # 模型名称 self.request.model_spec.signature_name = "predict_images" # 签名名称 print ('grpc is inited')
    最新回复(0)