您当前的位置:首页 > IT编程 > python
| C语言 | Java | VB | VC | python | Android | TensorFlow | C++ | oracle | 学术与代码 | cnn卷积神经网络 | gnn | 图像修复 | Keras | 数据集 | Neo4j | 自然语言处理 | 深度学习 | 医学CAD | 医学影像 | 超参数 | pointnet | pytorch | 异常检测 | Transformers | 情感分类 | 知识图谱 |

自学教程:Pytorch使用transforms

51自学网 2022-02-21 10:37:19
  python
这篇教程Pytorch使用transforms写得很实用,希望能帮到您。

首先,这次讲解的tansforms功能,通俗地讲,类似于在计算机视觉流程里的图像预处理部分的数据增强。

transforms的原理:

说明:图片(输入)通过工具得到结果(输出),这个工具,就是transforms模板工具,(tool=transforms.ToTensor()具体工具),使用工具result=tool(图片)

tansforms的调用与使用,由下图可得:

  • 先创建一个transforms.Tensor(),使用from torchvision import transforms调包
  • transforms去调init函数
  • init去调用真正的transforms类,里面就有很多的方法(绿色五角星标注),例如:resize,ToTensor,CenterCrop(从这些方法可以看出,许多都是数据增强的方法)。

接下来,上代码:

import osfrom torchvision import transformsfrom PIL import Imageroot_path = "D://data//basic//Image"label_path = "aligned"# 1.获取aligned第一张图的名字img_dir = os.path.join(root_path, label_path)img_list = os.listdir(img_dir)img_path = img_list[0]# 2.获取aligned第一张图的路径img = os.path.join(root_path, label_path, img_path)# 3.使用python自带的PIL获取图片img = Image.open(img)# 4.将PIL利用transforms转换成ToTensorto_tensor = transforms.ToTensor()  # 创建totensor ()img = to_tensor(img)  # 使用to_tensor直接将图片的PIL转化为tensorprint(img)# transforms

代码结果:


Python 如何手动编写一个自己的LRU缓存装饰器的方法实现
Python实现推送百度链接的示例代码
万事OK自学网:51自学网_软件自学网_CAD自学网自学excel、自学PS、自学CAD、自学C语言、自学css3实例,是一个通过网络自主学习工作技能的自学平台,网友喜欢的软件自学网站。