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):