图像通道转换[c h w] 转 [h w c]

cv2使用方法imread读出来图片的数据是[h w c] ,但是在pytorch中，我们往往使用的图片数据结构是[h w c]，

转换方法如下

import numpy as np
import cv2

img = cv2.imread(image_path)
img = np.float32(cv2.resize(img, (320, 192))) / 255
img = img.transpose(2, 0, 1)