使用ImageDataGenerator的flow_from_directory时RGB像素全部归零的问题

核心原因 #

你在flow_from_directory里设置的target_size=(20000, 1)参数顺序搞反了。Keras中target_size的格式是**(高度, 宽度)**，而你创建的图像尺寸是(宽度=20000, 高度=1)，原始图像shape为(1, 20000, 3)。当你把target_size设为(20000,1)时，相当于强制把图像拉伸成20000像素高、1像素宽，原本20000个水平像素被压缩到1个像素里，而你的图像大部分像素是0，最终输出就全变成0了。

修复步骤 #

1. 修正target_size参数顺序：
   将flow_from_directory中的target_size改为(1, 20000)，和原始图像的高宽匹配：

   train_generator = train_datagen.flow_from_directory(
       '/home/sam/sra_im/train1/',
       target_size=(1, 20000),  # 改为(高度, 宽度)
       batch_size=1,
       color_mode='rgb'
   )
   

2. 验证原始图像尺寸：
   你直接用PIL读取图像时输出的ima.shape应该是(1, 20000, 3)，对应高度1、宽度20000，target_size必须和这个一致，避免不必要的resize操作。

3. 可选：跳过resize（如果不需要）：
   如果不需要调整图像尺寸，可以去掉target_size参数，让生成器读取原始尺寸：

   train_generator = train_datagen.flow_from_directory(
       '/home/sam/sra_im/train1/',
       batch_size=1,
       color_mode='rgb'
   )
   

额外排查点 #

• 确认flow_from_directory读取的是你生成的目标图像，而非目录中其他全0的图像
• 检查图像保存格式：PNG格式支持无损保存你的低像素值图像，无需更换格式

内容的提问来源于stack exchange，提问作者Alexburn