Merge pull request AlexeyAB#6440 from guitarmind/master

Add free_network_ptr Support for Darknet in Python
zh2angijun · Aug 9, 2020 · 9dc897d · 9dc897d
2 parents af08480 + 59bb012
commit 9dc897d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/darknet.py b/darknet.py
@@ -276,6 +276,10 @@ def detect_image(network, class_names, image, thresh=.5, hier_thresh=.5, nms=.45
 load_net_custom.argtypes = [c_char_p, c_char_p, c_int, c_int]
 load_net_custom.restype = c_void_p
 
+free_network_ptr = lib.free_network_ptr
+free_network_ptr.argtypes = [c_void_p]
+free_network_ptr.restype = c_void_p
+
 do_nms_obj = lib.do_nms_obj
 do_nms_obj.argtypes = [POINTER(DETECTION), c_int, c_int, c_float]