MCPcopy Create free account

hub / github.com/alinlab/SelfPatch / functions

Functions386 in github.com/alinlab/SelfPatch

Method__init__
(self, in_dim, num_heads, k_num)
selfpatch_vision_transformer.py:363
Method__init__
(self, in_dim, out_dim, use_bn=False, norm_last_layer=True, nlayers=3, hidden_dim=2048, bottleneck_dim=256)
selfpatch_vision_transformer.py:446
Method__init__
(self, drop_prob=None)
detection/backbone/vit_SelfPatch.py:41
Method__init__
(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0.)
detection/backbone/vit_SelfPatch.py:69
Method__init__
(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., dr
detection/backbone/vit_SelfPatch.py:96
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768)
detection/backbone/vit_SelfPatch.py:119
Method__init__
Args: img_size (int, tuple): input image size patch_size (int, tuple): patch size in_chans (int): number
detection/backbone/vit_SelfPatch.py:142
Method__init__
(self, onnx_file: str, cfg: Any, device_id: int)
segmentation/tools/deploy_test.py:24
Method__init__
(self, radix, groups)
segmentation/backbones/resnest.py:24
Method__init__
(self, in_channels, channels, kernel_size,
segmentation/backbones/resnest.py:59
Method__init__
Bottleneck block for ResNeSt.
segmentation/backbones/resnest.py:166
Method__init__
(self, embed_dims, num_heads, attn_drop=0.,
segmentation/backbones/mit.py:118
Method__init__
(self, embed_dims, num_heads, feedforward_channels,
segmentation/backbones/mit.py:201
Method__init__
(self, in_channels=3, embed_dims=64, num_stages=4,
segmentation/backbones/mit.py:286
Method__init__
(self, in_channels, out_channels, num_convs=2,
segmentation/backbones/unet.py:46
Method__init__
(self, in_channels, out_channels, with_cp=False,
segmentation/backbones/unet.py:108
Method__init__
(self, in_channels, out_channels, with_cp=False,
segmentation/backbones/unet.py:182
Method__init__
(self, detail_channels=(64, 64, 128), in_channels=3, conv_c
segmentation/backbones/bisenetv2.py:33
Method__init__
(self, in_channels=3, out_channels=16, conv_cfg=None,
segmentation/backbones/bisenetv2.py:122
Method__init__
(self, in_channels=3, out_channels=16, conv_cfg=None,
segmentation/backbones/bisenetv2.py:322
Method__init__
(self, semantic_channels=(16, 32, 64, 128), in_channels=3,
segmentation/backbones/bisenetv2.py:383
Method__init__
(self, out_channels=128, align_corners=False, conv_cfg=None
segmentation/backbones/bisenetv2.py:454
Method__init__
(self, in_channels=3, detail_channels=(64, 64, 128), semant
segmentation/backbones/bisenetv2.py:579
Method__init__
(self, channel, reduction=16, with_cp=False)
segmentation/backbones/cgnet.py:27
Method__init__
(self, in_channels, out_channels, dilation=2,
segmentation/backbones/cgnet.py:79
Method__init__
(self, num_downsampling)
segmentation/backbones/cgnet.py:174
Method__init__
(self, arch='small', conv_cfg=None, norm_cfg=dict(type='BN'
segmentation/backbones/mobilenet_v3.py:73
Method__init__
(self, inplanes, planes, groups=1, base_wi
segmentation/backbones/resnext.py:19
Method__init__
(self, in_channels=3, num_channels=(64, 64, 64, 128), conv_
segmentation/backbones/bisenetv1.py:25
Method__init__
(self, in_channels, out_channel, conv_cfg=None,
segmentation/backbones/bisenetv1.py:94
Method__init__
(self, backbone_cfg, context_channels=(128, 256, 512), alig
segmentation/backbones/bisenetv1.py:147
Method__init__
(self, in_channels, out_channels, conv_cfg=None,
segmentation/backbones/bisenetv1.py:224
Method__init__
(self, num_branches, blocks, num_blocks, i
segmentation/backbones/hrnet.py:21
Method__init__
(self, in_channels, dw_channels, out_channels,
segmentation/backbones/fast_scnn.py:32
Method__init__
(self, in_channels=64, block_channels=(64, 96, 128), out_ch
segmentation/backbones/fast_scnn.py:116
Method__init__
(self, higher_in_channels, lower_in_channels, out_channels,
segmentation/backbones/fast_scnn.py:215
Method__init__
(self, embed_dims, num_heads, window_size,
segmentation/backbones/swin.py:148
Method__init__
(self, embed_dims, num_heads, feedforward_channels,
segmentation/backbones/swin.py:310
Method__init__
(self, embed_dims, num_heads, feedforward_channels,
segmentation/backbones/swin.py:406
Method__init__
(self, pretrain_img_size=224, in_channels=3, embed_dims=96,
segmentation/backbones/swin.py:515
Method__init__
(self, embed_dims, num_heads, feedforward_channels,
segmentation/backbones/vit_SelfPatch.py:44
Method__init__
(self, widen_factor=1., strides=(1, 2, 2, 2, 1, 2, 1), dila
segmentation/backbones/mobilenet_v2.py:53
Method__init__
(self, inplanes, planes, stride=1, dilatio
segmentation/backbones/resnet.py:19
Method__init__
(self, inplanes, planes, stride=1, dilatio
segmentation/backbones/resnet.py:108
Method__init__
(self, **kwargs)
segmentation/backbones/resnet.py:698
Method__init__
(self, **kwargs)
segmentation/backbones/resnet.py:712
Method__str__
(self)
utils.py:257
Method__str__
(self)
utils.py:313
Method_init_weights
(self, m)
selfpatch_vision_transformer.py:291
Method_init_weights
(self, m)
selfpatch_vision_transformer.py:383
Method_init_weights
(self, m)
selfpatch_vision_transformer.py:469
Method_init_weights
(m)
detection/backbone/vit_SelfPatch.py:232
Method_inner_forward
(x)
segmentation/backbones/resnest.py:229
Method_inner_forward
(x)
segmentation/backbones/cgnet.py:40
Method_inner_forward
(x)
segmentation/backbones/cgnet.py:145
Method_inner_forward
(x)
segmentation/backbones/swin.py:356
Method_inner_forward
(x)
segmentation/backbones/resnet.py:72
Method_inner_forward
(x)
segmentation/backbones/resnet.py:270
Functionaccuracy
Computes the accuracy over the k top predictions for the specified values of k
utils.py:482
Methodadd_meter
(self, name, meter)
utils.py:325
Methodaug_test
(self, imgs, img_metas, **kwargs)
segmentation/tools/deploy_test.py:97
Methodaug_test
(self, imgs, img_metas, **kwargs)
segmentation/tools/deploy_test.py:145
Methodavg
(self)
utils.py:241
Functionbool_flag
Parse boolean arguments from the command line.
utils.py:181
Functioncancel_gradients_last_layer
(epoch, model, freeze_last_layer)
utils.py:124
Functionclip_gradients
(model, clip)
utils.py:112
Functioncompute_map
Computes the mAP for a given set of returned results. Usage: map = compute_map (ranks, gnd) computes mean av
utils.py:723
Functionconcat_all_gather
Performs all_gather operation on the provided tensors. *** Warning ***: torch.distributed.all_gather has no gradient.
main_selfpatch.py:385
Functionconvert_json_to_label
(json_file)
segmentation/tools/convert_datasets/cityscapes.py:9
Functionconvert_mat
(mat_file, in_dir, out_dir)
segmentation/tools/convert_datasets/voc_aug.py:14
Functionconvert_to_trainID
(tuple_path, in_img_dir, in_ann_dir, out_img_dir, out_mask_dir, is_train)
segmentation/tools/convert_datasets/coco_stuff10k.py:189
Functionconvert_to_trainID
(maskpath, out_mask_dir, is_train)
segmentation/tools/convert_datasets/coco_stuff164k.py:189
Functioncosine_scheduler
(base_value, final_value, epochs, niter_per_ep, warmup_epochs=0, start_warmup_value=0)
utils.py:167
Methodencode_decode
(self, img, img_metas)
segmentation/tools/deploy_test.py:63
Methodencode_decode
(self, img, img_metas)
segmentation/tools/deploy_test.py:122
Methodextract_feat
(self, imgs)
segmentation/tools/deploy_test.py:60
Methodextract_feat
(self, imgs)
segmentation/tools/deploy_test.py:119
Functionfix_random_seeds
Fix random seeds.
utils.py:195
Methodforward
(self, x, head_only=False, loc=True)
main_selfpatch.py:146
Methodforward
Cross-entropy between softmax outputs of the teacher and student networks.
main_selfpatch.py:419
Methodforward
(self, x)
utils.py:590
Methodforward
(self, x)
selfpatch_vision_transformer.py:48
Methodforward
(self, x)
selfpatch_vision_transformer.py:62
Methodforward
(self, x, attention=False, mask=None)
selfpatch_vision_transformer.py:89
Methodforward
(self, x, x_cls, attention=False, mask=None)
selfpatch_vision_transformer.py:130
Methodforward
(self, x)
selfpatch_vision_transformer.py:157
Methodforward
(self, x_cls, x, return_attention=False)
selfpatch_vision_transformer.py:187
Methodforward
(self, x)
selfpatch_vision_transformer.py:208
Methodforward
(self, x, return_attention=False)
selfpatch_vision_transformer.py:235
Methodforward
(self, x)
selfpatch_vision_transformer.py:256
Methodforward
(self, x, loc=False)
selfpatch_vision_transformer.py:327
Methodforward
(self, x, loc=False)
selfpatch_vision_transformer.py:389
Methodforward
(self, x)
selfpatch_vision_transformer.py:475
Methodforward
(self, x)
detection/backbone/vit_SelfPatch.py:45
Methodforward
(self, x)
detection/backbone/vit_SelfPatch.py:59
Methodforward
(self, x)
detection/backbone/vit_SelfPatch.py:80
Methodforward
(self, x, return_attention=False)
detection/backbone/vit_SelfPatch.py:107
Methodforward
(self, x)
detection/backbone/vit_SelfPatch.py:128
Methodforward
(self, x)
detection/backbone/vit_SelfPatch.py:303
Methodforward
(self, x)
segmentation/backbones/resnest.py:29
← previousnext →201–300 of 386, ranked by callers