Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
File size: 796 Bytes
13170f7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | 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),
]
|