ошибка: обнаружена ошибка в рабочем процессе DataLoader 0 при работе с yolov8 в colab.

Question

У меня проблема с реализацией Yolov8 в колабе. Я скачал модель, используя,

      from ultralytics import YOLO
model = YOLO("yolov8n-cls.pt")

data directory,
data_dir='/content/gdrive/MyDrive/yolo'

У меня внутри есть две папки — train и val, которые, в свою очередь, имеют две папки, соответствующие двум классам.

Но когда я начинаю обучение,

      results = model.train(data=DATA_DIR, epochs=1, imgsz=64) 
I run into this issue
error: Caught error in DataLoader worker process 0.
Original Traceback (most recent call last):
  File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop
    data = fetcher.fetch(index)
  File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/_utils/fetch.py", line 51, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/usr/local/lib/python3.10/dist-packages/ultralytics/yolo/data/dataset.py", line 260, in __getitem__
    sample = self.album_transforms(image=cv2.cvtColor(im, cv2.COLOR_BGR2RGB))['image']
cv2.error: OpenCV(4.7.0) /io/opencv/modules/imgproc/src/color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function 'cvtColor'

Я ожидал, что обучение начнется, но оно не началось и выдает эту ошибку