Why for images, labels in train_ ds. Take (1): an error will be reported (Language Python)

The phenomenon and background of the problem

Machine learning cat and dog identification code , What happened

Problem related code , Do not paste screenshots

plt.figure(figsize=(15, 10)) # The width of the figure is 15 High for 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)) # The width of the figure is 15 High for 10

Operation results and error reporting contents

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

My solution ideas and tried methods

What I want to achieve