#python #image #classification #image-preprocessing #stopiteration
#питон #изображение #классификация #предварительная обработка изображений #стопитерация
Вопрос:
я сохраняю имена каждого изображения в массиве имен файлов и метки, связанные с этим изображением, в массиве меток. поэтому я написал этот код
import numpy as np subdirs, dirs, files = os.walk('/content/all_images').__next__() m = len(files) print(m) filenames = [] labels = np.zeros((m, 1)) import os import shutil images_dir = '/content/all_images' filenames_counter = 0 labels_counter = -1 for subdir, dirs, files in os.walk(train_dir): #print(files) for file in files: filenames.append(file) labels[filenames_counter, 0] = labels_counter filenames_counter = filenames_counter 1 labels_counter = labels_counter 1 print(len(filenames)) print(labels.shape) но я получил ошибку
Traceback (most recent call last): File "lt;ipython-input-4-4f63fd378da3gt;", line 3, in lt;modulegt; subdirs, dirs, files = os.walk('/content/all_images').__next__() StopIteration как устранить эту ошибку ?
Комментарии:
1. Пожалуйста, добавьте полный отчет об ошибке в свой вопрос!
2. Похоже
os.walk('/content/all_images'), что возвращен пустой итератор3. Обратная связь (последний последний вызов): Файл «lt;ipython-input-4-4f63fd378da3gt;», строка 3, в lt;ipython-input-4-4f63fd378da3gt;lt;модульgt; поддиры, dir, файлы = os.walk(‘/content/all_images’).__следующий__() Остановка