Pytorch transform
Pytorch
torchvision介绍
torchvision是pytorch的一个图形库,它服务于PyTorch深度学习框架的,主要用来构建计算机视觉模型。torchvision.transforms主要是用于常见的一些图形变换。以下是torchvision的构成:
torchvision.datasets: 一些加载数据的函数及常用的数据集接口
torchvison.models :包含常用的模型结构(含预训练模型),例如AlexNet , VGG, ResNet
tor
原创
2021-12-24 19:29:26 ·
2692 阅读 ·
0 评论