Pytorch – 在Pytroch中实现图片HWC与CHW相互转换、RGB与BGR相互转换
在Pytorch中对图片数据的tensor进行HWC和CHW转换、RGB和BGR的转换是常常使用到的数据预处理操作。
1 HWC与CHW互转
HWC转CHW
假设输入图片tensor格式为BHWC，比如1x224x224x3，在Pytorch转成BCHW其实也就是简单的维度变换
x = x.permute(0, 3, 1, 2) # 从BHWC转为BCHW
CHW转HWC
那如果输入图片tensor格式为BCHW，比如1x3x224x224，与上面一样也是变换维度就行
x = x.permute(0, 2, 3, 1)
2 RGB与BGR互转
RGB转BGR
假设输入图片tensor格式为BHWC，RGB，在Pytroch中实现RGB转BGR，就是最后一个维度C进行维度变换，交换R和B的位置，具体代码如下
x = x[..., [2, 1, 0]]
BGR转RGB
假设输入图片tensor格式为BHWC，BGR，在Pytroch中实现BGR转RGB，就是最后一个维度C进行维度变换，交换R和B的位置，具体代码如下
x = x[..., [2, 1, 0]]
欢迎扫码关注我的微信公众号，及时获取文章更新
本文作者：StubbornHuang
版权声明：本文为站长原创文章，如果转载请注明原文链接！
原文标题：Pytorch – 在Pytroch中实现图片HWC与CHW相互转换、RGB与BGR相互转换
原文链接：https://www.stubbornhuang.com/2648/
发布于：2023年06月07日 13:29:53
修改于：2023年06月07日 13:29:53
声明：本站所有文章，如无特殊说明或标注，均为本站原创发布。任何个人或组织，在未征得本站同意时，禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益，可联系我们进行处理。
文章末尾
评论
72