為什麼for images, labels in train_ds.take(1):會報錯呢(語言-python)

問題遇到的現象和發生背景

機器學習的貓狗識別代碼時，發生的錯誤

問題相關代碼，請勿粘貼截圖

plt.figure(figsize=(15, 10)) # 圖形的寬為15高為10

for images, labels in train_ds.take(1):
for i in range(8):

 ax = plt.subplot(5, 8, i + 1) plt.imshow(images[i]) plt.title(class_names[labels[i]]) plt.axis("off")plt.figure(figsize=(15, 10)) # 圖形的寬為15高為10

運行結果及報錯內容

InvalidArgumentError: Number of channels inherent in the image must be 1, 3 or 4, was 2
[[{ {node decode_image/DecodeImage}}]] [Op:IteratorGetNext]
<Figure size 1080x720 with 0 Axe

我的解答思路和嘗試過的方法

我想要達到的結果