MCPcopy Create free account

hub / github.com/ChunmingHe/WS-SAM / functions

Functions357 in github.com/ChunmingHe/WS-SAM

⨍ Functions 357◇ Types & classes 99
↓ 1 callersFunctionto_3d
(x)
lib/VIT/decoder/decoder_p.py:38
↓ 1 callersFunctionto_4d
(x,h,w)
lib/VIT/decoder/decoder_p.py:41
↓ 1 callersFunctiontrain
train function
Train.py:68
↓ 1 callersFunctionval
validation function
Train.py:187
↓ 1 callersFunctionwindow_reverse
Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image
lib/VIT/encoder/swin_encoder.py:51
FunctionCalParams
Usage: Calculate Params and FLOPs via [THOP](https://github.com/Lyken17/pytorch-OpCounter) Necessarity: from thop import prof
utils/utils.py:84
FunctionPYBIND11_MODULE
losses/kernels/lib_tree_filter/src/tree_filter.cpp:7
MethodThreshold_process
(self, a)
utils/data_val.py:317
MethodThreshold_process
(self, a)
utils/data_val.py:419
MethodThreshold_process
(self, a)
utils/data_val.py:534
MethodThreshold_process
(self, a)
utils/data_val.py:651
MethodThreshold_process
(self, a)
utils/data_val.py:824
Method__getitem__
(self, index)
utils/dataloader.py:28
Method__getitem__
(self, index)
utils/data_val.py:159
Method__getitem__
(self, index)
utils/data_val.py:266
Method__getitem__
(self, index)
utils/data_val.py:370
Method__getitem__
(self, index)
utils/data_val.py:472
Method__getitem__
(self, index)
utils/data_val.py:589
Method__getitem__
(self, index)
utils/data_val.py:788
Method__init__
(self, num=40)
utils/utils.py:62
Method__init__
(self, image_root, gt_root, trainsize)
utils/dataloader.py:11
Method__init__
(self, image_root, gt_root, testsize)
utils/dataloader.py:85
Method__init__
(self, image_root, gt_root, edge_root, trainsize)
utils/data_val.py:118
Method__init__
(self, image_root, gt_root, trainsize=384)
utils/data_val.py:225
Method__init__
(self, image_root, gt_root, imgName_root,trainsize=384)
utils/data_val.py:326
Method__init__
(self, image_root, gt_root, scribble_root, trainsize=384)
utils/data_val.py:428
Method__init__
(self, image_root, gt_root, scribble_root, imgName_root, trainsize=384)
utils/data_val.py:542
Method__init__
(self, image_root, gt_root, testsize)
utils/data_val.py:713
Method__init__
(self, image_root, scribble_root, trainsize=384)
utils/data_val.py:769
Method__init__
(self, channels=128, imagenet_pretrained=True)
lib/Network.py:43
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True,batch_norm
lib/GatedConv.py:11
Method__init__
Builds the soft position embedding layer. Args: hidden_size: Size of input feature dimension. resolution: Tuple of integers specifyin
lib/Slot.py:23
Method__init__
Builds the Slot Attention-based Auto-encoder. Args: resolution: Tuple of integers specifying width and height of input image
lib/Slot.py:158
Method__init__
(self, in_channels, out_channels, kernel_size, stride)
lib/Modules.py:22
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, need_relu=True,
lib/Modules.py:35
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, out_padding=0, need_relu=True,
lib/Modules.py:53
Method__init__
(self, dilation_series=[6, 12, 18], padding_series=[6, 12, 18], depth=128)
lib/Modules.py:105
Method__init__
(self, in_channels, out_channels)
lib/Modules.py:161
Method__init__
(self, channel, ratio=16)
lib/Modules.py:204
Method__init__
(self)
lib/Modules.py:223
Method__init__
(self, channel)
lib/Modules.py:237
Method__init__
(self, in_channels)
lib/Modules.py:249
Method__init__
(self, in_channels, out_channels)
lib/Modules.py:266
Method__init__
Builds the Slot Attention module. Args: iters: Number of iterations. num_slots: Number of slots. encoder_d
lib/Modules.py:294
Method__init__
(self, encoder_dims, resolution, num_slots, iters)
lib/Modules.py:380
Method__init__
(self, encoder_dims, resolution, num_slots, iters)
lib/Modules.py:411
Method__init__
(self, in_channels)
lib/Modules.py:476
Method__init__
(self, in_channels,num_slots_N,num_slots_M,iters,resolutions)
lib/Modules.py:493
Method__init__
(self, in_channels,num_slots_N,num_slots_M,iters,resolutions)
lib/Modules.py:507
Method__init__
(self, in_channels, out_channels)
lib/Modules.py:522
Method__init__
(self, in_channels, out_channels)
lib/Modules.py:549
Method__init__
(self, in_channels, out_channels)
lib/Modules.py:579
Method__init__
(self, in_channels, out_channels)
lib/Modules.py:603
Method__init__
(self, in_channels, out_channels)
lib/Modules.py:637
Method__init__
(self, channel, reduction=16)
lib/Modules.py:667
Method__init__
(self, n_feat, kernel_size=3, reduction=16, bias=True, bn=False, act=nn.ReLU(True), res_scale=1)
lib/Modules.py:689
Method__init__
(self, in_channels)
lib/Modules.py:711
Method__init__
(self, hidden_channels, out_channels)
lib/Modules.py:736
Method__init__
(self, in_channels)
lib/Modules.py:758
Method__init__
(self, in_channels)
lib/Modules.py:794
Method__init__
(self, in_channels, out_channels)
lib/Modules.py:826
Method__init__
(self, in_channels)
lib/Modules.py:846
Method__init__
(self, in_channels)
lib/Modules.py:867
Method__init__
(self, in_channels)
lib/Modules.py:898
Method__init__
(self, in_channels)
lib/Modules.py:926
Method__init__
(self, in_channels)
lib/Modules.py:953
Method__init__
(self, in_channels)
lib/Modules.py:975
Method__init__
(self, in_channels)
lib/Modules.py:1009
Method__init__
Constructor Args: inplanes: input channel dimensionality planes: output channel dimensionality stride: co
lib/Res2Net_v1b.py:18
Method__init__
(self, inplanes, planes, stride=1, downsample=None, groups=1, base_width=64, dilation=1, norm
lib/resnet.py:51
Method__init__
(self, inplanes, planes, stride=1, downsample=None, groups=1, base_width=64, dilation=1, norm
lib/resnet.py:92
Method__init__
(self, cfg, load_path=None)
lib/VIT/CamoFormer.py:61
Method__init__
(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0., sr_ratio=1)
lib/VIT/encoder/pvtv2_encoder.py:56
Method__init__
(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., dr
lib/VIT/encoder/pvtv2_encoder.py:119
Method__init__
(self, img_size=224, patch_size=7, stride=4, in_chans=3, embed_dim=768)
lib/VIT/encoder/pvtv2_encoder.py:161
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, num_classes=1000, embed_dims=[64, 128, 256, 512],
lib/VIT/encoder/pvtv2_encoder.py:201
Method__init__
(self, dim=768)
lib/VIT/encoder/pvtv2_encoder.py:372
Method__init__
(self, **kwargs)
lib/VIT/encoder/pvtv2_encoder.py:398
Method__init__
(self, **kwargs)
lib/VIT/encoder/pvtv2_encoder.py:408
Method__init__
(self, **kwargs)
lib/VIT/encoder/pvtv2_encoder.py:416
Method__init__
(self, **kwargs)
lib/VIT/encoder/pvtv2_encoder.py:424
Method__init__
(self, **kwargs)
lib/VIT/encoder/pvtv2_encoder.py:432
Method__init__
(self, **kwargs)
lib/VIT/encoder/pvtv2_encoder.py:441
Method__init__
(self, dim, window_size, num_heads, qkv_bias=True, qk_scale=None, attn_drop=0., proj_drop=0.)
lib/VIT/encoder/swin_encoder.py:82
Method__init__
(self, dim, input_resolution, num_heads, window_size=7, shift_size=0, mlp_ratio=4., qkv_bias=
lib/VIT/encoder/swin_encoder.py:185
Method__init__
(self, input_resolution, dim, norm_layer=nn.LayerNorm)
lib/VIT/encoder/swin_encoder.py:303
Method__init__
(self, dim, input_resolution, depth, num_heads, window_size, mlp_ratio=4., qkv_bias=True, qk_
lib/VIT/encoder/swin_encoder.py:363
Method__init__
(self, img_size=224, patch_size=4, in_chans=3, embed_dim=96, norm_layer=None)
lib/VIT/encoder/swin_encoder.py:424
Method__init__
(self, img_size=224, patch_size=4, in_chans=3, num_classes=1000, embed_dim=96, depths=[2, 2,
lib/VIT/encoder/swin_encoder.py:487
Method__init__
(self, normalized_shape)
lib/VIT/decoder/decoder_p.py:45
Method__init__
(self, normalized_shape)
lib/VIT/decoder/decoder_p.py:61
Method__init__
(self, dim, LayerNorm_type)
lib/VIT/decoder/decoder_p.py:82
Method__init__
(self, dim, ffn_expansion_factor, bias)
lib/VIT/decoder/decoder_p.py:97
Method__init__
(self, dim, num_heads, bias,mode)
lib/VIT/decoder/decoder_p.py:115
Method__init__
(self, mode='dilation',dim=128, num_heads=8, ffn_expansion_factor=4, bias=False, LayerNorm_type='WithBias')
lib/VIT/decoder/decoder_p.py:156
Method__init__
(self, dim=128)
lib/VIT/decoder/decoder_p.py:172
Method__init__
(self, channels)
lib/VIT/decoder/decoder_p.py:198
Method__init__
(self, configer=None)
losses/TreeLoss.py:45
Method__init__
(self, configer=None)
losses/TreeLoss.py:82
Method__init__
(self, use_sigmoid=False, use_kpos=False, partial=False,
losses/cross_entropy_loss.py:111
← previousnext →101–200 of 357, ranked by callers