新手Pytorch入门笔记-transforms.Compose()

在这里插入图片描述
我使用的图片是上图,直接下载即可
transforms.Compose 是PyTorch中的一个实用工具,用于创建一个包含多个数据变换操作的变换对象。这些变换操作通常用于数据预处理,例如图像数据的缩放、裁剪、旋转等。使用transforms.Compose 可以将多个数据变换组合在一起,以便将它们应用于数据。

在这个示例中,transforms.Compose被用于调整图像大小并将图像转换为张量,然后我们单独对张量执行了 unsqueeze(0) 操作以添加批次维度。这种分开执行的方式更清晰,并且通常更容易维护和理解。


from PIL import Image
from torchvision import transforms
import torch
from  torchvision.io import read_image
input_path = "D:\Document\ROG Strix 2019_1920x1080.png"
img_tensor = read_image(input_path)
print(img_tensor)


input_data=Image.open(input_path)

# 定义数据预处理操作
data_transforms = transforms.Compose([
        transforms.Resize((224, 224)),# 调整图像大小
        transforms.ToTensor(),#转化为张量
        # 归一化至 [0, 1] 范围内(假设图像为 RGB)
       transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
])
# 应用数据预处理操作到输入图像
transformed_data = data_transforms(input_data)
# 单独执行 unsqueeze 操作
transformed_data = transformed_data.unsqueeze(0)
print(transformed_data)

相关推荐

  1. pytorch入门笔记

    2024-04-27 01:34:05       28 阅读
  2. CSS新手入门笔记整理:CSS列表样式

    2024-04-27 01:34:05       36 阅读

最近更新

  1. TCP协议是安全的吗?

    2024-04-27 01:34:05       16 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2024-04-27 01:34:05       16 阅读
  3. 【Python教程】压缩PDF文件大小

    2024-04-27 01:34:05       15 阅读
  4. 通过文章id递归查询所有评论(xml)

    2024-04-27 01:34:05       18 阅读

热门阅读

  1. 什么是零拷贝?

    2024-04-27 01:34:05       12 阅读
  2. 【QEMU系统分析之启动篇(八)】

    2024-04-27 01:34:05       10 阅读
  3. 【C++】1 两数之和

    2024-04-27 01:34:05       11 阅读
  4. 算法:图遍历

    2024-04-27 01:34:05       12 阅读
  5. 启动MySQL服务后

    2024-04-27 01:34:05       13 阅读
  6. MySQL生成日期序列与表关联的 SQL 查询

    2024-04-27 01:34:05       15 阅读
  7. Scrapy

    Scrapy

    2024-04-27 01:34:05      12 阅读
  8. 【嵌入式笔试题】C语言笔试题(4)

    2024-04-27 01:34:05       11 阅读
  9. C# 异步编程模型(APM)的深入理解

    2024-04-27 01:34:05       11 阅读
  10. VUE DPlayer编译

    2024-04-27 01:34:05       14 阅读
  11. Docker

    Docker

    2024-04-27 01:34:05      11 阅读
  12. python高级进阶(四)[模块、包、异常]

    2024-04-27 01:34:05       13 阅读