diff --git a/pp.py b/pp.py index 58fe3e4..be78b65 100755 --- a/pp.py +++ b/pp.py @@ -18,7 +18,7 @@ import paddle from PIL import Image, ImageFile from paddleocr import PaddleOCR -paddle.set_flags({'FLAGS_fraction_of_gpu_memory_to_use': 0.6}) # 限制显存占用为GPU的80% +paddle.set_flags({'FLAGS_fraction_of_gpu_memory_to_use': 0.4}) # 限制显存占用为GPU的80% logging.disable(logging.DEBUG) # 关闭DEBUG日志的打印 logging.disable(logging.WARNING) # 关闭WARNING日志的打印 warnings.filterwarnings("ignore") @@ -94,6 +94,7 @@ def process_ocr(model, image): return result def process_images(conn, offset=0) -> int: + global EN, CH, JP, KR, RU with conn.cursor(pymysql.cursors.SSCursor) as cursor: cursor.execute("SELECT id, content FROM web_images WHERE text='' AND article_category_top_id=22 LIMIT 100 OFFSET %s", (offset,)) for id, content in cursor.fetchall():