1. 提供主流的model,和常用数据集
2. 对 torch.utils.data.Dataset进行扩充,针对不同类别图像放入不同文件夹的数据进行读取, torchvision.datasets.ImageFolder是torch.utils.data.Dataset 的子类。
3.提供transforms
Posted 哈哈哈
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了torchvision 作用相关的知识,希望对你有一定的参考价值。
1. 提供主流的model,和常用数据集
2. 对 torch.utils.data.Dataset进行扩充,针对不同类别图像放入不同文件夹的数据进行读取, torchvision.datasets.ImageFolder是torch.utils.data.Dataset 的子类。
3.提供transforms
以上是关于torchvision 作用的主要内容,如果未能解决你的问题,请参考以下文章
torchvision 笔记:transforms.Normalize()
torchvision.transforms.Compose()详解Pytorch入门手册