本文重點
在pytorch深度學習框架中,torchvision是一個非常優秀的視覺工具包,我們可以使用它加載一些著名的數據集,然后我們可以使用它來加載網絡模型,比如vgg,resnet等等,還可以使用它來預處理一些圖片數據,本節課程我們將學習一下它的使用方式。
torchvision的四部分
torchvision.datasets:流行視覺數據集的數據加載器
torchvision.models:流行的模型體系結構的定義,例如AlexNet,VGG和ResNet以及預先訓練的模型。
torchvision.transforms:常見的圖像預處理,例如隨機裁剪,旋轉等。
torchvision.utils:保存張量(3 x H x W)作為圖像到磁盤,給定一個小批量創建圖像網格等。
torchvision.datasets
torchvision.datasets的父類是torhc.utils.data.Dataset,封裝了很多常用的數據集,那么我們可以使用它來讀取一些常用的數據集,比如