目标检测——使用OpenCV读取图片要注意进行维度变换

注意：使用cv2.imread()读取彩色图片时，OpenCV获得的张量的数据顺序为h*w*c，其中张量的最后一个维度才是通道，所以在送入到torch中之前，需要对张量的维度顺序进行变换；

一个可行的方式为

img = img.transpose(2, 0, 1)
# 变换前: np.ndarray h*w*c
# 变换后: np.ndarray c*h*w

来源：CSDN

作者：songyuc

链接：https://blog.csdn.net/songyuc/article/details/103601790