Usually you would normalize your input images before passing it to the model, since this might speed up the training and also yield better results.
I think it’s not a good idea to somehow force the model to output activations in the uint8 range.
If you want to visualize the activations, matplotlib should automatically scale them if you are passing float values as far as I know.