From 341a7b14732ece85502a5f4378113365f0b17a43 Mon Sep 17 00:00:00 2001 From: Lynn Date: Mon, 15 Sep 2025 17:49:52 +0800 Subject: [PATCH] Fix: judge not empty before delete (#10099) ### What problem does this PR solve? judge not empty before delete session. ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- deepdoc/vision/ocr.py | 6 ++++-- deepdoc/vision/recognizer.py | 3 ++- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/deepdoc/vision/ocr.py b/deepdoc/vision/ocr.py index fb42fba27..d9f472aa1 100644 --- a/deepdoc/vision/ocr.py +++ b/deepdoc/vision/ocr.py @@ -351,7 +351,8 @@ class TextRecognizer: def close(self): # close session and release manually logging.info('Close TextRecognizer.') - del self.predictor + if hasattr(self, "predictor"): + del self.predictor gc.collect() def __call__(self, img_list): @@ -490,7 +491,8 @@ class TextDetector: def close(self): logging.info("Close TextDetector.") - del self.predictor + if hasattr(self, "predictor"): + del self.predictor gc.collect() def __call__(self, img): diff --git a/deepdoc/vision/recognizer.py b/deepdoc/vision/recognizer.py index a829c1cd3..65995a579 100644 --- a/deepdoc/vision/recognizer.py +++ b/deepdoc/vision/recognizer.py @@ -408,7 +408,8 @@ class Recognizer: def close(self): logging.info("Close recognizer.") - del self.ort_sess + if hasattr(self, "ort_sess"): + del self.ort_sess gc.collect() def __call__(self, image_list, thr=0.7, batch_size=16):