индекс списка ошибок вне диапазона, связанного с файлом метки - PullRequest
0 голосов
/ 05 ноября 2019

Я пытаюсь научиться переводному обучению (различая кошку и собаку) самостоятельно. И я не понимаю, что вызывает проблему индексации, связанную с маркировкой файлов. Ниже приведен мой код.

IMG_DIM = (150, 150)
test_files = glob.glob('test_data/*')
test_imgs = [img_to_array(load_img(img, target_size=IMG_DIM)) for img in test_files]
test_imgs = np.array(test_imgs)
test_labels = [fn.split('/')[1].split('.')[0].strip() for fn in test_files]
test_imgs_scaled = test_imgs.astype('float32')
test_imgs_scaled /= 255
test_labels_enc = class2num_label_transformer(test_labels)
print('Test dataset shape:', test_imgs.shape)
print(test_labels[0:4], test_labels_enc[0:4])

И это сообщение об ошибке.

    ---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
<ipython-input-166-d351a31bcadc> in <module>
      4 test_imgs = [img_to_array(load_img(img, target_size=IMG_DIM)) for img in test_files]
      5 test_imgs = np.array(test_imgs)
----> 6 test_labels = [fn.split('/')[1].split('.')[0].strip() for fn in test_files]
      7 
      8 test_imgs_scaled = test_imgs.astype('float32')

<ipython-input-166-d351a31bcadc> in <listcomp>(.0)
      4 test_imgs = [img_to_array(load_img(img, target_size=IMG_DIM)) for img in test_files]
      5 test_imgs = np.array(test_imgs)
----> 6 test_labels = [fn.split('/')[1].split('.')[0].strip() for fn in test_files]
      7 
      8 test_imgs_scaled = test_imgs.astype('float32')

IndexError: индекс списка вне диапазона

Я также прикрепляю папку, которая содержитфайлы, используемые в этом коде. file used in this code

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...