注意:使用cv2.imread()读取彩色图片时,OpenCV获得的张量的数据顺序为h*w*c,其中张量的最后一个维度才是通道,所以在送入到torch中之前,需要对张量的维度顺序进行变换;
一个可行的方式为
img = img.transpose(2, 0, 1)
# 变换前: np.ndarray h*w*c
# 变换后: np.ndarray c*h*w
注意:使用cv2.imread()读取彩色图片时,OpenCV获得的张量的数据顺序为h*w*c,其中张量的最后一个维度才是通道,所以在送入到torch中之前,需要对张量的维度顺序进行变换;
一个可行的方式为
img = img.transpose(2, 0, 1)
# 变换前: np.ndarray h*w*c
# 变换后: np.ndarray c*h*w