分类：opencv格式

2020
10-01

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)结果... 继续阅读 >