MCPcopy Create free account

hub / github.com/Cadene/pretrained-models.pytorch / functions

Functions373 in github.com/Cadene/pretrained-models.pytorch

↓ 21 callersFunctionload_pretrained
(model, num_classes, settings)
pretrainedmodels/models/torchvision_models.py:113
↓ 16 callersMethod__init__
(self)
pretrainedmodels/models/polynet.py:69
↓ 13 callersMethodfeatures
(self, x)
pretrainedmodels/models/senet.py:347
↓ 12 callersMethod__init__
(self, stem_filters, num_filters=42)
pretrainedmodels/models/nasnet.py:138
↓ 12 callersMethod__init__
(self, stem_filters, num_filters=42)
pretrainedmodels/models/nasnet_mobile.py:160
↓ 10 callersMethodupdate
(self, val, n=1)
examples/imagenet_eval.py:273
↓ 9 callersMethod__init__
(self)
pretrainedmodels/models/inceptionv4.py:57
↓ 9 callersMethodlogits
(self, features)
pretrainedmodels/models/dpn.py:375
↓ 8 callersFunctionmodify_vggs
(model)
pretrainedmodels/models/torchvision_models.py:456
↓ 7 callersMethod__init__
(self, scale=1.0, noReLU=False)
pretrainedmodels/models/inceptionresnetv2.py:205
↓ 7 callersMethod__init__
(self, in_channels_left, out_channels_left, in_channels_right, out_channels_right, is_reducti
pretrainedmodels/models/pnasnet.py:228
↓ 6 callersFunctioninitialize_pretrained_model
(model, num_classes, settings)
pretrainedmodels/models/senet.py:369
↓ 5 callersMethod__init__
(self, small=False, num_init_features=64, k_r=96, groups=32, b=False, k_sec=(3, 4, 20, 3), in
pretrainedmodels/models/dpn.py:313
↓ 5 callersFunctionmodify_resnets
(model)
pretrainedmodels/models/torchvision_models.py:314
↓ 4 callersMethod__init__
Parameters ---------- block (nn.Module): Bottleneck class. - For SENet154: SEBottleneck - For SE-ResN
pretrainedmodels/models/senet.py:209
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, groups, reduction, stride=1, downsample_kernel_size=1, downs
pretrainedmodels/models/senet.py:327
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
pretrainedmodels/models/fbresnet.py:133
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
pretrainedmodels/models/cafferesnet.py:127
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
pretrainedmodels/models/fbresnet/resnet152_load.py:124
↓ 4 callersFunctionconv2d
(input, params, base, stride=1, pad=0)
pretrainedmodels/models/wideresnet.py:16
↓ 4 callersFunctiongroup
(input, params, base, stride, n)
pretrainedmodels/models/wideresnet.py:20
↓ 4 callersFunctioninceptionresnetv2
r"""InceptionResNetV2 model architecture from the `"InceptionV4, Inception-ResNet..." <https://arxiv.org/abs/1602.07261>`_ paper.
pretrainedmodels/models/inceptionresnetv2.py:333
↓ 4 callersFunctioninceptionv4
(num_classes=1000, pretrained='imagenet')
pretrainedmodels/models/inceptionv4.py:314
↓ 4 callersFunctionmodify_densenets
(model)
pretrainedmodels/models/torchvision_models.py:183
↓ 3 callersMethod__init__
(self, *args)
pretrainedmodels/models/resnext_features/resnext101_64x4d_features.py:18
↓ 3 callersMethod__init__
(self, *args)
pretrainedmodels/models/resnext_features/resnext101_32x4d_features.py:18
↓ 3 callersMethodadd
Args: output (Tensor): NxK tensor that for each of the N examples indicates the probability of the example belong
pretrainedmodels/datasets/utils.py:110
↓ 3 callersFunctionextract_features_targets
(model, features_size, loader, path_data, cuda=False)
examples/voc2007_extract.py:27
↓ 3 callersMethodforward_prepare
(self, input)
pretrainedmodels/models/resnext_features/resnext101_64x4d_features.py:11
↓ 3 callersMethodforward_prepare
(self, input)
pretrainedmodels/models/resnext_features/resnext101_32x4d_features.py:11
↓ 2 callersMethod__init__
(self,in_filters,out_filters,reps,strides=1,start_with_relu=True,grow_first=True)
pretrainedmodels/models/xception.py:64
↓ 2 callersMethod__init__
(self, num_classes=1000)
pretrainedmodels/models/vggm.py:69
↓ 2 callersMethod__init__
(self, block, layers, num_classes=1000)
pretrainedmodels/models/fbresnet.py:105
↓ 2 callersMethod__init__
(self, block, layers, num_classes=1000)
pretrainedmodels/models/cafferesnet.py:102
↓ 2 callersMethod__init__
(self, block, layers, num_classes=1000)
pretrainedmodels/models/fbresnet/resnet152_load.py:100
↓ 2 callersFunctionaccuracy
Computes the precision@k for the specified values of k
examples/imagenet_eval.py:287
↓ 2 callersFunctionadaptive_avgmax_pool2d
Selectable global pooling function with dynamic input kernel size
pretrainedmodels/models/dpn.py:407
↓ 2 callersMethodcell_forward
(self, x_left, x_right)
pretrainedmodels/models/pnasnet.py:151
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
pretrainedmodels/models/fbresnet.py:27
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
pretrainedmodels/models/cafferesnet.py:23
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
pretrainedmodels/models/fbresnet/resnet152_load.py:20
↓ 2 callersFunctionmodify_squeezenets
(model)
pretrainedmodels/models/torchvision_models.py:401
↓ 2 callersFunctionvalidate
(val_loader, model, criterion)
examples/imagenet_eval.py:211
↓ 1 callersMethod__init__
(self, size, interpolation=Image.BILINEAR)
pretrainedmodels/datasets/utils.py:33
↓ 1 callersMethod__init__
(self, num_classes=1000)
pretrainedmodels/models/resnext.py:39
↓ 1 callersFunctionadjust_learning_rate
Sets the learning rate to the initial LR decayed by 10 every 30 epochs
examples/imagenet_eval.py:280
↓ 1 callersMethodaverage_precision
(output, target, difficult_examples=True)
pretrainedmodels/datasets/utils.py:180
↓ 1 callersFunctionbninception
r"""BNInception model architecture from <https://arxiv.org/pdf/1502.03167.pdf>`_ paper.
pretrainedmodels/models/bninception.py:497
↓ 1 callersFunctiondefine_model
(params)
pretrainedmodels/models/wideresnet.py:15
↓ 1 callersFunctiondownload_voc2007
(root)
pretrainedmodels/datasets/voc.py:114
↓ 1 callersFunctionequal
(x,y)
tests/test_pm_imagenet.py:20
↓ 1 callersFunctionf
(input, params, pooling_classif=True)
pretrainedmodels/models/wideresnet.py:41
↓ 1 callersMethodfeatures
(self, input)
pretrainedmodels/models/xception.py:172
↓ 1 callersMethodfeatures
(self, input)
pretrainedmodels/models/inceptionresnetv2.py:304
↓ 1 callersMethodfeatures
(self, x)
pretrainedmodels/models/pnasnet.py:340
↓ 1 callersMethodfeatures
(self, input)
pretrainedmodels/models/nasnet.py:563
↓ 1 callersMethodfeatures
(self, input)
pretrainedmodels/models/fbresnet.py:150
↓ 1 callersMethodfeatures
(self, x)
pretrainedmodels/models/polynet.py:439
↓ 1 callersMethodfeatures
(self, x)
pretrainedmodels/models/cafferesnet.py:144
↓ 1 callersMethodfeatures
(self, input)
pretrainedmodels/models/nasnet_mobile.py:579
↓ 1 callersMethodfeatures
(self, input)
pretrainedmodels/models/bninception.py:253
↓ 1 callersMethodforward
(self, x)
pretrainedmodels/models/fbresnet/resnet152_load.py:141
↓ 1 callersMethodforward_block
(self, x, block_index)
pretrainedmodels/models/polynet.py:259
↓ 1 callersMethodforward_block
(self, x, block_index)
pretrainedmodels/models/polynet.py:309
↓ 1 callersMethodforward_prepare
(self, input)
pretrainedmodels/models/vggm.py:57
↓ 1 callersFunctionload_imagenet_classes
(path_synsets='data/imagenet_synsets.txt', path_classes='data/imagenet_classes.txt')
pretrainedmodels/datasets/utils.py:9
↓ 1 callersMethodlogits
(self, features)
pretrainedmodels/models/xception.py:202
↓ 1 callersMethodlogits
(self, features)
pretrainedmodels/models/inceptionv4.py:300
↓ 1 callersMethodlogits
(self, features)
pretrainedmodels/models/inceptionresnetv2.py:322
↓ 1 callersMethodlogits
(self, input)
pretrainedmodels/models/resnext.py:46
↓ 1 callersMethodlogits
(self, input)
pretrainedmodels/models/resnext.py:67
↓ 1 callersMethodlogits
(self, features)
pretrainedmodels/models/pnasnet.py:358
↓ 1 callersMethodlogits
(self, x)
pretrainedmodels/models/senet.py:355
↓ 1 callersMethodlogits
(self, features)
pretrainedmodels/models/nasnet.py:594
↓ 1 callersMethodlogits
(self, features)
pretrainedmodels/models/fbresnet.py:163
↓ 1 callersMethodlogits
(self, x)
pretrainedmodels/models/polynet.py:448
↓ 1 callersMethodlogits
(self, x)
pretrainedmodels/models/cafferesnet.py:156
↓ 1 callersMethodlogits
(self, features)
pretrainedmodels/models/nasnet_mobile.py:604
↓ 1 callersMethodlogits
(self, features)
pretrainedmodels/models/bninception.py:485
↓ 1 callersFunctionmain
()
examples/voc2007_extract.py:135
↓ 1 callersFunctionmain
()
examples/imagenet_eval.py:64
↓ 1 callersFunctionmain
()
examples/imagenet_logits.py:27
↓ 1 callersFunctionmodify_alexnet
(model)
pretrainedmodels/models/torchvision_models.py:129
↓ 1 callersFunctionmy_hook
(t)
pretrainedmodels/datasets/utils.py:67
↓ 1 callersFunctionpooling_factor
(pool_type='avg')
pretrainedmodels/models/dpn.py:403
↓ 1 callersFunctionread_image_label
(file)
pretrainedmodels/datasets/voc.py:29
↓ 1 callersFunctionread_object_labels
(root, dataset, set)
pretrainedmodels/datasets/voc.py:43
↓ 1 callersFunctionread_object_labels_csv
(file, header=True)
pretrainedmodels/datasets/voc.py:82
↓ 1 callersMethodreset
Resets the meter with empty member variables
pretrainedmodels/datasets/utils.py:105
↓ 1 callersMethodreset
(self)
examples/imagenet_eval.py:267
↓ 1 callersFunctionresnet152
Constructs a ResNet-152 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
pretrainedmodels/models/fbresnet/resnet152_load.py:208
↓ 1 callersFunctionsave_checkpoint
(state, is_best, filename='checkpoint.pth.tar')
examples/imagenet_eval.py:255
↓ 1 callersFunctiontrain
(train_loader, model, criterion, optimizer, epoch)
examples/imagenet_eval.py:162
↓ 1 callersFunctiontrain_multilabel
(features, targets, classes, train_split, test_split, C=1.0, ignore_hard_examples=True, after_ReLU=False, norm
examples/voc2007_extract.py:59
↓ 1 callersFunctionupdate_state_dict
(state_dict)
pretrainedmodels/models/torchvision_models.py:98
↓ 1 callersFunctionwrite_object_labels_csv
(file, labeled_data)
pretrainedmodels/datasets/voc.py:64
Method__call__
(self, tensor)
pretrainedmodels/utils.py:14
Method__call__
(self, tensor)
pretrainedmodels/utils.py:28
Method__call__
(self, img)
pretrainedmodels/utils.py:79
Method__call__
(self, path_img)
pretrainedmodels/utils.py:89
next →1–100 of 373, ranked by callers