Обучение модели CV (компьютерное зрение). Модель YOLO.

Нужно обучить модель YOLO на детекцию 2 классов объектов. 

Датасет состоит из 100 инженерных схем. Нужно на них обнаруживать конкретные элементы 2 видов.