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的网络中了。

最新文章

  1. UWP开发之控件:用WebView做聊天框
  2. csdn的app打开贴子显示空白?
  3. HTML5移动Web开发(六)——定义一个内容策略
  4. 实习笔记-3:ef实体操作错误篇
  5. Git在下搭建下环境的工具
  6. 使用VideoView开发视频总结
  7. iOS9新特性之常见关键字、泛型
  8. Android隐藏软键盘
  9. Linux下网络配置与修改Centos7为列
  10. JavaScript函数表达式与函数声明
  11. 向comboboxEdit中动态添加数据库中保存的用户自定义单位制的名称
  12. vue环境配置脚手架环境搭建vue工程目录
  13. ArcEngine获取要素数据集的容差和分辨率
  14. 20165333实验一 JAVA开发环境的熟悉
  15. [从jQuery看JavaScript]-变量与作用域链
  16. angular的路由和监听路由的变化和用户超时的监听
  17. Myeclipse中设置jsp页面的默认编码格式
  18. C++继承具体解释之二——派生类成员函数具体解释(函数隐藏、构造函数与兼容覆盖规则)
  19. drill 试用
  20. MYSQL 创建常见问题

热门文章

  1. P4158[SCOI2009]粉刷匠
  2. 并行开发 2.task
  3. 攻防世界--no-strings-attached
  4. ToolStripComboBox 绑定数据
  5. 条款2:尽量使用const ,enum,inline替换define
  6. Calendar日历
  7. C++链接器
  8. 前端每日实战:76# 视频演示如何用纯 CSS 创作一组单元素办公用品(内含2个视频)
  9. python方法参数:*和**操作符
  10. SpringBoot---概述