torch.Tensor) —
The image input, can be a PIL image, numpy array or pytorch tensor. if it is a numpy array, should have
shape [batch, height, width] or [batch, height, width, channel] if it is a pytorch tensor, should
have shape [batch, channel, height, width].   height (int, optional, defaults to None) —
The height in preprocessed image. If None, will use the height of image input.   width (int, optional, defaults to None) -- The width in prep