numpy.ndarray 增加一维

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

before  (512,512)

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

after   (512,512,1)

猜你喜欢

转载自blog.csdn.net/a362682954/article/details/81220035