forked from mindspore-Ecosystem/mindspore
47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
|
# Copyright 2020 Huawei Technologies Co., Ltd
|
||
|
|
#
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
# you may not use this file except in compliance with the License.
|
||
|
|
# You may obtain a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
# See the License for the specific language governing permissions and
|
||
|
|
# limitations under the License.
|
||
|
|
# ============================================================================
|
||
|
|
import grpc
|
||
|
|
import numpy as np
|
||
|
|
import ms_service_pb2
|
||
|
|
import ms_service_pb2_grpc
|
||
|
|
|
||
|
|
|
||
|
|
def run():
|
||
|
|
channel = grpc.insecure_channel('localhost:50051')
|
||
|
|
stub = ms_service_pb2_grpc.MSServiceStub(channel)
|
||
|
|
# request = ms_service_pb2.EvalRequest()
|
||
|
|
# request.name = 'haha'
|
||
|
|
# response = stub.Eval(request)
|
||
|
|
# print("ms client received: " + response.message)
|
||
|
|
|
||
|
|
request = ms_service_pb2.PredictRequest()
|
||
|
|
request.data.tensor_shape.dims.extend([32, 1, 32, 32])
|
||
|
|
request.data.tensor_type = ms_service_pb2.MS_FLOAT32
|
||
|
|
request.data.data = (np.ones([32, 1, 32, 32]).astype(np.float32) * 0.01).tobytes()
|
||
|
|
|
||
|
|
request.label.tensor_shape.dims.extend([32])
|
||
|
|
request.label.tensor_type = ms_service_pb2.MS_INT32
|
||
|
|
request.label.data = np.ones([32]).astype(np.int32).tobytes()
|
||
|
|
|
||
|
|
result = stub.Test(request)
|
||
|
|
#result_np = np.frombuffer(result.result.data, dtype=np.float32).reshape(result.result.tensor_shape.dims)
|
||
|
|
print("ms client test call received: ")
|
||
|
|
#print(result_np)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
run()
|