Neural

Используйте самую быструю нейросеть

detect

def detect(image: List, conf: float = 0.3, max: int = 30) -> Dict[str, List[Tuple[int, int, int, int, float]]]:
НазваниеТипОписание

img

np.array

Изображение на котором будет искать объекты

conf

float

Минимальная уверенность нейросети от 0 до 1.

img = screen.capture(0,0,FOV,FOV)
result = neural.detect(img)
heads = result['heads']
bodys = result['bodys']

for head in heads:
    print(f'Голова - координаты: {head[0]}, уверенность: {head[1]}')
for body in bodys:
    print(f'Тело - координаты: {head[0]}, уверенность: {head[1]}'

В нашем случае, head[0] = [x,y,w,h], где x,y - координаты левого верхнего угла, а w, h - ширина и высота головы.

Вы можете использовать только модель, которую пользователь использовал на момент запуска вашего скрипта .napi.

Last updated