The image is in RGB, because I take the picture directly from a camera, and I have set mean and std contain three values
This is my list code of transform image
transform = tv.transforms.Compose([   tv.transforms.Resize(512),
tv.transforms.ToTensor(),
tv.transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
])