Are you passing the image_mean and image_std manually to the transformation?
If not, could you post a small code snippet showing, how you are calling the model including all input shapes, so that we can have a look?