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
当前分类随机文章推荐
- Pytorch - RuntimeError: No rendezvous handler for env://错误 阅读1540次,点赞0次
- Pytorch - torch.nn.Module的parameters()和named_parameters() 阅读801次,点赞0次
- Pytorch - torch.cat参数详解与使用 阅读2170次,点赞1次
- Pytorch - .to()和.cuda()的区别 阅读1227次,点赞0次
- Pytorch - Pytoch结合Tensorboard实现数据可视化 阅读442次,点赞0次
- Python - list/numpy/pytorch tensor相互转换 阅读1970次,点赞0次
- Pytorch - reshape和view的用法和区别 阅读782次,点赞0次
- Pytorch - 用Pytorch实现ResNet 阅读1006次,点赞0次
- Pytorch - 多GPU训练方式nn.DataParallel与nn.parallel.DistributedDataParallel的区别 阅读1490次,点赞0次
- Pytorch - 使用opencv-python解码视频文件并将视频帧转换为Pytorch tensor作为网络模型输入数据 阅读3061次,点赞0次
全站随机文章推荐
- VTK - 冠脉重建点匹配坐标数据下载 阅读4141次,点赞5次
- 资源分享 - Computational Geometry - An Introduction Through Randomized Algorithms 英文高清PDF下载 阅读1859次,点赞0次
- 资源分享 - GPU Pro 360 - Guide to Lighting 英文高清PDF下载 阅读3023次,点赞0次
- 资源分享 - 实时碰撞检测算法技术,Real-Time Collision Detection中文版 PDF下载 阅读2051次,点赞0次
- 资源分享 - GPU Zen 1:Advanced Rendering Techniques英文高清PDF下载 阅读6416次,点赞0次
- 资源分享 - OpenGL Development Cookbook 英文高清PDF下载 阅读1638次,点赞0次
- Pytorch - torch.cat参数详解与使用 阅读2170次,点赞1次
- WordPress - 防止用户在登录页面重复点击登录按钮造成重复登录 阅读1024次,点赞0次
- 资源分享 - C++标准库(第2版,侯捷译) 中文高清PDF下载 阅读5459次,点赞0次
- C++ - 拷贝构造函数与拷贝构造函数调用时机 阅读563次,点赞0次
评论
169