Method__init__(self, image_channel, block_expansion, num_down_blocks, max_features, reshape_channel, reshape_depth, num_resb
src/modules/appearance_feature_extractor.py:14
Method__init__(self, upscale=1, max_features=256, block_expansion=64, out_channels=64, num_down_blocks=2)
src/modules/spade_generator.py:14
Method__init__(self, block_expansion, num_blocks, max_features, num_kp, feature_channel, reshape_depth, compress, estimate_o
src/modules/dense_motion.py:14
Method__init__(
self,
in_chans=3,
depths=[3, 3, 9, 3],
dims=[96, 192, 384, 768],
dro
src/modules/convnextv2.py:60
Method__init__(self, model_config_path, model_checkpoint_path, embeddings_cache_path=None, cpu_only=False, **kwargs)
src/utils/animal_landmark_runner.py:26
Method__init__(self,
d_model=256, d_ffn=1024,
dropout=0.1, activation="relu",
src/utils/dependencies/XPose/models/UniPose/transformer_deformable.py:216
Method__init__(self, d_model=256, d_ffn=1024,
dropout=0.1, activation="relu",
n_levels=4,
src/utils/dependencies/XPose/models/UniPose/transformer_deformable.py:328
Method__init__(self, decoder_layer, num_layers, return_intermediate=False, use_dab=False, d_model=256, query_dim=4)
src/utils/dependencies/XPose/models/UniPose/transformer_deformable.py:516
Method__init__(self, d_model=256, nhead=8,
num_queries=300,
num_encoder_layers=6,
src/utils/dependencies/XPose/models/UniPose/deformable_transformer.py:35
Method__init__(self, decoder_layer, num_layers, norm=None,
return_intermediate=False,
d_mo
src/utils/dependencies/XPose/models/UniPose/deformable_transformer.py:692
Method__init__(self,
d_model=256, d_ffn=1024,
dropout=0.1, activation="relu",
src/utils/dependencies/XPose/models/UniPose/deformable_transformer.py:939
Method__init__(self, d_model=256, d_ffn=1024,
dropout=0.1, activation="relu",
n_levels=4,
src/utils/dependencies/XPose/models/UniPose/deformable_transformer.py:997
Method__init__(self, v_dim, l_dim, embed_dim, num_heads, dropout=0.1, cfg=None)
src/utils/dependencies/XPose/models/UniPose/fuse_modules.py:98
Method__init__(self, dim, window_size, num_heads, qkv_bias=True, qk_scale=None, attn_drop=0., proj_drop=0.)
src/utils/dependencies/XPose/models/UniPose/swin_transformer.py:78
Method__init__(self, dim, num_heads, window_size=7, shift_size=0,
mlp_ratio=4., qkv_bias=True, qk_scale=Non
src/utils/dependencies/XPose/models/UniPose/swin_transformer.py:163
Method__init__(self, embed_dim, num_heads, dropout=0., bias=True, add_bias_kv=False, add_zero_attn=False, kdim=None, vdim=No
src/utils/dependencies/XPose/models/UniPose/attention.py:65
Method__init__(self, d_model, nhead, dim_feedforward=2048, dropout=0.1, activation="relu", normalize_before=False)
src/utils/dependencies/XPose/models/UniPose/transformer_vanilla.py:59
Method__init__(self, num_pos_feats=64, temperatureH=10000, temperatureW=10000, normalize=False, scale=None)
src/utils/dependencies/XPose/models/UniPose/position_encoding.py:71