Pytorch使用transforms

作者:游客26024 时间:2023-11-01 20:34:33 

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

transforms的原理:

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

Pytorch使用transforms

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

  • 先创建一个transforms.Tensor(),使用from torchvision import transforms调包

  • transforms去调init函数

  • init去调用真正的transforms类,里面就有很多的方法(绿色五角星标注),例如:resize,ToTensor,CenterCrop(从这些方法可以看出,许多都是数据增强的方法)。

Pytorch使用transforms

接下来,上代码:


import os
from torchvision import transforms
from PIL import Image

root_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转换成ToTensor
to_tensor = transforms.ToTensor()  # 创建totensor ()
img = to_tensor(img)  # 使用to_tensor直接将图片的PIL转化为tensor

print(img)
# transforms

代码结果:

Pytorch使用transforms

来源:https://blog.csdn.net/XiaoyYidiaodiao/article/details/122158191

标签:Pytorch,transforms
0
投稿

猜你喜欢

  • 基于OpenCV的网络实时视频流传输的实现

    2022-10-18 03:03:57
  • Python爬虫实战之爬取携程评论

    2022-02-03 14:26:23
  • 使用keras根据层名称来初始化网络

    2021-12-09 10:51:40
  • url地址栏加密和解密函数 Base64

    2008-07-10 13:29:00
  • asp显示数据库中表名、字段名、字段内容

    2008-06-17 18:09:00
  • python求解水仙花数的方法

    2023-10-08 12:07:28
  • django如何根据现有数据库表生成model详解

    2024-01-17 02:29:21
  • Vue 使用 Mint UI 实现左滑删除效果CellSwipe

    2024-05-10 14:16:45
  • Python操作Access数据库基本步骤分析

    2024-01-18 22:32:20
  • 网页设计中的对比原则与接近性原则

    2010-03-30 14:51:00
  • vue cli+axios踩坑记录+拦截器使用方式,代理跨域proxy

    2023-07-02 16:38:30
  • Pandas实现DataFrame的简单运算、统计与排序

    2022-03-29 02:09:54
  • Python中itertools模块用法详解

    2023-05-28 05:00:27
  • php将textarea数据提交到mysql出现很多空格的解决方法

    2023-09-11 01:10:08
  • 一个免刷新页面的JavaScript日历

    2007-12-26 12:57:00
  • 5个保护MySQL数据仓库的小技巧

    2024-01-24 09:35:34
  • Linux系统中MySQL的常用操作命令

    2024-01-18 22:00:10
  • 将MSSQL Server 导入/导出到远程服务器教程的图文方法分享

    2024-01-13 21:55:42
  • 利用Python求解阿基米德分牛问题

    2023-11-10 09:36:48
  • python实现简单飞机大战小游戏

    2023-12-19 06:55:37
  • asp之家 网络编程 m.aspxhome.com