The size of tensor a must match the size of tensor b at non-singleton dimension 0

Posted 2022-03-15 AI浩

  File "D:\\cv\\captcha.Pytorch-master\\train.py", line 352, in <module>
    train_original(net)
  File "D:\\cv\\captcha.Pytorch-master\\train.py", line 234, in train_original
    for circle, input in enumerate(trainDataLoader, 0):
  File "D:\\ProgramData\\Anaconda3\\lib\\site-packages\\torch\\utils\\data\\dataloader.py", line 521, in __next__
    data = self._next_data()
  File "D:\\ProgramData\\Anaconda3\\lib\\site-packages\\torch\\utils\\data\\dataloader.py", line 561, in _next_data
    data = self._dataset_fetcher.fetch(index)  # may raise StopIteration
  File "D:\\ProgramData\\Anaconda3\\lib\\site-packages\\torch\\utils\\data\\_utils\\fetch.py", line 49, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "D:\\ProgramData\\Anaconda3\\lib\\site-packages\\torch\\utils\\data\\_utils\\fetch.py", line 49, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "D:\\cv\\captcha.Pytorch-master\\lib\\dataset.py", line 93, in __getitem__
    data = self.transform(data)
  File "D:\\ProgramData\\Anaconda3\\lib\\site-packages\\torchvision\\transforms\\transforms.py", line 61, in __call__
    img = t(img)
  File "D:\\ProgramData\\Anaconda3\\lib\\site-packages\\torch\\nn\\modules\\module.py", line 1102, in _call_impl
    return forward_call(*input, **kwargs)
  File "D:\\ProgramData\\Anaconda3\\lib\\site-packages\\torchvision\\transforms\\transforms.py", line 226, in forward
    return F.normalize(tensor, self.mean, self.std, self.inplace)
  File "D:\\ProgramData\\Anaconda3\\lib\\site-packages\\torchvision\\transforms\\functional.py", line 351, in normalize
    tensor.sub_(mean).div_(std)
RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimension 0

Process finished with exit code 1

解决方法：将图像转为RGB

 data = Image.open(imgPath).convert('RGB')