sensAI-Generic-Object-Detection / shared /eve_python /structs /EveObjectDetectionStructs.py
beaupreda's picture
Upload sensAI-Generic-Object-Detection with upload_repo.py
13170f7 verified
Raw
History Blame Contribute Delete
796 Bytes
import ctypes
from ctypes_enum import CtypesEnum
from .CDetectionStructs import *
from .EveControlOption import *
from .EveErrors import *
EVE_MAX_OBJECT_DETECTION_CLASSES = 128
class EveObjectDetectionOptions(ctypes.Structure):
_fields_ = [
("enabled", ctypes.c_int),
("modelPath", ctypes.c_byte * 512),
("nmsThreshold", ctypes.c_float),
("iouThreshold", ctypes.c_float),
("classNames", ctypes.c_byte * EVE_CLASS_ID_NAME_SIZE * EVE_MAX_OBJECT_DETECTION_CLASSES),
("classCount", ctypes.c_int),
("error", ctypes.c_int),
]
class EvePersonDetectionOptions(ctypes.Structure):
_fields_ = [
("enabled", ctypes.c_int),
("error", ctypes.c_int),
]
class EveDetectionData(ctypes.Structure):
_fields_ = [
("data", ctypes.POINTER(CDetectionData)),
("error", ctypes.c_int),
]