Numpy数组随机生成/维度增加/维度转换

Posted 2023-03-01 调参小飞侠

一、numpy随机生成(np.random.random())

import numpy as np
im0 = np.random.random((640, 640, 3)) # 随机生成一个shape为(640, 640, 3)的三维数组
print('img: ', im0.shape) # img:  (640, 640, 3)

二、numpy维度转换(np.transpose())

im1 = np.transpose(im0, (2,0,1)) # 将生成的三维数组shape转换为(3, 640, 640)
print('im1: ', im1.shape) # im1:  (3, 640, 640)

三、numpy维度增加(np.newaxis)

im2 = im1[np.newaxis, :, :, :] # 将转换后的三维数组新增一个维度变为(1，3, 640, 640)	
print('im2: ', im2.shape)
im3 = im1[:, np.newaxis, :, :] # 将转换后的三维数组新增一个维度变为(3，1, 640, 640)	
print('im3: ', im3.shape)

注：有任何问题欢迎评论区一起交流讨论！