-
大小: 227KB文件類型: .whl金幣: 1下載: 0 次發布日期: 2021-01-07
- 語言: 其他
- 標簽: Pytorch??torchvision??
資源簡介
torchvision 主要包含以下三部分:
models : 提供深度學習中各種經典的網絡結構以及訓練好的模型,包括Alex Net, VGG系列、ResNet系列、Inception系列等;
datasets:提供常用的數據集加載,設計上都是繼承torch.utils.data.Dataset,主要包括MMIST、CIFAR10/100、ImageNet、COCO等;
transforms: 提供常用的數據預處理操作,主要包括對Tensor及PIL Image對象的操作。
代碼片段和文件信息
評論
共有 條評論