我正在按照本教程学习不同的数据集,以尝试不同的方法。但是,图像和文件夹格式与本教程相同。我使用的是.jpeg而不是.ppm。
我将代码更改为此:
def load_data(data_directory):
directories = [d for d in os.listdir(data_directory)
if os.path.isdir(os.path.join(data_directory, d))]
labels = []
images = []
for d in directories:
label_directory = os.path.join(data_directory,d)
file_names = [os.path.join(label_directory, f)
for f in os.listdir(label_directory)
if f.endswith(".jpeg")]
print(file_names)
for f in file_names:
images.append(skimage.io.imread(f))
labels.append(d)
return images, labels
我得到的错误是:
TypeError: Value passed to parameter 'labels' has DataType float32 not in list of allowed values: int32, int64
有人知道如何解决此问题吗?我试图整天用谷歌搜索修复它而没有成功... 谢谢