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

问题描述

我有一个来自TensorFlow DeepLab Model Zoo的经过预训练的模型,在通过在我自己的48bit(16bit /颜色通道)TIFF数据集上进行转移学习对其进行训练之前,我想将输入层数据类型从uint8更改为uint16。

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

enter image description here

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)