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
声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。
评论
51