m0_64936084的博客

原创 使用opencv读图片是需要注意的问题

直接使用cv2.imread()读取彩色图片，获得的张量的数据顺序为h*w*c，其中张量的最后一个维度才是通道，会出现报错 Given groups=1, weight of size [48, 3, 6, 6], expected input[1, 640, 640, 3] to have 3 channels, but got 640 channels instead 需要对张量的维度顺序进行变换： img = img.transpose(2, 0, 1) # 变换前: np.ndarray h*