проблема индексации при предварительной обработке изображений

Question

По моим данным, код уже разделял имя файла (строка и номер / кат.1500). Таким образом, код автоматически распознает собаку или кошку после присвоения номера рисунка.

У меня 1500 изображений кошек в случайном порядке от (кат.1 до кат.12294).

img_id = 3000
cat_generator = train_datagen.flow(train_imgs[img_id:img_id+1], train_labels[img_id:img_id+1],
                               batch_size=1)
cat = [next(cat_generator) for i in range(0,5)]
fig, ax = plt.subplots(1,5, figsize=(16, 6)
print('Labels:', [item[1][0] for item in cat])
l = [ax[i].imshow(cat[i][0][0]) for i in range(0,5)]

Приведенный выше код работает только для числа до 2999. Как я должен получить выше 3000, как я получаю ошибку ниже:

---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
<ipython-input-229-dc867a5e35bd> in <module>
      5 cat = [next(cat_generator) for i in range(0,6)]
      6 fig, ax = plt.subplots(1,5, figsize=(16, 6))
----> 7 print('Labels:', [item[1][0] for item in cat])
      8 l = [ax[i].imshow(cat[i][0][0]) for i in range(0,6)]
      9 

<ipython-input-229-dc867a5e35bd> in <listcomp>(.0)
      5 cat = [next(cat_generator) for i in range(0,6)]
      6 fig, ax = plt.subplots(1,5, figsize=(16, 6))
----> 7 print('Labels:', [item[1][0] for item in cat])
      8 l = [ax[i].imshow(cat[i][0][0]) for i in range(0,6)]
      9 

IndexError: index 0 is out of bounds for axis 0 with size 0