fix ocr_serving bug

This commit is contained in:
yunyaoXYY
2023-03-06 02:17:27 +00:00
parent 81c2c75c5c
commit 6ff9e0fc14

View File

@@ -190,7 +190,7 @@ class TritonPythonModel:
index] > self.cls_threshold:
image_list[index] = cv2.rotate(
image_list[index].astype(np.float32), 1)
image_list[index] = np.astype(np.uint8)
image_list[index] = image_list[index].astype(np.uint8)
rec_pre_tensors = self.rec_preprocessor.run(image_list)
rec_dlpack_tensor = rec_pre_tensors[0].to_dlpack()