mirror of
https://github.com/kerberos-io/openalpr-base.git
synced 2025-10-07 15:20:55 +08:00
Added OpenALPR Python bindings
This commit is contained in:
91
bindings/python/openalpr.py
Normal file
91
bindings/python/openalpr.py
Normal file
@@ -0,0 +1,91 @@
|
||||
import ctypes
|
||||
import json
|
||||
import platform
|
||||
|
||||
class Alpr():
|
||||
def __init__(self, country, config_file, runtime_dir):
|
||||
|
||||
# Load the .dll for Windows and the .so for Unix-based
|
||||
if platform.system().lower().find("windows") != -1:
|
||||
self._openalprpy_lib = ctypes.cdll.LoadLibrary("openalprpy.dll")
|
||||
else:
|
||||
self._openalprpy_lib = ctypes.cdll.LoadLibrary("libopenalprpy.so")
|
||||
|
||||
|
||||
self._initialize_func = self._openalprpy_lib.initialize
|
||||
self._initialize_func.argtypes = [ctypes.c_char_p, ctypes.c_char_p, ctypes.c_char_p]
|
||||
|
||||
self._dispose_func = self._openalprpy_lib.dispose
|
||||
|
||||
self._is_loaded_func = self._openalprpy_lib.isLoaded
|
||||
self._is_loaded_func.restype = ctypes.c_bool
|
||||
|
||||
self._recognize_file_func = self._openalprpy_lib.recognizeFile
|
||||
self._recognize_file_func.restype = ctypes.c_void_p
|
||||
self._recognize_file_func.argtypes = [ctypes.c_char_p]
|
||||
|
||||
self._recognize_array_func = self._openalprpy_lib.recognizeArray
|
||||
self._recognize_array_func.restype = ctypes.c_void_p
|
||||
self._recognize_array_func.argtypes = [ctypes.POINTER(ctypes.c_ubyte), ctypes.c_uint]
|
||||
|
||||
self._free_json_mem_func = self._openalprpy_lib.freeJsonMem
|
||||
|
||||
|
||||
self._set_default_region_func = self._openalprpy_lib.setDefaultRegion
|
||||
self._set_default_region_func.argtypes = [ctypes.c_char_p]
|
||||
|
||||
self._set_detect_region_func = self._openalprpy_lib.setDetectRegion
|
||||
self._set_detect_region_func.argtypes = [ctypes.c_bool]
|
||||
|
||||
|
||||
self._set_top_n_func = self._openalprpy_lib.setTopN
|
||||
self._set_top_n_func.argtypes = [ctypes.c_int]
|
||||
|
||||
self._get_version_func = self._openalprpy_lib.getVersion
|
||||
self._get_version_func.restype = ctypes.c_void_p
|
||||
|
||||
self._initialize_func(country, config_file, runtime_dir)
|
||||
|
||||
|
||||
def unload(self):
|
||||
self._openalprpy_lib.dispose()
|
||||
|
||||
def is_loaded(self):
|
||||
return self._is_loaded_func()
|
||||
|
||||
def recognize_file(self, file_path):
|
||||
ptr = self._recognize_file_func(file_path)
|
||||
json_data = ctypes.cast(ptr, ctypes.c_char_p).value
|
||||
response_obj = json.loads(json_data)
|
||||
self._free_json_mem_func(ptr)
|
||||
|
||||
return response_obj
|
||||
|
||||
def recognize_array(self, byte_array):
|
||||
|
||||
pb = ctypes.cast(byte_array, ctypes.POINTER(ctypes.c_ubyte))
|
||||
ptr = self._recognize_array_func(pb, len(byte_array))
|
||||
json_data = ctypes.cast(ptr, ctypes.c_char_p).value
|
||||
response_obj = json.loads(json_data)
|
||||
self._free_json_mem_func(ptr)
|
||||
|
||||
return response_obj
|
||||
|
||||
def get_version(self):
|
||||
|
||||
ptr = self._get_version_func()
|
||||
version_number = ctypes.cast(ptr, ctypes.c_char_p).value
|
||||
self._free_json_mem_func(ptr)
|
||||
|
||||
return version_number
|
||||
|
||||
def set_top_n(self, topn):
|
||||
self._set_top_n_func(topn)
|
||||
|
||||
def set_default_region(self, region):
|
||||
self._set_default_region_func(region)
|
||||
|
||||
def set_detect_region(self, enabled):
|
||||
self._set_detect_region_func(enabled)
|
||||
|
||||
|
Reference in New Issue
Block a user