Merge pull request #2 from persona-id/bjou/fix-memory-leak

fix: resolves memory leak caused by using CRAFT detector with detect() or readtext()

Author: bjou-persona

easyocr/detection.py

@@ -58,6 +58,9 @@ def test_net(
     with torch.no_grad():
         y, feature = net(x)
 
+    # remove x and feature from device, whether GPU or CPU
+    del x, feature
+
     boxes_list, polys_list = [], []
     for out in y:
         # make score and link map
@@ -82,6 +85,11 @@ def test_net(
         boxes_list.append(boxes)
         polys_list.append(polys)
 
+    # remove y from device, whether GPU or CPU, and check if cuda was used before calling empty_cache() to clean up
+    del y
+    if device == 'cuda':
+        torch.cuda.empty_cache()
+
     return boxes_list, polys_list
 
 def get_detector(trained_model, device='cpu', quantize=True, cudnn_benchmark=False):