11、torchvision中的数据集的使用

transforms要用来处理数据集中的每一张图片,

本节主要讲解怎么把数据集和transforms结合在一起;在实际应用中该如何去下载数据集,如何去组织,查看以及使用

去pytorch官网,找到torchvision模块,里面会有数据集的api,使用的时候可以利用一些参数直接自动下载

1、torchvision models

  主要提供一些比较常见的神经网络,比如分类的,语义分割的,目标检测,视频分类等模型

2、torchvision datasets如何与transforms联合使用?

去pytorch官网上找内置数据集,点开你想要的数据集,里面会有方法以及参数设置

 

 

 

 

 原始的图片是PIL.Image,如果和pytorch使用的话需要转换成tensor数据类型

将数据集中的每一个部分都转换成tensor类型,使用transform

 

 如果有的数据集没有显示下载地址,可以按住ctrl+数据集名称,查看它的源代码

 

posted @ 2023-02-17 21:09  bokeAR  阅读(60)  评论(0)    收藏  举报