- 博客(2)
- 收藏
- 关注
原创 torchvision包介绍
torchvision.datasets:提供常用的数据集,设计上继承 torch.utils.data.Dataset,主要包括:MNIST、CIFAR10/100、ImageNet、COCO等。torchvision.models: 提供深度学习中各种经典的网络结构、预训练好的模型,如:Alex-Net、VGG、ResNet、Inception等。torchvision.models 中为我们提供了已经训练好的模型,让我们可以加载之后,直接使用。
2023-07-01 10:18:39 313
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人