MCPcopy Create free account

hub / github.com/IrisRainbowNeko/ML-Danbooru-webui / functions

Functions177 in github.com/IrisRainbowNeko/ML-Danbooru-webui

↓ 56 callersFunction_cfg
(url='', **kwargs)
mldanbooru/caformer/metaformer_baselines.py:39
↓ 12 callersMethod__init__
(self, dim, mlp_ratio=4, out_features=None, act_layer=StarReLU, drop=0., bias=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:449
↓ 11 callersFunctionconv2d_ABN
(ni, nf, stride, activation="leaky_relu", kernel_size=3, activation_param=1e-2, groups=1)
mldanbooru/tresnet/tresnet.py:39
↓ 8 callersFunctionconv2d_FReLU
(ni, nf, stride, kernel_size=3, groups=1)
mldanbooru/tresnet/tresnet_f.py:18
↓ 4 callersMethod__init__
(self, channels, reduction_channels, inplace=True)
mldanbooru/tresnet/layers/general_layers.py:78
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1, use_se=True, anti_alias_layer=None)
mldanbooru/tresnet/tresnet.py:186
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1, use_se=True, anti_alias_layer=None)
mldanbooru/tresnet/tresnet_f.py:160
↓ 3 callersFunctionconv2d_BN
(ni, nf, stride, kernel_size=3)
mldanbooru/tresnet/tresnet_f.py:11
↓ 3 callersMethodwith_pos_embed
(self, tensor, pos: Optional[torch.Tensor])
mldanbooru/caformer/ms_decoder.py:43
↓ 2 callersMethod__init__
(self, layers, in_chans=3, num_classes=1000, width_factor=1.0, first_two_layers=BasicBlock)
mldanbooru/tresnet/tresnet.py:134
↓ 2 callersMethod__init__
(self, layers, in_chans=3, num_classes=1000, width_factor=1.0, first_two_layers=BasicBlock)
mldanbooru/tresnet/tresnet_f.py:108
↓ 2 callersFunctionbuild_caformer
(model_name, args)
mldanbooru/caformer/ml_caformer.py:57
↓ 2 callersMethodbuild_transform
(self, image_size, keep_ratio=False)
mldanbooru/interface.py:97
↓ 2 callersFunctioncreate_model
Create a model
mldanbooru/utils/factory.py:9
↓ 2 callersMethodinfer_
(self, img: Image.Image, thr: float)
mldanbooru/interface.py:111
↓ 2 callersMethodload_model
(self, path=DEFAULT_MODEL)
mldanbooru/interface.py:57
↓ 1 callersMethod__init__
(self, decoder_layer, num_layers, norm=None, return_intermediate=False)
mldanbooru/caformer/ms_decoder.py:95
↓ 1 callersMethod__init__
(self, num_pos_feats=64, temperature=10000, normalize=False, scale=None)
mldanbooru/caformer/position_encoding.py:15
↓ 1 callersMethod__init__
(self, embed_len_decoder: int)
mldanbooru/ml_decoder/ml_decoder.py:34
↓ 1 callersMethod__init__
(self, d_model, nhead=8, dim_feedforward=2048, dropout=0.1, activation="relu", layer_norm_eps
mldanbooru/ml_decoder/layer.py:13
↓ 1 callersMethod__init__
(self, filt_size: int = 3, stride: int = 2, channels: int = 0)
mldanbooru/tresnet/layers/anti_aliasing.py:20
↓ 1 callersMethod__setstate__
(self, state)
mldanbooru/ml_decoder/layer.py:33
↓ 1 callersFunction_get_clones
(module, N)
mldanbooru/caformer/ms_decoder.py:90
↓ 1 callersFunctionbuild_position_encoding
(position_embedding, hidden_dim=512)
mldanbooru/caformer/position_encoding.py:76
↓ 1 callersFunctionbuild_ui
(infer)
scripts/mld_main.py:8
↓ 1 callersMethoddecode
(self, feat_list)
mldanbooru/caformer/ml_caformer.py:41
↓ 1 callersMethodencode
(self, x)
mldanbooru/caformer/ml_caformer.py:32
↓ 1 callersMethodforward_features
(self, x)
mldanbooru/caformer/metaformer_baselines.py:668
↓ 1 callersMethodload_class_map
(self)
mldanbooru/interface.py:92
↓ 1 callersMethodreset_parameters
(self)
mldanbooru/caformer/position_encoding.py:59
FunctionInplacABN_to_ABN
(module: nn.Module)
mldanbooru/tresnet/tresnet.py:15
FunctionTResnetD
Constructs a large TResnet model.
mldanbooru/tresnet/tresnet.py:229
FunctionTResnetD
Constructs a large TResnet model.
mldanbooru/tresnet/tresnet_f.py:204
FunctionTResnetL
Constructs a large TResnet model.
mldanbooru/tresnet/tresnet.py:238
FunctionTResnetL
Constructs a large TResnet model.
mldanbooru/tresnet/tresnet_f.py:214
FunctionTResnetM
Constructs a medium TResnet model.
mldanbooru/tresnet/tresnet.py:221
FunctionTResnetM
Constructs a medium TResnet model.
mldanbooru/tresnet/tresnet_f.py:195
FunctionTResnetS
Constructs a small TResnet model.
mldanbooru/tresnet/tresnet.py:212
FunctionTResnetS
Constructs a small TResnet model.
mldanbooru/tresnet/tresnet_f.py:185
FunctionTResnetXL
Constructs a large TResnet model.
mldanbooru/tresnet/tresnet.py:248
FunctionTResnetXL
Constructs a large TResnet model.
mldanbooru/tresnet/tresnet_f.py:223
Method__call__
(self, h: torch.Tensor, duplicate_pooling: torch.Tensor, out_extrap: torch.Tensor)
mldanbooru/ml_decoder/ml_decoder.py:37
Method__call__
(self, x: torch.Tensor)
mldanbooru/tresnet/layers/general_layers.py:55
Method__init__
(self)
mldanbooru/interface.py:39
Method__init__
(self, encoder: MetaFormer, decoder: MSDecoder, num_queries=50, d_model=512, num_classes=1000, scale_skip=0)
mldanbooru/caformer/ml_caformer.py:14
Method__init__
(self, d_model, nhead, dim_feedforward=2048, dropout=0.1, activation=nn.GELU(), normalize_bef
mldanbooru/caformer/ms_decoder.py:17
Method__init__
(self, in_channels, out_channels, kernel_size, stride=1, padding=0, pre_norm=None, post_norm
mldanbooru/caformer/metaformer_baselines.py:209
Method__init__
(self, dim, init_value=1.0, trainable=True)
mldanbooru/caformer/metaformer_baselines.py:234
Method__init__
(self, inplace=False)
mldanbooru/caformer/metaformer_baselines.py:246
Method__init__
(self, scale_value=1.0, bias_value=0.0, scale_learnable=True, bias_learnable=True, mode=None,
mldanbooru/caformer/metaformer_baselines.py:257
Method__init__
(self, dim, head_dim=32, num_heads=None, qkv_bias=False, attn_drop=0., proj_drop=0., proj_bias=False,
mldanbooru/caformer/metaformer_baselines.py:276
Method__init__
(self, num_tokens=196, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:332
Method__init__
(self, affine_shape=None, normalized_dim=(-1, ), scale=True, bias=True, eps=1e-5)
mldanbooru/caformer/metaformer_baselines.py:378
Method__init__
(self, dim, expansion_ratio=2, act1_layer=StarReLU, act2_layer=nn.Identity, bias=False, kerne
mldanbooru/caformer/metaformer_baselines.py:403
Method__init__
(self, pool_size=3, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:433
Method__init__
(self, dim, num_classes=1000, mlp_ratio=4, act_layer=SquaredReLU, norm_layer=nn.LayerNorm, head_dropou
mldanbooru/caformer/metaformer_baselines.py:474
Method__init__
(self, dim, token_mixer=nn.Identity, mlp=Mlp, norm_layer=nn.LayerNorm,
mldanbooru/caformer/metaformer_baselines.py:498
Method__init__
(self, in_chans=3, num_classes=1000, depths=[2, 2, 6, 2], dims=[64, 128, 32
mldanbooru/caformer/metaformer_baselines.py:583
Method__init__
(self, num_pos_feats=256)
mldanbooru/caformer/position_encoding.py:53
Method__init__
(self, num_classes, num_of_groups=-1, decoder_embedding=768, initial_num_features=2048, zsl=0
mldanbooru/ml_decoder/ml_decoder.py:47
Method__init__
(self, d_model, nhead=8, dim_feedforward=2048, dropout=0.1, activation="relu", layer_norm_eps
mldanbooru/ml_decoder/layer.py:58
Method__init__
(self, inplanes, planes, stride=1, downsample=None, use_se=True, anti_alias_layer=None)
mldanbooru/tresnet/tresnet.py:50
Method__init__
(self, inplanes, planes, stride=1, downsample=None, use_se=True, anti_alias_layer=None)
mldanbooru/tresnet/tresnet.py:89
Method__init__
(self, inplanes, planes, stride=1, downsample=None, use_se=True, anti_alias_layer=None)
mldanbooru/tresnet/tresnet_f.py:29
Method__init__
(self, inplanes, planes, stride=1, downsample=None, use_se=True, anti_alias_layer=None)
mldanbooru/tresnet/tresnet_f.py:68
Method__init__
(self, filt_size: int = 3, stride: int = 2, channels: int = 0)
mldanbooru/tresnet/layers/anti_aliasing.py:9
Method__init__
(self, flatten=False)
mldanbooru/tresnet/layers/avg_pool.py:8
Method__init__
(self, block_size)
mldanbooru/tresnet/layers/general_layers.py:15
Method__init__
(self, remove_model_jit=False)
mldanbooru/tresnet/layers/general_layers.py:28
Method__init__
(self, block_size=4)
mldanbooru/tresnet/layers/general_layers.py:40
Method__init__
(self, inplace=True)
mldanbooru/tresnet/layers/general_layers.py:65
Method__init__
(self, in_channel)
mldanbooru/tresnet/layers/frelu.py:6
Method__setstate__
(self, state)
mldanbooru/ml_decoder/layer.py:85
Method_init_weights
(self, m)
mldanbooru/caformer/metaformer_baselines.py:658
Functionadd_ml_decoder_head
(model, num_classes=-1, num_of_groups=-1, decoder_embedding=768, zsl=0, learn_query=False)
mldanbooru/ml_decoder/ml_decoder.py:11
Functioncaformer_b36
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1493
Functioncaformer_b36_384
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1509
Functioncaformer_b36_384_in21ft1k
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1541
Functioncaformer_b36_in21ft1k
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1525
Functioncaformer_b36_in21k
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1557
Functioncaformer_d30
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1482
Functioncaformer_m36
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1403
Functioncaformer_m364_in21k
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1467
Functioncaformer_m36_384
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1419
Functioncaformer_m36_384_in21ft1k
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1451
Functioncaformer_m36_in21ft1k
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1435
Functioncaformer_s18
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1243
Functioncaformer_s18_384
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1259
Functioncaformer_s18_384_in21ft1k
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1291
Functioncaformer_s18_in21ft1k
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1275
Functioncaformer_s18_in21k
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1307
Functioncaformer_s36
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1323
Functioncaformer_s36_384
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1339
Functioncaformer_s36_384_in21ft1k
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1371
Functioncaformer_s36_in21ft1k
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1355
Functioncaformer_s36_in21k
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1387
Functionconv2d
(ni, nf, stride)
mldanbooru/tresnet/tresnet.py:31
Functionconvformer_b36
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1163
Functionconvformer_b36_384
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1179
Functionconvformer_b36_384_in21ft1k
(pretrained=False, **kwargs)
mldanbooru/caformer/metaformer_baselines.py:1211
next →1–100 of 177, ranked by callers