YOLO выдает ошибку на черно-белые изображения

Пофиксить обёртку: выдавать ошибку в функции либо сделать возможность, чтобы YOLO обрабатывала одноканальные изображения, либо дублировать каналы.