fix ocr

pur1fying · Feb 18, 2024 · 6e5de60 · 6e5de60
1 parent 9de58c0
commit 6e5de60
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/core/ocr/ocr.py b/core/ocr/ocr.py
@@ -78,8 +78,8 @@ def get_region_num(self, img, region, category=int, ratio=1.0):
         t1 = time.time()
         res = self.ocrNUM.ocr_for_single_line(img)['text']
         ocr_time = round(time.time() - t1, 3)
-        res.replace('<unused3>', '')
-        res.replace('<unused2>', '')
+        res = res.replace('<unused3>', '')
+        res = res.replace('<unused2>', '')
         self.logger.info("ocr res : " + res + " time: " + str(ocr_time))
         temp = ''
         for i in range(0, len(res)):
@@ -97,8 +97,8 @@ def get_region_pure_english(self, img, region, ratio=1.0):
         t1 = time.time()
         res = self.ocrEN.ocr_for_single_line(img)['text']
         ocr_time = round(time.time() - t1, 3)
-        res.replace('<unused3>', '')
-        res.replace('<unused2>', '')
+        res = res.replace('<unused3>', '')
+        res = res.replace('<unused2>', '')
         self.logger.info("ocr res : " + res + " time: " + str(ocr_time))
         temp = ''
         for i in range(0, len(res)):
@@ -111,8 +111,8 @@ def get_region_pure_chinese(self, img, region, ratio=1.0):
         t1 = time.time()
         res = self.ocrCN.ocr_for_single_line(img)['text']
         ocr_time = round(time.time() - t1, 3)
-        res.replace('<unused3>', '')
-        res.replace('<unused2>', '')
+        res = res.replace('<unused3>', '')
+        res = res.replace('<unused2>', '')
         self.logger.info("ocr res : " + res + " time: " + str(ocr_time))
         temp = ''
         for i in range(0, len(res)):
@@ -149,8 +149,8 @@ def get_region_res(self, img, region, model='CN', ratio=1.0):
         elif model == 'JP':
             res = self.ocrJP.ocr_for_single_line(img)['text']
         ocr_time = round(time.time() - t1, 3)
-        res.replace('<unused3>', '')
-        res.replace('<unused2>', '')
+        res = res.replace('<unused3>', '')
+        res = res.replace('<unused2>', '')
         self.logger.info("ocr res : " + res + " time: " + str(ocr_time))
         return res