pytorch 如何实现HWC转CHW
看代码吧~
import torch
import numpy as np
from torchvision.transforms import ToTensor
t = torch.tensor(np.arange(24).reshape(2,4,3))
print(t)
#HWC 转CHW
print(t.transpose(0,2).transpose(1,2))
print(t.permute(2,0,1))
print(ToTensor()(t.numpy()))
D:\anaconda\python.exe C:/Users/liuxinyu/Desktop/pytorch_test/day3/hwc转chw.py tensor([[[ 0, 1, 2], [ 3, 4, 5], [ 6, 7, 8], [ 9, 10, 11]],
[[12, 13, 14], [15, 16, 17], [18, 19, 20], [21, 22, 23]]], dtype=torch.int32) tensor([[[ 0, 3, 6, 9], [12, 15, 18, 21]],
[[ 1, 4, 7, 10], [13, 16, 19, 22]],
[[ 2, 5, 8, 11], [14, 17, 20, 23]]], dtype=torch.int32) tensor([[[ 0, 3, 6, 9], [12, 15, 18, 21]],
[[ 1, 4, 7, 10], [13, 16, 19, 22]],
[[ 2, 5, 8, 11], [14, 17, 20, 23]]], dtype=torch.int32) tensor([[[ 0, 3, 6, 9], [12, 15, 18, 21]],
[[ 1, 4, 7, 10], [13, 16, 19, 22]],
[[ 2, 5, 8, 11], [14, 17, 20, 23]]], dtype=torch.int32)
Process finished with exit code 0
补充:opencv python 把图(cv2下)BGR转RGB,且HWC转CHW
如下所示:
img = cv2.imread("001.jpg") img_ = img[:,:,::-1].transpose((2,0,1))
① 在opencv里,图格式HWC,其余都是CHW,故transpose((2,0,1))
② img[:,:,::-1]对应H、W、C,彩图是3通道,即C是3层。opencv里对应BGR,故通过C通道的 ::-1 就是把BGR转为RGB
注: [::-1] 代表顺序相反操作
③ 若不涉及C通道的BGR转RGB,如Img[:,:,0]代表B通道,也就是蓝色分量图像;Img[:,:,1]代表G通道,也就是绿色分量图像;
Img[:,:,2]代表R通道,也就是红色分量图像。
补充:python opencv 中将图像由BGR转换为CHW用于后期的深度训练
BGR HWC -> CHW 12 -> HCW 01 -> CHW
import cv2 as cv import numpy as np img = cv.imread("lenna.png") #BGR HWC -> CHW 12 -> HCW 01 -> CHW transform_img = img.swapaxes(1,2).swapaxes(0,1) print(img.shape) print(transform_img.shape) cv.imshow("image0 ",transform_img[0]) cv.imshow("image1",transform_img[1]) cv.imshow("image2",transform_img[2]) cv.waitKey(0) cv.destroyAllWindows()
(资源库 www.zyku.net)
原文链接:https://blog.csdn.net/u010970956/article/details/104338072
栏 目:Python教程
下一篇:Keras多线程机制与flask多线程冲突的解决方案
本文标题:pytorch 如何实现HWC转CHW
本文地址:https://www.zyku.net/python/9786.html
您可能感兴趣的文章
- 08-14苹果手机如何过滤未知发件人
- 01-07iQOO9怎么更换桌面布局风格
- 01-15小米12如何设置息屏样式
- 12-17华为nova5怎么设置杂志锁屏
- 12-23HTML <div> 标签
- 07-05Linux locate命令
- 04-21小米11ultra心率健康测试教程
- 05-05excel表格中自动生成手机号码
- 01-12左家右厨-左家右厨应用软件功能介绍
- 02-23Linux注释掉系统不需要的用户和用户组
- 02-18ajax实现异步文件或图片上传功能
- 11-06bose蓝牙耳机如何配对手机
- 12-31一生帮-一生帮应用软件功能介绍
- 09-19iqoo8pro在哪里查看隐藏应用
- 01-31一加8pro反向无线充电启用教程
- 03-05小米flipbudspro升级版本教程
- 12-02捏咔创作二次元人物玩法分享
- 09-27苹果音响homepod怎样配对手机
- 05-16Discuz门户网易云跟贴代码接入教程
- 02-12vivoy30投屏功能使用教程
最近更新
阅读排行
猜你喜欢
- 01-12安尼威尔摄像头app-安尼威尔摄像头app
- 09-24oppo手机退出专注模式方法介绍
- 10-28华为FreebudsLipstick耳机怎么样
- 02-18MySQL/MariaDB启动时处于进度条状态导
- 09-19oppo手机怎么找回蓝牙耳机
- 01-11金题护考-金题护考应用软件功能介绍
- 03-15opporeno6显示运动步数教程
- 01-10vlog提词神器-vlog提词神器应用软件功
- 02-09opporeno4se通讯录同步到新手机步骤
- 02-08小米11手机更改桌面图标大小方法