fix: 'numpy' has no attribute 'astype' (#13554)

pull/13559/head
laolitou 2024-07-31 12:36:31 +08:00 committed by GitHub
parent 358560f03d
commit cab3fcbcdf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -202,7 +202,7 @@ class TritonPythonModel:
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()