无法在张量流上训练CNN

时间:2019-12-16 18:55:24

标签: tensorflow

我正在按照本教程学习不同的数据集,以尝试不同的方法。但是,图像和文件夹格式与本教程相同。我使用的是.jpeg而不是.ppm。

link

我将代码更改为此:

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

有人知道如何解决此问题吗?我试图整天用谷歌搜索修复它而没有成功... 谢谢

0 个答案:

没有答案