🧹 sbb_textline_detector: Delete textline session/model after using it

5 years ago · 5d440857e7
parent 9b2c415125
commit 5d440857e7
1 changed files with 6 additions and 0 deletions
--- a/qurator/sbb_textline_detector/main.py
+++ b/qurator/sbb_textline_detector/main.py
@ -683,6 +683,12 @@ class textlineerkenner:

            y_predi = mask_true
            y_predi = cv2.resize(y_predi, (img_org.shape[1], img_org.shape[0]), interpolation=cv2.INTER_NEAREST)
+
+        session_textline.close()
+
+        del model_textline
+        del session_textline
+        gc.collect()
        return y_predi

    def get_textlines_for_each_textregions(self, textline_mask_tot, boxes):