tensorflow - 将预训练检查点的输入层从 uint8 更改为 uint16

Question

我有一个来自 TensorFlow DeepLab Model Zoo 的预训练模型，我想将输入层数据类型从 uint8 更改为 uint16，然后通过我自己的 48 位（16 位/颜色通道）TIFF 数据集上的迁移学习对其进行训练。

但是，我想不出一种简单的方法来加载检查点并更改第一层的 dtype。预训练模型以及我的训练代码使用的是 TFv1.15。