我有以下 python 代碼下載一組影像來訓練一個模型......
from matplotlib import pyplot as plt
import numpy as np
import tensorflow as tf
(train_images, train_labels), (_, _) = tf.keras.datasets.mnist.load_data()
train_images = train_images.reshape(train_images.shape[0], 28, 28, 1).astype('float32')
train_images = (train_images - 127.5) / 127.5
print(str(type(train_images[0,0])))
for i in range(len(train_images)):
plt.imsave('train_images\\' str(train_labels[i]) '.jpg', train_images[i, :])
我正在嘗試將這些影像保存到檔案中，在收到有關嘗試索引標量的錯誤訊息之前，我已根據我可以對其進行索引的次數確定這是一個 4d 陣列。
PS C:\Users\David\Desktop\ai> python .\images.py
<class 'numpy.ndarray'>
Traceback (most recent call last):
File "C:\Users\David\Desktop\ai\images.py", line 13, in <module>
plt.imsave('train_images\\00' str(train_labels[i]) '.jpg', train_images[i, :])
File "C:\python3.9.9\lib\site-packages\matplotlib\pyplot.py", line 2144, in imsave
return matplotlib.image.imsave(fname, arr, **kwargs)
File "C:\python3.9.9\lib\site-packages\matplotlib\image.py", line 1641, in imsave
rgba = sm.to_rgba(arr, bytes=True)
File "C:\python3.9.9\lib\site-packages\matplotlib\cm.py", line 434, in to_rgba
raise ValueError("Third dimension must be 3 or 4")
ValueError: Third dimension must be 3 or 4
PS C:\Users\David\Desktop\ai>
誰能告訴我如何做到這一點？
uj5u.com熱心網友回復：
我一問我就找到了答案……最后一行應該是這樣的：
plt.imsave('train_images\\' str(train_labels[i]) '.jpg', train_images[i, :, :, 0])
uj5u.com熱心網友回復：
問題是你[i,:])的 plt.imsave 在回圈內的 i in range (len (train_images))。你必須用 替換它[i,:,:, 0])。解決辦法是這樣
from matplotlib import pyplot as plt
import numpy as np
import tensorflow as tf
(train_images, train_labels), (_, _) = tf.keras.datasets.mnist.load_data()
train_images = train_images.reshape(train_images.shape[0], 28, 28, 1).astype('float32')
train_images = (train_images - 127.5) / 127.5
print(str(type(train_images[0,0])))
for i in range(len(train_images)):
plt.imsave('train_images\\' str(train_labels[i]) '.jpg', train_images[i, :, :, 0])
更新：抱歉，只是更新頁面，我在下面閱讀了您的回答。我沒看過，我正專心解決你的代碼
轉載請註明出處，本文鏈接：https://www.uj5u.com/yidong/391934.html