From 12b8f5d04b799146376380a4293b2241c7d40e6f Mon Sep 17 00:00:00 2001 From: "yuanzhi.zyz" Date: Tue, 20 Sep 2022 15:29:57 +0800 Subject: [PATCH] [to #42322933]fix cpu-used bug Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10184817 --- modelscope/pipelines/cv/ocr_recognition_pipeline.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/modelscope/pipelines/cv/ocr_recognition_pipeline.py b/modelscope/pipelines/cv/ocr_recognition_pipeline.py index 4b095042..c20d020c 100644 --- a/modelscope/pipelines/cv/ocr_recognition_pipeline.py +++ b/modelscope/pipelines/cv/ocr_recognition_pipeline.py @@ -91,7 +91,8 @@ class OCRRecognitionPipeline(Pipeline): data.append(mask) data = torch.FloatTensor(data).view( - len(data), 1, IMG_HEIGHT, IMG_WIDTH).cuda() / 255. + len(data), 1, IMG_HEIGHT, IMG_WIDTH) / 255. + data = data.to(self.device) result = {'img': data}