Functions — github.com/ChongQingNoSubway/SelfReg-UNet

hub / github.com/ChongQingNoSubway/SelfReg-UNet / functions

Functions3,840 in github.com/ChongQingNoSubway/SelfReg-UNet

⨍ Functions 3,840◇ Types & classes 731

↓ 1 callersMethod__init__

( self, cfg: List[Any], num_classes: int = 1000, in_chans: int

src/train_acdc/lib/models_timm/vgg.py:86

↓ 1 callersMethod__init__

( self, block_args, num_classes=1000, num_features=1280, in_chans=3, stem_size=32, fix_stem=False,

src/train_acdc/lib/models_timm/efficientnet.py:473

↓ 1 callersMethod__init__

(self, drop_prob: float = 0., scale_by_keep: bool = True)

src/train_acdc/lib/models_timm/layers/drop.py:160

↓ 1 callersMethod__init__

(self, feat_size, dim_head, scale)

src/train_acdc/lib/models_timm/layers/bottleneck_attn.py:61

↓ 1 callersMethod__init__

( self, in_features: int, feat_size: Union[int, Tuple[int, int]],

src/train_acdc/lib/models_timm/layers/attention_pool2d.py:88

↓ 1 callersMethod__init__

(self, kernel_size: int, stride=None, padding=0, ceil_mode=False, count_include_pad=True)

src/train_acdc/lib/models_timm/layers/pool2d_same.py:24

↓ 1 callersMethod__init__

(self, in_channels, out_channels, kernel_size=3, stride=1, dilation=1, padding='', bias=False,

src/train_acdc/lib/models_timm/layers/separable_conv.py:54

↓ 1 callersMethod__init__

(self, max_res: int = 224, num_bands: int = 64, concat_grid=True, keep_spatial=False)

src/train_acdc/lib/models_timm/layers/pos_embed.py:114

↓ 1 callersMethod__init__

( self, channels=None, kernel_size=3, gamma=2, beta=1, act_layer=None, gate_layer='sigmoid',

src/train_acdc/lib/models_timm/layers/eca.py:60

↓ 1 callersMethod__init__

( self, in_channels, out_channels, kernel_size=1, stride=1, padding='', dilation=1, groups=1,

src/train_acdc/lib/models_timm/layers/conv_bn_act.py:13

↓ 1 callersMethod__init__

(self, in_channels, out_channels=None, kernel_size=3, stride=1, padding=None, dilation=1, gro

src/train_acdc/lib/models_timm/layers/split_attn.py:36

↓ 1 callersMethod__init__

( self, dim, dim_out=None, feat_size=None, stride=1, num_heads=8, dim_head=None, block_size=8, hal

src/train_acdc/lib/models_timm/layers/halo_attn.py:125

↓ 1 callersMethod__init__

(self, num_features, apply_act=True, eps=1e-5, rms=True, **_)

src/train_acdc/lib/models_timm/layers/filter_response_norm.py:20

↓ 1 callersMethod__init__

Selective Kernel Convolution Module As described in Selective Kernel Networks (https://arxiv.org/abs/1903.06586) with some modifications.

src/train_acdc/lib/models_timm/layers/selective_kernel.py:51

↓ 1 callersMethod__init__

(self, k=9, dilation=1, stochastic=False, epsilon=0.0)

src/train_acdc/lib/gcn_lib/torch_edge.py:116

↓ 1 callersMethod__init__

(self, channels, act='relu', norm=None, bias=True)

src/train_acdc/lib/gcn_lib/torch_nn.py:43

↓ 1 callersMethod__init__

(self, block_units, width_factor)

src/train_acdc/networks_trans/vit_seg_modeling_resnet_skip.py:115

↓ 1 callersMethod__init__

(self, n_channels, n_classes, bilinear=False)

src/train_acdc/unet/unet_model.py:51

↓ 1 callersMethod__init__

(self, in_chans=384, out_chans=[128], upsample=True)

src/train_acdc/modelsHiFormer/Decoder.py:4

↓ 1 callersMethod__init__

(self, num=40)

src/train_synase/utils/utils.py:56

↓ 1 callersMethod__init__

(self, block_units, width_factor)

src/train_synase/networks_trans/vit_seg_modeling_resnet_skip.py:115

↓ 1 callersMethod__init__

(self, n_channels, n_classes, bilinear=False)

src/train_synase/unet/unet_model.py:51

↓ 1 callersMethod__init__

(self, in_chans=384, out_chans=[128], upsample=True)

src/train_synase/modelsHiFormer/Decoder.py:4

↓ 1 callersMethod_attn

(self, x)

src/train_acdc/lib/models_timm/swin_transformer_v2.py:322

↓ 1 callersFunction_block_cfg

(width_mult=1.0, depth_mult=1.0, initial_chs=16, final_chs=180, se_ratio=0., ch_div=1)

src/train_acdc/lib/models_timm/rexnet.py:101

↓ 1 callersFunction_build_blocks

( block_cfg, prev_chs, width_mult, ch_div=1, act_layer='swish', dw_act_layer='relu6', drop_path_rate=0

src/train_acdc/lib/models_timm/rexnet.py:121

↓ 1 callersMethod_calc_window_shift

(self, target_window_size, target_shift_size)

src/train_acdc/lib/models_timm/swin_transformer_v2.py:315

↓ 1 callersMethod_check_branches

(self, num_branches, blocks, num_blocks, num_in_chs, num_channels)

src/train_acdc/lib/models_timm/hrnet.py:406

↓ 1 callersFunction_compute_num_patches

(img_size, patches)

src/train_acdc/lib/models_timm/crossvit.py:260

↓ 1 callersFunction_convert_openai_clip

(state_dict, model)

src/train_acdc/lib/models_timm/vision_transformer.py:713

↓ 1 callersFunction_create_gluon_xception

(variant, pretrained=False, **kwargs)

src/train_acdc/lib/models_timm/gluon_xception.py:256

↓ 1 callersFunction_create_inception_v4

(variant, pretrained=False, **kwargs)

src/train_acdc/lib/models_timm/inception_v4.py:321

↓ 1 callersFunction_create_nasnet

(variant, pretrained=False, **kwargs)

src/train_acdc/lib/models_timm/nasnet.py:576

↓ 1 callersFunction_create_pnasnet

(variant, pretrained=False, **kwargs)

src/train_acdc/lib/models_timm/pnasnet.py:347

↓ 1 callersFunction_decode_block_str

Decode block definition string Gets a list of block arg (dicts) through a string notation of arguments. E.g. ir_r2_k3_s2_e1_i32_o16_se0.25_n

src/train_acdc/lib/models_timm/efficientnet_builder.py:76

↓ 1 callersMethod_dice_loss

(self, score, target)

src/train_acdc/utils/utils.py:122

↓ 1 callersMethod_dice_loss

(self, score, target)

src/train_acdc/modelsHiFormer/utils.py:564

↓ 1 callersMethod_dice_loss

(self, score, target)

src/train_synase/utils/utils.py:122

↓ 1 callersMethod_dice_loss

(self, score, target)

src/train_synase/modelsHiFormer/utils.py:564

↓ 1 callersFunction_fast_hist

(label_pred, label_true, num_classes)

src/train_acdc/utils/misc.py:60

↓ 1 callersFunction_fast_hist

(label_pred, label_true, num_classes)

src/train_synase/utils/misc.py:60

↓ 1 callersMethod_forward

(self, x)

src/train_acdc/lib/models_timm/inception_v3.py:69

↓ 1 callersMethod_forward

(self, x)

src/train_acdc/lib/models_timm/inception_v3.py:102

↓ 1 callersMethod_forward

(self, x)

src/train_acdc/lib/models_timm/inception_v3.py:140

↓ 1 callersMethod_forward

(self, x)

src/train_acdc/lib/models_timm/inception_v3.py:178

↓ 1 callersMethod_forward

(self, x)

src/train_acdc/lib/models_timm/inception_v3.py:215

↓ 1 callersMethod_forward

(self, x)

src/train_acdc/lib/models_timm/vision_transformer.py:353

↓ 1 callersMethod_forward_batch

This function performs standard (non-sequential) scaled cosine self-attention.

src/train_acdc/lib/models_timm/swin_transformer_v2_cr.py:242

↓ 1 callersMethod_forward_cross_attn

(self, x: torch.Tensor, x_prev: Optional[torch.Tensor] = None)

src/train_acdc/lib/models_timm/mobilevit.py:399

↓ 1 callersMethod_forward_jit

(self, x)

src/train_acdc/lib/models_timm/vision_transformer.py:347

↓ 1 callersMethod_forward_self_attn

(self, x: torch.Tensor)

src/train_acdc/lib/models_timm/mobilevit.py:374

↓ 1 callersMethod_forward_sequential

src/train_acdc/lib/models_timm/swin_transformer_v2_cr.py:232

↓ 1 callersFunction_gen_fbnetc

FBNet-C Paper: https://arxiv.org/abs/1812.03443 Ref Impl: https://github.com/facebookresearch/maskrcnn-benchmark/blob/master/maskrcn

src/train_acdc/lib/models_timm/efficientnet.py:770

↓ 1 callersFunction_gen_mnasnet_small

Creates a mnasnet-b1 model. Ref impl: https://github.com/tensorflow/tpu/tree/master/models/official/mnasnet Paper: https://arxiv.org/pdf/1807

src/train_acdc/lib/models_timm/efficientnet.py:711

↓ 1 callersFunction_gen_mobilenet_v3_rw

Creates a MobileNet-V3 model. Ref impl: ? Paper: https://arxiv.org/abs/1905.02244 Args: channel_multiplier: multiplier to number o

src/train_acdc/lib/models_timm/mobilenetv3.py:297

↓ 1 callersFunction_gen_spnasnet

Creates the Single-Path NAS model from search targeted for Pixel1 phone. Paper: https://arxiv.org/abs/1904.02877 Args: channel_multipl

src/train_acdc/lib/models_timm/efficientnet.py:800

↓ 1 callersFunction_get_attn_fn

(stage_args)

src/train_acdc/lib/models_timm/cspnet.py:791

↓ 1 callersFunction_get_block_fn

(stage_args)

src/train_acdc/lib/models_timm/cspnet.py:780

↓ 1 callersFunction_get_grouping

(name)

src/train_acdc/lib/models_timm/helpers.py:662

↓ 1 callersMethod_get_rel_pos_bias

(self)

src/train_acdc/lib/models_timm/beit.py:174

↓ 1 callersMethod_get_rel_pos_bias

(self)

src/train_acdc/lib/models_timm/swin_transformer.py:186

↓ 1 callersMethod_get_relative_pos

(self, relative_pos, H, W)

src/train_acdc/lib/gcn_lib/torch_vertex.py:165

↓ 1 callersFunction_get_return_layers

(feature_info, out_map)

src/train_acdc/lib/models_timm/features.py:145

↓ 1 callersMethod_get_stage_args

(self, cfg: RegNetCfg, default_stride=2, output_stride=32, drop_path_rate=0.)

src/train_acdc/lib/models_timm/regnet.py:424

↓ 1 callersFunction_get_stage_fn

(stage_args)

src/train_acdc/lib/models_timm/cspnet.py:765

↓ 1 callersMethod_init_respostnorm

(self)

src/train_acdc/lib/models_timm/swin_transformer_v2.py:460

↓ 1 callersMethod_init_weights

(self)

src/train_acdc/networks_trans/vit_seg_modeling.py:107

↓ 1 callersMethod_init_weights

(self)

src/train_synase/networks_trans/vit_seg_modeling.py:107

↓ 1 callersMethod_initialize_weights

(self)

src/train_acdc/lib/models_timm/vgg.py:167

↓ 1 callersFunction_is_contiguous

(tensor: torch.Tensor)

src/train_acdc/lib/models_timm/layers/norm.py:75

↓ 1 callersFunction_kernel_valid

(k)

src/train_acdc/lib/models_timm/layers/selective_kernel.py:15

↓ 1 callersFunction_layer_norm_cf

(x: torch.Tensor, weight: torch.Tensor, bias: torch.Tensor, eps: float)

src/train_acdc/lib/models_timm/layers/norm.py:84

↓ 1 callersFunction_load_weights

(model: nn.Module, checkpoint_path: str, prefix: str = 'resnet/')

src/train_acdc/lib/models_timm/resnetv2.py:468

↓ 1 callersFunction_load_weights

Load weights from .npz checkpoints for official Google Brain Flax implementation

src/train_acdc/lib/models_timm/vision_transformer.py:607

↓ 1 callersMethod_make_block

(self, ba, block_idx, block_count)

src/train_acdc/lib/models_timm/efficientnet_builder.py:312

↓ 1 callersMethod_make_branches

(self, num_branches, block, num_blocks, num_channels)

src/train_acdc/lib/models_timm/hrnet.py:435

↓ 1 callersMethod_make_fuse_layers

(self)

src/train_acdc/lib/models_timm/hrnet.py:442

↓ 1 callersMethod_make_one_branch

(self, branch_index, block, num_blocks, num_channels, stride=1)

src/train_acdc/lib/models_timm/hrnet.py:418

↓ 1 callersFunction_num_groups

(num_channels, num_groups, group_size)

src/train_acdc/lib/models_timm/layers/norm_act.py:172

↓ 1 callersMethod_one_hot_encoder

(self, input_tensor)

src/train_acdc/utils/utils.py:114

↓ 1 callersMethod_one_hot_encoder

(self, input_tensor)

src/train_acdc/modelsHiFormer/utils.py:556

↓ 1 callersMethod_one_hot_encoder

(self, input_tensor)

src/train_synase/utils/utils.py:114

↓ 1 callersMethod_one_hot_encoder

(self, input_tensor)

src/train_synase/modelsHiFormer/utils.py:556

↓ 1 callersMethod_padding

(self, x)

src/train_acdc/lib/models_timm/layers/median_pool.py:25

↓ 1 callersMethod_partition_attn

(self, x)

src/train_acdc/lib/maxxvit_4out.py:1212

↓ 1 callersMethod_partition_attn

(self, x)

src/train_acdc/lib/maxxvit_4out.py:1285

↓ 1 callersMethod_partition_attn

(self, x)

src/train_acdc/lib/maxxvit_4out.py:1383

↓ 1 callersMethod_partition_attn

(self, x)

src/train_acdc/lib/models_timm/maxxvit.py:1212

↓ 1 callersMethod_partition_attn

(self, x)

src/train_acdc/lib/models_timm/maxxvit.py:1285

↓ 1 callersMethod_partition_attn

(self, x)

src/train_acdc/lib/models_timm/maxxvit.py:1383

↓ 1 callersMethod_pos_embed

(self, x)

src/train_acdc/lib/models_timm/vision_transformer.py:516

↓ 1 callersMethod_relative_positional_encodings

Method computes the relative positional encodings Returns: relative_position_bias (torch.Tensor): Relative positional encodings

src/train_acdc/lib/models_timm/swin_transformer_v2_cr.py:217

↓ 1 callersMethod_shifted_window_attn

(self, x)

src/train_acdc/lib/models_timm/swin_transformer_v2_cr.py:411

↓ 1 callersMethod_shortcut_pool

(self, x, feat_size: List[int])

src/train_acdc/lib/models_timm/mvitv2.py:647

↓ 1 callersFunction_update_config_from_file

(config, cfg_file)

src/train_acdc/config.py:175

↓ 1 callersFunction_update_config_from_file

(config, cfg_file)

src/train_synase/config.py:175

↓ 1 callersFunction_weight_init

(m)

src/train_acdc/lib/models_timm/senet.py:63

↓ 1 callersMethod_window_attn

(self, x, q_global: Optional[torch.Tensor] = None)

src/train_acdc/lib/models_timm/gcvit.py:303

↓ 1 callersFunction_xception

(variant, pretrained=False, **kwargs)

src/train_acdc/lib/models_timm/xception.py:240

↓ 1 callersFunctionact_with_gamma

(act_type, gamma: float = 1.)

src/train_acdc/lib/models_timm/nfnet.py:271

← previous next →501–600 of 3,840, ranked by callers