修复重复check问题

YaoFANGUK · Apr 12, 2022 · 6d34de5 · 6d34de5
1 parent adff19e
commit 6d34de5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/backend/config.py b/backend/config.py
@@ -16,7 +16,6 @@
 from paddle import fluid
 from tools.constant import *
 
-fluid.install_check.run_check()
 # 判断代码路径是否合法
 IS_LEGAL_PATH = True
 config = configparser.ConfigParser()
@@ -42,11 +41,9 @@
 interface_config.read(interface_file, encoding='utf-8')
 # 设置识别语言
 REC_CHAR_TYPE = config['DEFAULT']['Language']
-print(f"{interface_config['Main']['RecSubLang']}：{REC_CHAR_TYPE}")
 
 # 设置识别模式
 MODE_TYPE = config['DEFAULT']['Mode']
-print(f"{interface_config['Main']['RecMode']}：{MODE_TYPE}")
 ACCURATE_MODE_ON = False
 if MODE_TYPE == 'accurate':
     ACCURATE_MODE_ON = True
@@ -110,7 +107,6 @@
     if len(fluid.cuda_places()) > 0:
         # 如果有GPU则使用GPU
         USE_GPU = True
-        print(interface_config['Main']['GPUSpeedUp'])
 
 # 使用快速字幕检测算法时，背景颜色
 BG_MOD = BackgroundColor.DARK

diff --git a/backend/main.py b/backend/main.py
@@ -20,7 +20,8 @@
 import numpy as np
 from tqdm import tqdm
 import sys
-
+from paddle import fluid
+fluid.install_check.run_check()
 sys.path.insert(0, os.path.dirname(__file__))
 import config
 from config import interface_config
@@ -90,6 +91,10 @@ def __init__(self, vd_path, sub_area=None):
         self.raw_subtitle_path = os.path.join(self.subtitle_output_dir, 'raw.txt')
         # 自定义ocr对象
         self.ocr = OcrRecogniser()
+        print(f"{interface_config['Main']['RecSubLang']}：{config.REC_CHAR_TYPE}")
+        print(f"{interface_config['Main']['RecMode']}：{config.MODE_TYPE}")
+        if config.USE_GPU:
+            print(interface_config['Main']['GPUSpeedUp'])
         # 处理进度
         self.progress = 0
         # 是否完成