Используйте самую быструю нейросеть
Last updated 8 months ago
def detect(image: List, conf: float = 0.3, max: int = 30) -> Dict[str, List[Tuple[int, int, int, int, float]]]:
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.
img
np.array
Изображение на котором будет искать объекты
conf
float
Минимальная уверенность нейросети от 0 до 1.