回答此问题可获得 20 贡献值,回答如果被采纳可获得 50 分。
<p>我从tensorflow基本图像分类指南中得到了一组名为train\u images和train\u labels的图像:</p>
<p><a href="https://www.tensorflow.org/tutorials/keras/classification" rel="nofollow noreferrer">https://www.tensorflow.org/tutorials/keras/classification</a></p>
<p>我加载数据集时使用:</p>
<pre><code>fashion_mnist = keras.datasets.fashion_mnist
(train_images, train_labels), (test_images, test_labels) = fashion_mnist.load_data()
</code></pre>
<p>这两个列表的形状分别是:
(60000,28,28)(60000,)</p>
<p>之后,我想使用ImageDataGenerator水平翻转一些图像,但是当我将模型与我的火车列表相匹配时,它会返回一个错误,说x应该是一个秩4的数组</p>
<p>我已经试过了</p>
<pre><code>train_images = (np.expand_dims(train_images,0))
</code></pre>
<p>所以形状变成(160000,28,28)
(我必须这样做,才能让模型检查单个图像)
但它不适合模型</p>
<p>以下是代码的其余部分:</p>
<pre><code>aug = ImageDataGenerator(rotation_range=20, horizontal_flip=True)
model = keras.Sequential([
keras.layers.Flatten(input_shape=(28,28)),
keras.layers.Dense(128, activation='relu'),
keras.layers.Dense(10, activation='softmax')
])
model.compile(
optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy']
)
BS=32
EPOCHS=10
H = model.fit_generator(
aug.flow(train_images, train_labels, batch_size=BS),
validation_data=(test_images, test_labels),
steps_per_epoch=len(train_images) // BS,
epochs=EPOCHS)
</code></pre>
<p>这就是产生的错误:</p>
<pre><code>---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
<ipython-input-65-e49da92bcb89> in <module>()
5 #train_images.shape
6 H = model.fit_generator(
----> 7 aug.flow(train_images, train_labels, batch_size=BS),
8 validation_data=(test_images, test_labels),
9 steps_per_epoch=len(train_images) // BS,
1 frames
/usr/local/lib/python3.6/dist-packages/keras_preprocessing/image/numpy_array_iterator.py in __init__(self, x, y, image_data_generator, batch_size, shuffle, sample_weight, seed, data_format, save_to_dir, save_prefix, save_format, subset, dtype)
115 raise ValueError('Input data in `NumpyArrayIterator` '
116 'should have rank 4. You passed an array '
--> 117 'with shape', self.x.shape)
118 channels_axis = 3 if data_format == 'channels_last' else 1
119 if self.x.shape[channels_axis] not in {1, 3, 4}:
ValueError: ('Input data in `NumpyArrayIterator` should have rank 4. You passed an array with shape', (60000, 28, 28))
</code></pre>
<p>实际上,列车图像是(N°的图像,宽度,高度)什么是第四轴它正在等待?
如何执行此操作?你知道吗</p>