Hi,
As the error mentions, the input does not have the right shape.
GIven that it expects to have 3 channels and the size of your input, I guess you need to convert it from batch x height x width x channel to batch x channel x height x width. You can use the permute function for that.