experiencor / keras-yolo3

Training and Detecting Objects with YOLO3
MIT License
1.6k stars 861 forks source link

TypeError: 'Image' object is not subscriptable #317

Open Mai-Magdy opened 3 years ago

Mai-Magdy commented 3 years ago

for i, (input, target) in enumerate(val_loader): File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 521, in next data = self._next_data() File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 1203, in _next_data return self._process_data(data) File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 1229, in _process_data data.reraise() File "/usr/local/lib/python3.7/dist-packages/torch/_utils.py", line 425, in reraise raise self.exc_type(msg) TypeError: Caught TypeError in DataLoader worker process 0. Original Traceback (most recent call last): File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop data = fetcher.fetch(index) File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/fetch.py", line 44, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/content/gdrive/My Drive/Colab Notebooks/dataset.py", line 219, in getitem img_group = self._get_frames(record, video_reader,segment_indices) File "/content/gdrive/My Drive/Colab Notebooks/dataset.py", line 173, in _get_frames process_data = self.transform(images) File "/usr/local/lib/python3.7/dist-packages/torchvision/transforms/transforms.py", line 60, in call img = t(img) File "/content/gdrive/My Drive/Colab Notebooks/transforms.py", line 442, in call if img_group[0].mode == 'L': TypeError: 'Image' object is not subscriptable