[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)
还没有任何评论哟~
