Ошибка про обучении модели

У меня есть скаченные изображения для обучения мой модели. Но когда идёт сам процесс обучения постоянно возникает ошибка:

image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)  # RGB image
cv2.error: OpenCV(4.5.1) C:\Users\appveyor\AppData\Local\Temp\1\pip-req-build-oduouqig\opencv\modules\imgproc\src\color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function 'cv::cvtColor'

Как я понял изображение загружается некорректно. Что можно сделать в этом случае?

Вот код:

from imageai.Detection.Custom import DetectionModelTrainer

trainer = DetectionModelTrainer()
trainer.setModelTypeAsYOLOv3()
trainer.setDataDirectory(data_directory=r"C:\rec\data")
trainer.setTrainConfig(object_names_array=["hat","person"], batch_size=10, num_experiments=15, train_from_pretrained_model="pretrained-yolov3.h5")
trainer.trainModel()