Numpy 配列のランダム生成/次元増加/次元変換

1. 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)

2 次元変換 (np.transpose())

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

3、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)

注: ご質問がある場合は、コメント欄でご相談ください。