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