torchvision模型和预训练参数

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

参考:

https://blog.csdn.net/qq_41542989/article/details/122929126

https://pytorch.org/vision/stable/models.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值