MCPcopy Create free account

hub / github.com/alinlab/SelfPatch / functions

Functions386 in github.com/alinlab/SelfPatch

⨍ Functions 386◇ Types & classes 81
↓ 80 callersFunctionprint
(*args, **kwargs)
utils.py:439
↓ 11 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
selfpatch_vision_transformer.py:53
↓ 10 callersMethodupdate
(self, **kwargs)
utils.py:298
↓ 7 callersMethodnorm1
nn.Module: the normalization layer named "norm1"
segmentation/backbones/hrnet.py:425
↓ 7 callersFunctiontrunc_normal_
(tensor, mean=0., std=1., a=-2., b=2.)
utils.py:528
↓ 6 callersMethod__init__
(self, in_channels, out_channels, exp_ratio=6,
segmentation/backbones/bisenetv2.py:201
↓ 6 callersMethodforward_plugin
Forward function for plugins.
segmentation/backbones/resnet.py:245
↓ 6 callersMethodmax
(self)
utils.py:250
↓ 6 callersMethodnorm2
nn.Module: the normalization layer named "norm2"
segmentation/backbones/hrnet.py:430
↓ 5 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
detection/backbone/vit_SelfPatch.py:50
↓ 5 callersMethodapply
(self, x)
utils.py:669
↓ 4 callersMethod__init__
(self, backbone_cfg, in_channels=3, spatial_channels=(64, 6
segmentation/backbones/bisenetv1.py:294
↓ 4 callersMethod__init__
(self, embed_dims, num_heads, window_size,
segmentation/backbones/swin.py:39
↓ 4 callersMethod__init__
(self, depth, in_channels=3, stem_channels=64,
segmentation/backbones/resnet.py:396
↓ 4 callersFunction_retrieve_data_cfg
(_data_cfg, skip_type, show_origin)
segmentation/tools/browse_dataset.py:113
↓ 4 callersFunctionshow_result_pyplot
(img: Union[str, np.ndarray], result: np.ndarray, palette: Optio
segmentation/tools/onnx2tensorrt.py:76
↓ 3 callersMethod__init__
(self, groups=1, base_width=4, radix=2, re
segmentation/backbones/resnest.py:295
↓ 3 callersMethod__init__
(self, embed_dims, feedforward_channels, act_cfg=dict(type=
segmentation/backbones/mit.py:40
↓ 3 callersMethod__init__
(self, in_channels=3, base_channels=64, num_stages=5,
segmentation/backbones/unet.py:283
↓ 3 callersMethod__init__
(self, in_channels=3, num_channels=(32, 64, 128), num_block
segmentation/backbones/cgnet.py:219
↓ 3 callersMethod__init__
(self, in_channels=3, downsample_dw_channels=(32, 48), glob
segmentation/backbones/fast_scnn.py:326
↓ 3 callersMethod_make_layer
(self, in_channels, out_channels, blocks,
segmentation/backbones/fast_scnn.py:160
↓ 3 callersMethod_make_stage
Make each stage.
segmentation/backbones/hrnet.py:529
↓ 3 callersMethod_make_transition_layer
Make transition layer.
segmentation/backbones/hrnet.py:434
↓ 3 callersFunction_run
(command)
utils.py:386
↓ 3 callersFunctionis_dist_avail_and_initialized
()
utils.py:403
↓ 3 callersMethodmake_block_plugins
make plugins for block. Args: in_channels (int): Input channels of plugin. plugins (list[dict]): List of plugins cfg
segmentation/backbones/resnet.py:222
↓ 3 callersFunctionun_gz
(src, dst)
segmentation/tools/convert_datasets/stare.py:15
↓ 2 callersMethod__init__
(self, dim=256, whit=0.5)
utils.py:638
↓ 2 callersMethod_freeze_stages
Freeze stages param and norm stats.
segmentation/backbones/hrnet.py:570
↓ 2 callersMethod_freeze_stages
Freeze stages param and norm stats.
segmentation/backbones/resnet.py:640
↓ 2 callersFunction_update_input_img
(img_list, img_meta_list, update_ori_shape=False)
segmentation/tools/pytorch2onnx.py:101
↓ 2 callersFunctiondigit_version
(version_str)
segmentation/tools/pytorch2torchscript.py:17
↓ 2 callersFunctionextract_feature
Extract one frame feature everytime.
eval_video_segmentation.py:153
↓ 2 callersFunctiongenerate_aug_list
(merged_list, excluded_list)
segmentation/tools/convert_datasets/voc_aug.py:21
↓ 2 callersMethodget_local_index
(N_patches, k_size)
selfpatch_vision_transformer.py:426
↓ 2 callersFunctionimwrite_indexed
Save indexed png for DAVIS.
eval_video_segmentation.py:166
↓ 2 callersMethodnorm1
(self)
segmentation/backbones/vit_SelfPatch.py:84
↓ 2 callersMethodnorm1
nn.Module: normalization layer after the first convolution layer
segmentation/backbones/resnet.py:253
↓ 2 callersMethodnorm2
(self)
segmentation/backbones/vit_SelfPatch.py:88
↓ 2 callersMethodnorm3
nn.Module: normalization layer after the third convolution layer
segmentation/backbones/resnet.py:263
↓ 2 callersFunctionnorm_cdf
(x)
utils.py:495
↓ 2 callersMethodprepare_tokens
(self, x)
selfpatch_vision_transformer.py:320
↓ 2 callersFunctionread_frame
read a single frame & preprocess
eval_video_segmentation.py:197
↓ 2 callersMethodresize_pos_embed
Resize pos_embed weights. Resize pos_embed using bicubic interpolate method. Args: pos_embed (torch.Tensor): Position emb
segmentation/backbones/vit_SelfPatch.py:341
↓ 2 callersMethodstep
(self)
utils.py:545
↓ 2 callersMethodupdate
(self, value, n=1)
utils.py:217
↓ 2 callersMethodwindow_partition
Args: x: (B, H, W, C) Returns: windows: (num_windows*B, window_size, window_size, C)
segmentation/backbones/swin.py:269
↓ 1 callersMethod__init__
(self, out_dim, out_dim_selfpatch, ncrops, warmup_teacher_temp, teacher_temp, warmup_teacher_
main_selfpatch.py:401
↓ 1 callersMethod__init__
(self, trt_file: str, cfg: Any, device_id: int)
segmentation/tools/deploy_test.py:103
↓ 1 callersMethod__init__
(self, groups=1, base_width=4, **kwargs)
segmentation/backbones/resnext.py:139
↓ 1 callersMethod__init__
(self, extra, in_channels=3, conv_cfg=None,
segmentation/backbones/hrnet.py:299
↓ 1 callersMethod__init__
(self, img_size=224, patch_size=16, in_channels=3,
segmentation/backbones/vit_SelfPatch.py:148
↓ 1 callersMethod_check_branches
Check branches configuration.
segmentation/backbones/hrnet.py:50
↓ 1 callersMethod_check_input_divisible
(self, x)
segmentation/backbones/unet.py:427
↓ 1 callersFunction_class_to_index
(mask, _mapping, _key)
segmentation/tools/convert_datasets/pascal_context.py:23
↓ 1 callersFunction_convert_batchnorm
(module)
segmentation/tools/pytorch2onnx.py:25
↓ 1 callersFunction_convert_batchnorm
(module)
segmentation/tools/pytorch2torchscript.py:38
↓ 1 callersFunction_demo_mm_inputs
Create a superset of inputs needed to run test or train batches. Args: input_shape (tuple): input batch dimensions nu
segmentation/tools/pytorch2onnx.py:46
↓ 1 callersFunction_demo_mm_inputs
Create a superset of inputs needed to run test or train batches. Args: input_shape (tuple): input batch dimensions nu
segmentation/tools/pytorch2torchscript.py:59
↓ 1 callersMethod_freeze_stages
(self)
segmentation/backbones/mobilenet_v3.py:254
↓ 1 callersMethod_freeze_stages
(self)
segmentation/backbones/swin.py:639
↓ 1 callersMethod_freeze_stages
(self)
segmentation/backbones/mobilenet_v2.py:181
↓ 1 callersMethod_make_branches
Build multiple branch.
segmentation/backbones/hrnet.py:115
↓ 1 callersMethod_make_fuse_layers
Build fuse layer.
segmentation/backbones/hrnet.py:125
↓ 1 callersMethod_make_layer
(self)
segmentation/backbones/mobilenet_v3.py:129
↓ 1 callersMethod_make_layer
Make each layer.
segmentation/backbones/hrnet.py:481
↓ 1 callersMethod_make_one_branch
Build one branch.
segmentation/backbones/hrnet.py:68
↓ 1 callersMethod_make_stem_layer
Make stem layer for ResNet.
segmentation/backbones/resnet.py:591
↓ 1 callersFunction_no_grad_trunc_normal_
(tensor, mean, std, a, b)
utils.py:492
↓ 1 callersMethod_pos_embeding
Positiong embeding method. Resize the pos_embed, if the input image size doesn't match the training size. Args:
segmentation/backbones/vit_SelfPatch.py:309
↓ 1 callersFunction_prepare_input_img
(img_path: str, test_pipeline: Iterable[dict], shape: Optional[I
segmentation/tools/onnx2tensorrt.py:26
↓ 1 callersFunction_prepare_input_img
(img_path, test_pipeline, shape=None, res
segmentation/tools/pytorch2onnx.py:76
↓ 1 callersFunction_update_input_img
(img_list: Iterable, img_meta_list: Iterable)
segmentation/tools/onnx2tensorrt.py:51
↓ 1 callersFunctioncheck_torch_version
()
segmentation/tools/pytorch2torchscript.py:29
↓ 1 callersFunctioncolor_normalize
(x, mean=[0.485, 0.456, 0.406], std=[0.228, 0.224, 0.225])
eval_video_segmentation.py:244
↓ 1 callersFunctioncompute_ap
Computes average precision for given ranked indexes. Arguments --------- ranks : zerro-based ranks of positive images nres : num
utils.py:688
↓ 1 callersFunctionconvert_mit
(ckpt)
segmentation/tools/model_converters/mit2mmseg.py:11
↓ 1 callersFunctionconvert_swin
(ckpt)
segmentation/tools/model_converters/swin2mmseg.py:11
↓ 1 callersFunctionconvert_vit
(ckpt)
segmentation/tools/model_converters/vit2mmseg.py:11
↓ 1 callersFunctioncorrect_unfold_norm_order
(x)
segmentation/tools/model_converters/swin2mmseg.py:21
↓ 1 callersFunctioncorrect_unfold_reduction_order
(x)
segmentation/tools/model_converters/swin2mmseg.py:14
↓ 1 callersMethoddouble_step_seq
(step1, len1, step2, len2)
segmentation/backbones/swin.py:119
↓ 1 callersFunctiondrop_path
(x, drop_prob: float = 0., training: bool = False)
selfpatch_vision_transformer.py:30
↓ 1 callersFunctiondrop_path
(x, drop_prob: float = 0., training: bool = False)
detection/backbone/vit_SelfPatch.py:27
↓ 1 callersFunctioneval_video_tracking_davis
Evaluate tracking on a video given first frame & segmentation
eval_video_segmentation.py:38
↓ 1 callersMethodforward_features
(self, x)
detection/backbone/vit_SelfPatch.py:283
↓ 1 callersFunctiongenerate_coco_list
(folder)
segmentation/tools/convert_datasets/coco_stuff10k.py:209
↓ 1 callersFunctionget_GiB
return x GiB.
segmentation/tools/onnx2tensorrt.py:21
↓ 1 callersFunctionget_args_parser
()
main_selfpatch.py:44
↓ 1 callersMethodget_intermediate_layers
(self, x, n=1)
selfpatch_vision_transformer.py:333
↓ 1 callersFunctionget_rank
()
utils.py:417
↓ 1 callersFunctionget_world_size
()
utils.py:411
↓ 1 callersFunctionimshow_semantic
Draw `result` over `img`. Args: img (str or Tensor): The image to be displayed. seg (Tensor): The semantic segmentation results t
segmentation/tools/browse_dataset.py:49
↓ 1 callersMethodinit_weights
(self)
segmentation/backbones/swin.py:75
↓ 1 callersMethodinterpolate_pos_encoding
(self, x, w, h)
selfpatch_vision_transformer.py:300
↓ 1 callersMethodinterpolate_pos_encoding
(self, x, w, h)
detection/backbone/vit_SelfPatch.py:251
↓ 1 callersFunctionis_main_process
()
utils.py:423
↓ 1 callersFunctionlabel_propagation
propagate segs of frames in list_frames to frame_tar
eval_video_segmentation.py:113
↓ 1 callersFunctionload_json_logs
(json_logs)
segmentation/tools/analyze_logs.py:100
next →1–100 of 386, ranked by callers