Changing input layer of pretrained checkpoint from uint8 to uint16

131 Views Asked by At

I have a pretrained model from TensorFlow DeepLab Model Zoo and I would like to change the input layer datatype from uint8 to uint16 before training it through transfer learning on my own 48bit (16bit/color channel) TIFF dataset.

However, I cannot figure out a simple way to load the checkpoint and change the dtype of the first layer. The pretrained model as well as my train code is using TFv1.15.

enter image description here

0

There are 0 best solutions below