栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

小知识点【python rollaxis()函数。TypeError: Invalid shape (3, 32, 32) for image data】

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

小知识点【python rollaxis()函数。TypeError: Invalid shape (3, 32, 32) for image data】

项目场景: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

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/980349.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号