从目录加载图像数据集代替tf.load中的MNIST数据集

Question

我想在我的数据集上实现堆叠式胶囊自动编码器（请参见here）。在实施过程中，我面临以下问题：

MNIST数据集加载如下：

tfds.load(name='mnist', split=subset, **kwargs).repeat().batch(batch_size)
<DatasetV1Adapter shapes: {label: (?,), image: (?, 28, 28, 1)}, types: {label: tf.int64, image: tf.uint8}>

我通过替换上面的代码来加载我的数据集，如下所示：

img_list = []
for img in glob.glob("update_207/*.png"):
  n= cv2.imread(img,0)
  img_1 = tf.convert_to_tensor(np.array(n.resize((2000, 1200, 1))))
  img_list.append(img_1) 
tf.data.Dataset.from_tensor_slices(img_list).repeat().batch(batch_size)
<DatasetV1Adapter shapes: (?,), types: tf.float32>

我是Tensorflow的新手，所以请提出对我的代码的建议，以便使我的数据集具有与MNIST相似的形状和类型。

Answer 1

很好地阐述“相似形状和类型”的意思。在您的情况下，看起来输入图像比mnist图像大得多。这意味着您必须将其尺寸调整为28x28像素。

如果您让我们知道当前缺少什么，我们可能可以为您提供更好的帮助！