Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Cadene/pretrained-models.pytorch
/ functions
Functions
373 in github.com/Cadene/pretrained-models.pytorch
⨍
Functions
373
◇
Types & classes
120
Method
forward
(self, x)
pretrainedmodels/models/polynet.py:339
Method
forward
(self, x)
pretrainedmodels/models/polynet.py:455
Method
forward
(self, x)
pretrainedmodels/models/cafferesnet.py:42
Method
forward
(self, x)
pretrainedmodels/models/cafferesnet.py:77
Method
forward
(self, x)
pretrainedmodels/models/cafferesnet.py:162
Method
forward
(self, x)
pretrainedmodels/models/nasnet_mobile.py:55
Method
forward
(self, x)
pretrainedmodels/models/nasnet_mobile.py:69
Method
forward
(self, x)
pretrainedmodels/models/nasnet_mobile.py:87
Method
forward
(self, x)
pretrainedmodels/models/nasnet_mobile.py:105
Method
forward
(self, x)
pretrainedmodels/models/nasnet_mobile.py:131
Method
forward
(self, x)
pretrainedmodels/models/nasnet_mobile.py:147
Method
forward
(self, x)
pretrainedmodels/models/nasnet_mobile.py:183
Method
forward
(self, x_conv0, x_stem_0)
pretrainedmodels/models/nasnet_mobile.py:248
Method
forward
(self, x, x_prev)
pretrainedmodels/models/nasnet_mobile.py:318
Method
forward
(self, x, x_prev)
pretrainedmodels/models/nasnet_mobile.py:381
Method
forward
(self, x, x_prev)
pretrainedmodels/models/nasnet_mobile.py:435
Method
forward
(self, x, x_prev)
pretrainedmodels/models/nasnet_mobile.py:493
Method
forward
(self, input)
pretrainedmodels/models/nasnet_mobile.py:612
Method
forward
(self, x)
pretrainedmodels/models/wideresnet.py:65
Method
forward
(self, input)
pretrainedmodels/models/bninception.py:492
Method
forward
(self, x)
pretrainedmodels/models/fbresnet/resnet152_load.py:39
Method
forward
(self, x)
pretrainedmodels/models/fbresnet/resnet152_load.py:74
Method
forward
(self, input)
pretrainedmodels/models/resnext_features/resnext101_64x4d_features.py:22
Method
forward
(self, input)
pretrainedmodels/models/resnext_features/resnext101_64x4d_features.py:30
Method
forward
(self, input)
pretrainedmodels/models/resnext_features/resnext101_64x4d_features.py:38
Method
forward
(self, input)
pretrainedmodels/models/resnext_features/resnext101_32x4d_features.py:22
Method
forward
(self, input)
pretrainedmodels/models/resnext_features/resnext101_32x4d_features.py:30
Method
forward
(self, input)
pretrainedmodels/models/resnext_features/resnext101_32x4d_features.py:38
Method
get_number_classes
(self)
pretrainedmodels/datasets/voc.py:260
Function
identity
(x)
pretrainedmodels/models/resnext_features/resnext101_64x4d_features.py:41
Function
identity
(x)
pretrainedmodels/models/resnext_features/resnext101_32x4d_features.py:41
Function
inceptionv3
r"""Inception v3 model architecture from `"Rethinking the Inception Architecture for Computer Vision" <http://arxiv.org/abs/1512.00567>`_.
pretrainedmodels/models/torchvision_models.py:252
Function
inner
(b=1, bsize=1, tsize=None)
pretrainedmodels/datasets/utils.py:70
Function
logits
(self, features)
pretrainedmodels/models/torchvision_models.py:152
Function
nasnetalarge
r"""NASNetALarge model architecture from the `"NASNet" <https://arxiv.org/abs/1707.07012>`_ paper.
pretrainedmodels/models/nasnet.py:608
Function
nasnetamobile
r"""NASNetALarge model architecture from the `"NASNet" <https://arxiv.org/abs/1707.07012>`_ paper.
pretrainedmodels/models/nasnet_mobile.py:618
Function
pnasnet5large
r"""PNASNet-5 model architecture from the `"Progressive Neural Architecture Search" <https://arxiv.org/abs/1712.00559>`_ paper.
pretrainedmodels/models/pnasnet.py:372
Function
polynet
PolyNet architecture from the paper 'PolyNet: A Pursuit of Structural Diversity in Very Deep Networks' https://arxiv.org/abs/1611.05725
pretrainedmodels/models/polynet.py:461
Function
print_info
(self, input, output)
examples/visu_arch.py:75
Function
resnet101
Constructs a ResNet-101 model.
pretrainedmodels/models/torchvision_models.py:378
Function
resnet101
Constructs a ResNet-101 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
pretrainedmodels/models/fbresnet/resnet152_load.py:196
Function
resnet152
Constructs a ResNet-152 model.
pretrainedmodels/models/torchvision_models.py:388
Function
resnet18
Constructs a ResNet-18 model.
pretrainedmodels/models/torchvision_models.py:348
Function
resnet18
Constructs a ResNet-18 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
pretrainedmodels/models/fbresnet/resnet152_load.py:160
Function
resnet34
Constructs a ResNet-34 model.
pretrainedmodels/models/torchvision_models.py:358
Function
resnet34
Constructs a ResNet-34 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
pretrainedmodels/models/fbresnet/resnet152_load.py:172
Function
resnet50
Constructs a ResNet-50 model.
pretrainedmodels/models/torchvision_models.py:368
Function
resnet50
Constructs a ResNet-50 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
pretrainedmodels/models/fbresnet/resnet152_load.py:184
Function
resnext101_32x4d
(num_classes=1000, pretrained='imagenet')
pretrainedmodels/models/resnext.py:79
Function
resnext101_64x4d
(num_classes=1000, pretrained='imagenet')
pretrainedmodels/models/resnext.py:93
Function
save_activation
(self, input, output)
examples/visu_arch.py:177
Function
se_resnet101
(num_classes=1000, pretrained='imagenet')
pretrainedmodels/models/senet.py:401
Function
se_resnet152
(num_classes=1000, pretrained='imagenet')
pretrainedmodels/models/senet.py:412
Function
se_resnet50
(num_classes=1000, pretrained='imagenet')
pretrainedmodels/models/senet.py:390
Function
se_resnext101_32x4d
(num_classes=1000, pretrained='imagenet')
pretrainedmodels/models/senet.py:434
Function
se_resnext50_32x4d
(num_classes=1000, pretrained='imagenet')
pretrainedmodels/models/senet.py:423
Function
senet154
(num_classes=1000, pretrained='imagenet')
pretrainedmodels/models/senet.py:381
Function
squeezenet1_0
r"""SqueezeNet model architecture from the `"SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size" <https://arxi
pretrainedmodels/models/torchvision_models.py:428
Function
squeezenet1_1
r"""SqueezeNet 1.1 model from the `official SqueezeNet repo <https://github.com/DeepScale/SqueezeNet/tree/master/SqueezeNet_v1.1>`_. SqueezeNe
pretrainedmodels/models/torchvision_models.py:440
Function
test_pm_imagenet
(model_name, pretrained)
tests/test_pm_imagenet.py:24
Function
test_torch_save
(model_name, pretrained, tmp_path)
tests/test_torch_save.py:15
Method
value
Returns the model's average precision for each class Return: ap (FloatTensor): 1xK tensor, with avg precision for each class k
pretrainedmodels/datasets/utils.py:158
Function
vgg11
VGG 11-layer model (configuration "A")
pretrainedmodels/models/torchvision_models.py:495
Function
vgg11_bn
VGG 11-layer model (configuration "A") with batch normalization
pretrainedmodels/models/torchvision_models.py:505
Function
vgg13
VGG 13-layer model (configuration "B")
pretrainedmodels/models/torchvision_models.py:515
Function
vgg13_bn
VGG 13-layer model (configuration "B") with batch normalization
pretrainedmodels/models/torchvision_models.py:525
Function
vgg16
VGG 16-layer model (configuration "D")
pretrainedmodels/models/torchvision_models.py:535
Function
vgg16_bn
VGG 16-layer model (configuration "D") with batch normalization
pretrainedmodels/models/torchvision_models.py:545
Function
vgg19
VGG 19-layer model (configuration "E")
pretrainedmodels/models/torchvision_models.py:555
Function
vgg19_bn
VGG 19-layer model (configuration 'E') with batch normalization
pretrainedmodels/models/torchvision_models.py:565
Function
vggm
(num_classes=1000, pretrained='imagenet')
pretrainedmodels/models/vggm.py:105
Function
wideresnet50
(pooling)
pretrainedmodels/models/wideresnet.py:70
Function
xception
(num_classes=1000, pretrained='imagenet')
pretrainedmodels/models/xception.py:216
← previous
301–373 of 373, ranked by callers