在SRGAN当中,如果要对单通道的灰度图进行训练可以将输入层的尺寸改为(: ,: ,1),但是此时网络中使用的VGG19的 输入仍然为RGB三通道的图像,因此会报错。
VGG网络能否用于SRGAN对灰度图像的训练中?如果可以因当如何修改?
使用VGG19的代码如下:
def build_vgg(self):
vgg=VGG19(weights='imagenet')
vgg.outputs = [vgg.layers[9].output]
img = Input(shape=self.hr_shape)
img_features = vgg(img)
return Model(img, img_features)