If you are loading a standard model lets say VGG, you can very easily find the shape of the input tensor from a google search.
Most vision models have an input shape of 3x224x224(AFAIK)
But do check once…