Ошибка загрузки изображения классификации товаров в python

Question

Недавно я выполнял проект по классификации продуктов, у меня есть предварительно классифицированный набор данных «train» с 41 папкой, соответствующей каждой категории продуктов, и его файл csv, в котором перечислены имя изображения и его категория.

Затем у меня есть еще один «тестовый» набор данных с кучей неклассифицированных продуктов, проект хотел классифицировать эти изображения и вывести файл csv с «именем» и «категорией»

Я использовал Google Colab в этом проекте, после того, как я успешно загрузил и смонтировал все файлы, и я готов сканировать обученное изображение, у меня возникла ошибка, ниже мой код

train_image = []

for i in tqdm(range(train.shape[0])):

    img = image.load_img('content/train/train/'+train[i].astype('str')+'.jpg', target_size=(28,28,1))

    img = image.img_to_array(img)

    img = img/255

    train_image.append(img)

X = np.array(train_image)

, и это ошибка, которую я получаю:

/usr/local/lib/python3.6/dist-packages/pandas/core/indexes/base.py in get_loc(self, key, method, tolerance)
   2645             try:
-> 2646                 return self._engine.get_loc(key)
   2647             except KeyError:

pandas/_libs/index.pyx in pandas._libs.index.IndexEngine.get_loc()

pandas/_libs/index.pyx in pandas._libs.index.IndexEngine.get_loc()

pandas/_libs/hashtable_class_helper.pxi in pandas._libs.hashtable.PyObjectHashTable.get_item()

pandas/_libs/hashtable_class_helper.pxi in pandas._libs.hashtable.PyObjectHashTable.get_item()

KeyError: 0

During handling of the above exception, another exception occurred:

KeyError                                  Traceback (most recent call last)
2 frames
/usr/local/lib/python3.6/dist-packages/pandas/core/indexes/base.py in get_loc(self, key, method, tolerance)
   2646                 return self._engine.get_loc(key)
   2647             except KeyError:
-> 2648                 return self._engine.get_loc(self._maybe_cast_indexer(key))
   2649         indexer = self.get_indexer([key], method=method, tolerance=tolerance)
   2650         if indexer.ndim > 1 or indexer.size > 1:

pandas/_libs/index.pyx in pandas._libs.index.IndexEngine.get_loc()

pandas/_libs/index.pyx in pandas._libs.index.IndexEngine.get_loc()

pandas/_libs/hashtable_class_helper.pxi in pandas._libs.hashtable.PyObjectHashTable.get_item()

pandas/_libs/hashtable_class_helper.pxi in pandas._libs.hashtable.PyObjectHashTable.get_item()

KeyError: 0

Вот как выглядит моя обучающая директория в google colab

--content
  ↳train  
    ↳train  
      ↳0  
      ↳1  
      ....  
      ↳41  

Что делать, чтобы устранить эту ошибку?

Answer 1

Хорошо, я понял, это проблема формата load_image

img = image.load_img(r'train/train/' + str(train_df['category'][i]) + '/' + train_df['filename'][i], target_size=(28,28))