预测图像还是黑白的

Question

支持了彩色模式，素材中也有彩色图片，训练后预测生成的图像还是黑白图像，这是什么原因呢？
老师视频中的黑白图像是预测生成的，高清图像应该是打开的原图图像吧？
这里是不是讲错了…

Answer 1

金色麦穗 2023-12-01 20:02:09

通常是指原图像是彩色的，而生成的图像是黑白（灰度）图像，然后通过支持GRB模式后，图像可以变成RGB三色的，主要是想表达这个意思。视频时间较久了，我明天看下课程中对应原来的代码，然后写个小demo说明一下估计更好一些，然后给同学留言。

• 提问者 慕后端8225145 #1

  好的，麻烦老师了

• 金色麦穗 回复 提问者 慕后端8225145 #2

  1.首先在建模最后一行增加一个输出层代码
  model.add(Conv2D(filters=3, kernel_size=(3, 3), padding='same', activation='linear'))
  激活函数尽量用linear，如果用softmax也行，但是误差函数就需要从长记忆了；
  2.然后，在最后预测图像代码最后追加一个 【*255】，如下：
  pred = model.predict(test_x)*255
  这样就不会因为归一化[0,1]之间，显得预测出来的图像是灰度图像了。

• 金色麦穗 #3

  补充，建模的时候
  卷积核调整的小一点， 把kernel_size=(9, 9),调整为 kernel_size=(3, 3),
  # 添加卷积层
  model.add(
      Conv2D(
          filters=64,
          kernel_size=(3, 3),
          padding='same',
          strides=(1, 1),
          activation='relu'
      )
  )