Advertisement

[pytorch]-torchvision.transformsCompose()介绍及其代码示例解析数据变换过程

阅读量:

简介

tprchvision属于PyTorch的一个图形库其旨在服务于PyTorch的深度学习框架主要用于构建计算机视觉模型而torchvision则主要用于执行一系列常见的图形变换操作以下是torchvision的主要组成部分

  • 1.torchvision.datasets:提供了一系列用于加载数据的函数及其常见接口。
  • 2.torchvision.models:涵盖了系列经典的模型架构设计:如AlexNet、VGG、ResNet等预训练模型。
  • 3.torchvision.transforms:包括一系列常见的图片处理操作包括裁剪、旋转等多种变换。
  • 4.torchvision.utils:提供了若干其他有用的辅助功能。
    本文的核心内容聚焦于 torch.utils imaging Compose() 类,
    该类的主要功能在于将多个图像处理操作依次连接起来进行处理。
    其构造过程较为直观且易于理解。
复制代码
    class torchvision.transforms.Compose(transforms):
     # Composes several transforms together.
     # Parameters: transforms (list of Transform objects) – list of transforms t

全部评论 (0)

还没有任何评论哟~