Я пытался следовать этому уроку: Передача обучения для лодки или без лодки
Фрагмент интереса, который я не могу запустить:
# used a fixed dataset for evaluating the algorithm
valid_x, valid_y = next(flow_from_dataframe(valid_idg,
valid_df,
path_col = 'path',
y_col = 'has_ship_vec',
target_size = IMG_SIZE,
color_mode = 'rgb',
batch_size = VALID_IMG_COUNT)) # one big batch
print(valid_x.shape, valid_y.shape)
Выдает следующую ошибку:
IndexError Traceback (most recent call
last) <ipython-input-14-26b3a3a98978> in <module>
6 target_size = IMG_SIZE,
7 color_mode = 'rgb',
8 batch_size = VALID_IMG_COUNT)) # one big batch
9 print(valid_x.shape, valid_y.shape)
~/venv/lib/python3.7/site-packages/keras_preprocessing/image/iterator.py
in __next__(self, *args, **kwargs)
102
103 def __next__(self, *args, **kwargs):
104 return self.next(*args, **kwargs)
105
106 def next(self):
114 # The transformation of images is not under thread lock
115 # so it can be done in parallel
116 return self._get_batches_of_transformed_samples(index_array)
117
118 def _get_batches_of_transformed_samples(self, index_array):
~/venv/lib/python3.7/site-packages/keras_preprocessing/image/iterator.py
in _get_batches_of_transformed_samples(self, index_array)
225 filepaths = self.filepaths
226 for i, j in enumerate(index_array):
227 img = load_img(filepaths[j],
228 color_mode=self.color_mode,
229 target_size=self.target_size,
IndexError: list index out of range
Кто-нибудь может объяснить, что здесь происходит не так?