我使用卷积网络对具有不同形状的图像进行分类。我找不到在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加载不同大小的图像,随机播放和批处理它们?
备注:我想使用空间金字塔池来处理不同的图像大小。