torchvision包是服務(wù)于PyTorch深度學(xué)習(xí)框架的，主要用來(lái)構(gòu)建計(jì)算機(jī)視覺(jué)模型。

torchvision 主要由以下幾部分構(gòu)成：

• torchvision.datasets：一些加載數(shù)據(jù)的函數(shù)及常用的數(shù)據(jù)集接口；
• torchvision.models：包含常用的模型結(jié)構(gòu)（含預(yù)訓(xùn)練模型），例如AlexNet，VGG，ResNet等；
• torchvision.transforms：常用的圖片變換，例如裁剪、旋轉(zhuǎn)等；
• torchvision.utils：其他的一些有用的方法；

總結(jié)

以上是生活随笔為你收集整理的torchvision包的主要构成的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。