transpose order error

[amberyic]

code in https://github.com/kcct-fujimotolab/3DCNN/blob/master/3dcnn.py function in loaddata()

if color: return np.array(X).transpose((0, 2, 3, 4, 1)), labels else: return np.array(X).transpose((0, 2, 3, 1)), labels

when the color is True

if color: return np.array(X).transpose((0, 2, 3, 1, 4)), labels else: return np.array(X).transpose((0, 2, 3, 1)), labels