【解决错误】ValueError: axes don't match array

Original Traceback (most recent call last):
  File "/home/cody/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 178, in _worker_loop
    data = fetcher.fetch(index)
  File "/home/cody/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/home/cody/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 44, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/home/cody/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/utils/data/dataset.py", line 256, in __getitem__
    return self.dataset[self.indices[idx]]
  File "/home/cody/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/utils/data/dataset.py", line 206, in __getitem__
    return self.datasets[dataset_idx][sample_idx]
  File "/home/cody/PycharmProjects/ONET/Facial_Landmark_Detection/train_test/datasets.py", line 314, in __getitem__
    image_t = np.transpose(image_t, (2, 0, 1))
  File "<__array_function__ internals>", line 6, in transpose
  File "/home/cody/anaconda3/envs/pytorch/lib/python3.6/site-packages/numpy/core/fromnumeric.py", line 650, in transpose
    return _wrapfunc(a, 'transpose', axes)
  File "/home/cody/anaconda3/envs/pytorch/lib/python3.6/site-packages/numpy/core/fromnumeric.py", line 61, in _wrapfunc
    return bound(*args, **kwds)
ValueError: axes don't match array

解决办法：

image = np.array(Image.open(img_name))

更改为：

image = np.array(Image.open(img_name).convert("RGB"))

图片文件下的图，有的是单通道，有的是三通道，转化下读图模式即可

来源：CSDN

作者：不会飞的鹰08

链接：https://blog.csdn.net/weixin_41449637/article/details/103578188

标签

axes

python3