from ultralytics import YOLO

model = YOLO('yolov8m.pt')
model.save_dir = './predict/'

result = model.predict(
    './Assets/cat_dog.jpg',
    save=True,
    device='cpu',
    show=False,
    classes=[16, 15],
)[0]
result = model(
    './Assets/cat_dog.jpg',
    save=True,
    project='test',
    name='ololo',
    device='cpu',
    show=False,
)[0]

# print(len(result.boxes))

# for _ in result.boxes:
#     box = result.boxes[_]
#     print("Object type:", box.cls)
#     print("Coordinates:", box.xyxy)
# print("Probability:", box.conf)