在用pytorch处理图像矩阵的时候,我们通常会将三通道RGB图像[256,256,3] 切割成bachsize为128,大小为n*n的小块,得到[128,3,n,n]
测试时,输入图像是大小为[256,256,3]的三通道RGB图像,此时要将图像送入网络,需要经过维度变换处理,将(0,1,2),变到(2,0,1)。
经过model之后,得到输出图像的维度是[3,256,256]。要想正确显示图像,又需要图像变换处理。将(0,1,2)换成(1,2,0).
此处需要用到numpy的transpose操作.
具体我就不写了,参考这篇博客吧
numpy之转置(transpose)和轴对换https://www.jianshu.com/p/90d4cdfa4000
网友评论