fix: python API that caused Memory leak

AlexeyAB · Sep 15, 2020 · 5aa9fdd · 5aa9fdd · brandhsu · Nov 16, 2020
1 parent eb0272f
commit 5aa9fdd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/darknet_images.py b/darknet_images.py
@@ -111,6 +111,7 @@ def image_detection(image_path, network, class_names, class_colors, thresh):
 
     darknet.copy_image_from_bytes(darknet_image, image_resized.tobytes())
     detections = darknet.detect_image(network, class_names, darknet_image, thresh=thresh)
+    darknet.free_image(darknet_image)
     image = darknet.draw_boxes(detections, image_resized, class_colors)
     return cv2.cvtColor(image, cv2.COLOR_BGR2RGB), detections
 

diff --git a/darknet_video.py b/darknet_video.py
@@ -84,6 +84,7 @@ def inference(darknet_image_queue, detections_queue, fps_queue):
         fps_queue.put(fps)
         print("FPS: {}".format(fps))
         darknet.print_detections(detections, args.ext_output)
+        darknet.free_image(darknet_image)
     cap.release()