Здравствуйте! Если Вы используйте метод predict или просто обращаетесь к экземпляру model(), то Вы можете использовать атрибут classes. Он принимает список с id классов, которые будут распознаны. <br/> <br/> id классов можно найти в .yaml файле для обучения модели. в случае в датасетом COCO, на котором обучена модель YOLO, которая идет в этой библиотеке, этот файл можно найти по ссылке <a href="https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/coco.yaml" rel="nofollow">https://github.com/ultralytics/ultralytics/blob/ma...</a> <br/> <br/> <pre><code class="python">from ultralytics import YOLO
# Load a pretrained YOLOv8n model
model = YOLO('yolov8n.pt')
# Run inference on 'bus.jpg' with arguments
model.predict('bus.jpg', save=True, imgsz=320, conf=0.5, classes=[0,1])
#или так
#model('bus.jpg', save=True, imgsz=320, conf=0.5, classes=[0,1])</code></pre> <br/> <br/> вот ссылка на документацию <a href="https://docs.ultralytics.com/modes/predict/#inference-arguments" rel="nofollow">https://docs.ultralytics.com/modes/predict/#infere...</a> <br/> <br/> и если нужен будет какой то более сложный функционал, то можно наследовать свой класс от стандартной <a href="https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/model.py" rel="nofollow">модели</a> и делать в нем изменения. но в большинстве случаев это не пригодится