diff --git a/main.py b/main.py index d3d490a..4fc3a71 100644 --- a/main.py +++ b/main.py @@ -4,7 +4,6 @@ import traceback os.environ["CUDA_VISIBLE_DEVICES"] = "1" -# import paddle from auto_email.error_email import send_an_error_email from config.log import LOGGING_CONFIG from photo_review.photo_review import main @@ -22,5 +21,3 @@ if __name__ == '__main__': except Exception as e: log.error(traceback.format_exc()) send_an_error_email(program_name='照片审核关键信息抽取脚本', error_name=repr(e), error_detail=traceback.format_exc()) - # 释放显存 - # paddle.device.cuda.empty_cache() diff --git a/photo_review/photo_review.py b/photo_review/photo_review.py index b709e79..caf1a3d 100644 --- a/photo_review/photo_review.py +++ b/photo_review/photo_review.py @@ -2,7 +2,6 @@ import json import logging import math import os -import pdb import sys import tempfile import time @@ -55,7 +54,6 @@ def split_image(img_path, max_ratio=2.82, best_ration=1.41, overlap=0.05): split_result = [] # 打开图片 img = open_image(img_path) - pdb.set_trace() # 获取图片的宽度和高度 height, width = img.shape[:2] # 计算宽高比