图像通道转换[c h w] 转 [h w c]
cv2使用方法imread读出来图片的数据是[h w c] ,但是在pytorch中，我们往往使用的图片数据结构是[h w c]，转换方法如下import numpy as npimport cv2img = cv2.imread(image_path)img = np.float32(cv2.resize(img, (320, 192))) / 255img = img.transpose(2,
·
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)
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
4
0- 0
已为社区贡献7条内容
所有评论(0)