Hello everyone, I have a problem when I want to test my training file in real-time. When I grab the picture from the camera, there is error like this
Traceback (most recent call last):
File “/home/riza/Documents/P R O G R A M /NEW_GAN/Test.py”, line 145, in 
real_img = transform(real_img).cuda()
File “/usr/local/lib/python3.5/dist-packages/torchvision/transforms/transforms.py”, line 60, in call
img = t(img)
File “/usr/local/lib/python3.5/dist-packages/torchvision/transforms/transforms.py”, line 163, in call
return F.normalize(tensor, self.mean, self.std, self.inplace)
File “/usr/local/lib/python3.5/dist-packages/torchvision/transforms/functional.py”, line 208, in normalize
tensor.sub_(mean[:, None, None]).div_(std[:, None, None])
RuntimeError: output with shape [1, 512, 512] doesn’t match the broadcast shape [3, 512, 512]
Please your help. Thank you