Pytorch: ValueError: 太多維度: 3 > 2. 9/opt/anaconda3/envs/pytorch/lib/python3.7/site-packages/PIL/Image.py in fromarray(obj, mode)

Question

我正在使用 MNIST 數據通過 pytorch 運行我的 python。 我喜歡只訓練數字 0 和 1 的部分數據。當我嘗試打印第一張圖像的大小時，它遇到了這個錯誤：

值錯誤：維度太多：3 > 2

我對 Python 很陌生。 如果我不分割訓練數據，程序運行良好。 這是代碼片段

subset_indices = ((train_data.train_labels == 0) + (train_data.train_labels == 1)).nonzero()
train_loader = torch.utils.data.DataLoader(train_data,batch_size=batch_size, shuffle=False,sampler=SubsetRandomSampler(subset_indices))

Answer 1

該錯誤是由於您將一個 3 維數組傳遞到函數Image.fromarray ，該函數可能設置為錯誤的模式。 您需要確保mode設置為RGB ，使其看起來像Image.fromarray(data, mode='RGB') 。