随机weights创建model

import torchvision.models as models
resnet18 = models.resnet18()
alexnet = models.alexnet()
vgg16 = models.vgg16()
squeezenet = models.squeezenet1_0()
densenet = models.densenet161()
inception = models.inception_v3()
googlenet = models.googlenet()
shufflenet = models.shufflenet_v2_x1_0()
mobilenet = models.mobilenet_v2()
resnext50_32x4d = models.resnext50_32x4d()
wide_resnet50_2 = models.wide_resnet50_2()
mnasnet = models.mnasnet1_0()

pre-trained创建model

import torchvision.models as models
resnet18 = models.resnet18(pretrained=True)
alexnet = models.alexnet(pretrained=True)
squeezenet = models.squeezenet1_0(pretrained=True)
vgg16 = models.vgg16(pretrained=True)
densenet = models.densenet161(pretrained=True)
inception = models.inception_v3(pretrained=True)
googlenet = models.googlenet(pretrained=True)
shufflenet = models.shufflenet_v2_x1_0(pretrained=True)
mobilenet = models.mobilenet_v2(pretrained=True)
resnext50_32x4d = models.resnext50_32x4d(pretrained=True)
wide_resnet50_2 = models.wide_resnet50_2(pretrained=True)
mnasnet = models.mnasnet1_0(pretrained=True)

这些权重系数将被保存在.cache\torch\checkpoints\中，通过TORCH_MODEL_ZOO来更改。
一些模型坑你有不同的训练与验证的行为，比如batch normalization。所以在不同的modes中切换，使用mode.train()和model.eval()
所有的pre-trained模型传入的图像都是经过normalized，比如mini-batches of 3-channel RGB(3HW)，图像会被normalized成[0,1]，然后使用mean=[0.485,0.456,0.406]和std = [0.299,0.244,0.225]

normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406],std=[0.229, 0.224, 0.225])

import torch
from torchvision import datasets, transforms as Ttransform = T.Compose([T.Resize(256), T.CenterCrop(224), T.ToTensor()])
dataset = datasets.ImageNet(".", split="train", transform=transform)means = []
stds = []
for img in subset(dataset):means.append(torch.mean(img))stds.append(torch.std(img))mean = torch.mean(torch.tensor(means))
std = torch.mean(torch.tensor(stds))

参考：
https://pytorch.org/docs/stable/torchvision/models.html

TORCHVISION MODELS相关推荐

使用torchvision.models.inception_v3(pretrained=True)加载预训练的模型每次都特别慢
欢迎大家关注笔者,你的关注是我持续更博的最大动力原创文章,转载告知,盗版必究使用torchvision.models.inception_v3(pretrained=True)加载预训练的模型每次 ...
PyTorch torchvision.models小结
前言 PyTorch框架中有一个非常重要且好用的包:torchvision,该包主要由3个子包组成,分别是: torchvision.datasets torchvision.models torch ...
PyTorch 1.0 中文文档：torchvision.models
译者:BXuan694 models子包定义了以下模型架构: AlexNet VGG ResNet SqueezeNet DenseNet Inception v3 你可以通过调用以下构造函数构造随机 ...
使用tensorwatchimport pdb import tensorwatch as tw import torchvision.models alexnet_model = torchv可视化
pip install tensorwatch 使用: import pdb import tensorwatch as tw import torchvision.models alexnet_mo ...
PyTorch源码解读之torchvision.models
PyTorch框架中有一个非常重要且好用的包:torchvision,该包主要由3个子包组成,分别是:torchvision.datasets.torchvision.models.torchvisi ...
使用torchvision.models
调用torchvision.models torchvision地址 torchvision有一些可以使用的模型可以直接导入 PyTorch框架中有一个非常重要且好用的包:torchvision,该包 ...
pytorch torchvision.models
torchvision.models 模型子包包含以下模型体系结构的定义: 目录 torchvision.models Alexnet VGG ResNet SqueezeNet DenseNet I ...
No module named ‘torchvision.models.feature_extraction‘
No module named 'torchvision.models.feature_extraction'解决办法解决办法一: 首先有这样几种可能,是因为所用的torch和torch visio ...
学习torchvision.models._dict_
torchvision.models._dict_[arch](pretrained=False)与python _dict_ 的关系之前一直在写面向过程的C代码,转换到Python面向对象了有 ...

TORCHVISION MODELS

随机weights创建model

pre-trained创建model

TORCHVISION MODELS相关推荐

最新文章

热门文章