1. generación aleatoria 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)
Conversión de dos dimensiones numpy (np.transpose())
im1 = np.transpose(im0, (2,0,1)) # 将生成的三维数组shape转换为(3, 640, 640)
print('im1: ', im1.shape) # im1: (3, 640, 640)
Tres, aumento de dimensión 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)
Nota: Si tiene alguna pregunta, ¡bienvenido a discutir en el área de comentarios!