torchvision.models子包包含用于处理不同任务的模型定义,包括:图像分类、逐像素语义分割、对象检测、实例分割、人物关键点检测、视频分类和光流。
import torch
import torchvision
import torchvision.models as models
from torchkeras import summary
## torch,torchvision 版本号
print("PyTorch Version: ",torch.__version__)
print("Torchvision Version: ",torchvision.__version__)
print(dir(torchvision.models)) # 查看所有的模型
# 查看模型结构
vgg16 = models.vgg16()
#print(vgg16)
resnet18 = models.resnet18()
#print(resnet18)
alexnet = models.alexnet()
#print(alexnet)
densenet = models.densenet161()
#print(densenet)
inception = models.inception_v3()
#print(inception)
googlenet = models.googlenet()
#print(googlenet)
###模型介绍及预训练参数下载
# https://pytorch.org/vision/stable/models.html
参考: