项目场景:pytorch生成图片
描述:网络输出的数组转化成可视的图片,但是维度不匹配。
问题描述
报错:TypeError: Invalid shape (3, 32, 32) for image data
原理分析:
提示:下面展示一些
代码片
。
rollaxis(a, axis, start = 0)
参数:
a 输入的数组
axis:需要改变的轴,编码从0开始
start:要把axis轴放到哪个轴前,编码从0开始,取值在最后多一位(有n+1个位置可以放)
比如三维数组:33232,想变成32323
axis:012,要变换的是0轴
start: 0123,要放在最后一位,也就是编号3之前,写start = 3
rollaxis(a, axis=0, start = 3)
img = np.rollaxis(AAE_x_hat[i], 0, 3)
参考链接
感谢:https://www.cnblogs.com/yqs-0705/p/10174586.html