在讀入多張影象並且在第3維上進行合併時（image[:, :, :]），出現IndexError: too many indices for array錯誤

data1 = imread('E:/dataset/Train_gray/2011_09_26_drive_0001_extract0000000000.png')
data2 = imread('E:/dataset/Train_gray/2011_09_26_drive_0001_extract0000000001.png')
data=np.concatenate((data1[:,:,0],data2[:,:,0]),axis=2)

錯誤資訊提示如下

Traceback (most recent call last):
  File "E:/project/GAN/test.py", line 28, in <module>
    data=np.concatenate((data1[:,:,0],data2[:,:,0]),axis=2)

IndexError: too many indices for array

原因在於讀入方式，當cv2.imread錯誤得到解除

data1 = cv2.imread('E:/dataset/Train_gray/2011_09_26_drive_0001_extract0000000000.png')
data2 = cv2.imread('E:/dataset/Train_gray/2011_09_26_drive_0001_extract0000000001.png')
data=np.concatenate((data1[:,:,0:1],data2[:,:,0:1]),axis=2)
print(data.shape)

控制檯輸出結果如下

E:\anaconda1\python.exe "E:/project/GAN/test.py"
(256, 512, 2)

Process finished with exit code 0