小编pro*_*eja的帖子

具有不同形状的Tensorflow数据集

我使用卷积网络对具有不同形状的图像进行分类。我找不到在Tensorflow中加载图像的方法。基于此问题,它应与tf.data.Dataset()一起使用。我以这种方式创建数据集:

    dataset = tf.data.Dataset.from_tensor_slices(filenames)
    dataset = dataset.map(read_file)
    dataset = dataset.shuffle(samples_in_buffer)
    dataset = dataset.batch(batch_size)
    dataset = dataset.repeat()
Run Code Online (Sandbox Code Playgroud)

执行时出现此错误

HandleElementToSlice Cannot copy slice: number of elements does not match.  Shapes are: [element]: [295,256,3], [parent slice]: [276,128,3]
Run Code Online (Sandbox Code Playgroud)

是否可以使用Tensorflow加载不同大小的图像,随机播放和批处理它们?

备注:我想使用空间金字塔池来处理不同的图像大小。

python tensorflow

1
推荐指数
1
解决办法
2683
查看次数

标签 统计

python ×1

tensorflow ×1