0

我将每个图像的名称存储在文件名数组中，并将与该图像关联的标签存储在标签数组中。所以我写了这段代码

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 "<ipython-input-4-4f63fd378da3>", line 3, in <module>
    subdirs, dirs, files = os.walk('/content/all_images').__next__()

StopIteration

如何解决这个错误？

4

0 回答 0