pytorch读取图像数据转成opencv格式实例

pytorch读取图像数据转成opencv格式方法：先转成numpy通用的格式，再将其转换成opencv格式。 pytorch读取的数据使用loaddata这类函数实现。pytorch网络输入图像的格式为（C, H, W)，就是（通道数，高，宽）而numpy中图像的格式为（H,W,C）。 那就将其通道调换一下。用到函数transpose。 转换方法如下 例如A 的格式为（c,h,w） 那么经过 A = A.transpose(1,2,0) 后就变成了（h,w,c）了 然后用语句 B= cv2.cvtColor(A,cv2.COLOR_RGB2BGR) 结果就变成opencv可用的图像了