[Serving]update np.object to np.object_ (#1021)

np.object to np.object_
This commit is contained in:
heliqi
2022-12-30 16:43:47 +08:00
committed by GitHub
parent 665f2e549a
commit 6310ddc8d6
5 changed files with 5 additions and 5 deletions

View File

@@ -141,7 +141,7 @@ class TritonPythonModel:
self.uie_model_.set_schema(schema) self.uie_model_.set_schema(schema)
results = self.uie_model_.predict(texts, return_dict=True) results = self.uie_model_.predict(texts, return_dict=True)
results = np.array(results, dtype=np.object) results = np.array(results, dtype=np.object_)
out_tensor = pb_utils.Tensor(self.output_names[0], results) out_tensor = pb_utils.Tensor(self.output_names[0], results)
inference_response = pb_utils.InferenceResponse( inference_response = pb_utils.InferenceResponse(
output_tensors=[out_tensor, ]) output_tensors=[out_tensor, ])

View File

@@ -92,7 +92,7 @@ class TritonPythonModel:
results = self.postprocess_.run([infer_outputs, ]) results = self.postprocess_.run([infer_outputs, ])
r_str = fd.vision.utils.fd_result_to_json(results) r_str = fd.vision.utils.fd_result_to_json(results)
r_np = np.array(r_str, dtype=np.object) r_np = np.array(r_str, dtype=np.object_)
out_tensor = pb_utils.Tensor(self.output_names[0], r_np) out_tensor = pb_utils.Tensor(self.output_names[0], r_np)
inference_response = pb_utils.InferenceResponse( inference_response = pb_utils.InferenceResponse(
output_tensors=[out_tensor, ]) output_tensors=[out_tensor, ])

View File

@@ -95,7 +95,7 @@ class TritonPythonModel:
results = self.postprocess_.run(infer_outputs) results = self.postprocess_.run(infer_outputs)
r_str = fd.vision.utils.fd_result_to_json(results) r_str = fd.vision.utils.fd_result_to_json(results)
r_np = np.array(r_str, dtype=np.object) r_np = np.array(r_str, dtype=np.object_)
out_tensor = pb_utils.Tensor(self.output_names[0], r_np) out_tensor = pb_utils.Tensor(self.output_names[0], r_np)
inference_response = pb_utils.InferenceResponse( inference_response = pb_utils.InferenceResponse(
output_tensors=[out_tensor, ]) output_tensors=[out_tensor, ])

View File

@@ -96,7 +96,7 @@ class TritonPythonModel:
results = self.postprocessor_.run([infer_outputs], im_infos) results = self.postprocessor_.run([infer_outputs], im_infos)
r_str = fd.vision.utils.fd_result_to_json(results) r_str = fd.vision.utils.fd_result_to_json(results)
r_np = np.array(r_str, dtype=np.object) r_np = np.array(r_str, dtype=np.object_)
out_tensor = pb_utils.Tensor(self.output_names[0], r_np) out_tensor = pb_utils.Tensor(self.output_names[0], r_np)
inference_response = pb_utils.InferenceResponse( inference_response = pb_utils.InferenceResponse(

View File

@@ -95,7 +95,7 @@ class TritonPythonModel:
dlpack_tensor) dlpack_tensor)
output_tensor_1 = pb_utils.Tensor( output_tensor_1 = pb_utils.Tensor(
self.output_names[1], np.array( self.output_names[1], np.array(
im_infos, dtype=np.object)) im_infos, dtype=np.object_))
inference_response = pb_utils.InferenceResponse( inference_response = pb_utils.InferenceResponse(
output_tensors=[output_tensor_0, output_tensor_1]) output_tensors=[output_tensor_0, output_tensor_1])
responses.append(inference_response) responses.append(inference_response)