修改TF2功能模型的输入层

时间:2020-06-20 03:38:11

标签: tensorflow keras tensorflow2.0

我有一个从预先训练的砝码加载的TF2模型。该模型的前三层是:

  1. Input,带有3个输入通道
  2. ZeroPadding
  3. Conv2D,带有3个输入通道

现在,我只想替换这三层,因为我的输入是4通道图像。我的目标是为输入图像的前3个通道重用预先训练的3个通道权重,并为第4个输入通道使用随机初始化的权重。

什么是这样做的好方法?

0 个答案:

没有答案