pytorch与numpy中的通道交换问题
2024-10-21 20:45:30
pytorch网络输入图像的格式为(C, H, W),而numpy中的图像的shape为(H,W,C)
所以一般需要变换通道,将numpy中的shape变换为torch中的shape。
方法如下:
# A是numpy数据类型
A = A.transpose(0,1,2) # 没有改变:(h,w,c)
A = A.transpose(2,0,1) # 转换为:(c,h,w)
这样就可以直接输入到torch的网络中了。
最新文章
- UWP开发之控件:用WebView做聊天框
- csdn的app打开贴子显示空白?
- HTML5移动Web开发(六)——定义一个内容策略
- 实习笔记-3:ef实体操作错误篇
- Git在下搭建下环境的工具
- 使用VideoView开发视频总结
- iOS9新特性之常见关键字、泛型
- Android隐藏软键盘
- Linux下网络配置与修改Centos7为列
- JavaScript函数表达式与函数声明
- 向comboboxEdit中动态添加数据库中保存的用户自定义单位制的名称
- vue环境配置脚手架环境搭建vue工程目录
- ArcEngine获取要素数据集的容差和分辨率
- 20165333实验一 JAVA开发环境的熟悉
- [从jQuery看JavaScript]-变量与作用域链
- angular的路由和监听路由的变化和用户超时的监听
- Myeclipse中设置jsp页面的默认编码格式
- C++继承具体解释之二——派生类成员函数具体解释(函数隐藏、构造函数与兼容覆盖规则)
- drill 试用
- MYSQL 创建常见问题