numpy.ndarray 增加一维

使用np.newaxis为数组增加维数,函数加在哪里维数就哪里加1

before  (512,512)

label=label[:, : ,np.newaxis]

after   (512,512,1)

你可能感兴趣的:(python)