fix OCR color extraction

zyddnys · Nov 12, 2023 · 88349e7 · 88349e7
1 parent 20d6a87
commit 88349e7
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 12 deletions.
diff --git a/manga_translator/ocr/model_32px.py b/manga_translator/ocr/model_32px.py
@@ -130,10 +130,6 @@ async def _infer(self, image: np.ndarray, textlines: List[Quadrilateral], args:
                 out_regions.append(cur_region)
 
         if is_quadrilaterals:
-            for region in out_regions :
-                if isinstance(region, TextBlock):
-                    region.fg_colors /= float(len(region.lines))
-                    region.bg_colors /= float(len(region.lines))
             return out_regions
         return textlines
 

diff --git a/manga_translator/ocr/model_48px.py b/manga_translator/ocr/model_48px.py
@@ -153,10 +153,6 @@ async def _infer(self, image: np.ndarray, textlines: List[Quadrilateral], verbos
                 out_regions.append(cur_region)
 
         if is_quadrilaterals:
-            for region in out_regions :
-                if isinstance(region, TextBlock):
-                    region.fg_colors /= float(len(region.lines))
-                    region.bg_colors /= float(len(region.lines))
             return out_regions
         return textlines
 

diff --git a/manga_translator/ocr/model_48px_ctc.py b/manga_translator/ocr/model_48px_ctc.py
@@ -149,10 +149,6 @@ async def _infer(self, image: np.ndarray, textlines: List[Quadrilateral], args:
                 out_regions.append(cur_region)
 
         if is_quadrilaterals:
-            for region in out_regions :
-                if isinstance(region, TextBlock):
-                    region.fg_colors /= float(len(region.lines))
-                    region.bg_colors /= float(len(region.lines))
             return out_regions
         return textlines