MCPcopy Create free account

hub / github.com/TencentARC/T2I-Adapter / functions

Functions109 in github.com/TencentARC/T2I-Adapter

⨍ Functions 109◇ Types & classes 22
↓ 9 callersFunctionresize_numpy_image
(image, max_resolution=1024 * 1024, resize_short_edge=None)
Adapter/utils.py:24
↓ 6 callersMethod__init__
(self, channels)
Adapter/extra_condition/model_edge.py:360
↓ 6 callersFunctioninstantiate_from_config
(config)
configs/utils.py:76
↓ 3 callersFunctionmake_layers
(block, no_relu_layers)
Adapter/extra_condition/openpose/model.py:6
↓ 3 callersFunctionpidinet
()
Adapter/extra_condition/model_edge.py:638
↓ 2 callersMethod__init__
(self, channels, use_conv, dims=2, out_channels=None, padding=1)
Adapter/models/adapters.py:61
↓ 2 callersFunctionimport_model_class_from_model_name_or_path
( pretrained_model_name_or_path: str, revision: str, subfolder: str = "text_encoder" )
train_sketch.py:90
↓ 2 callersFunctionimport_model_class_from_model_name_or_path
( pretrained_model_name_or_path: str, revision: str, subfolder: str = "text_encoder" )
Adapter/utils.py:5
↓ 1 callersMethod__init__
(self)
Adapter/extra_condition/openpose/api.py:18
↓ 1 callersMethod__init__
(self)
Adapter/extra_condition/openpose/model.py:23
↓ 1 callersFunctionavg_pool_nd
Create a 1D, 2D, or 3D average pooling module.
Adapter/models/adapters.py:19
↓ 1 callersFunctioncompute_embeddings
(batch, proportion_empty_prompts, text_encoders, tokenizers, is_train=True)
train_sketch.py:554
↓ 1 callersMethodcompute_embeddings
(self, prompt_batch, proportion_empty_prompts, text_encoders, tokenizers, size, is_train=True)
Adapter/Sampling.py:133
↓ 1 callersFunctionconfig_model
(model)
Adapter/extra_condition/model_edge.py:623
↓ 1 callersFunctionconv_nd
Create a 1D, 2D, or 3D convolution module.
Adapter/models/adapters.py:6
↓ 1 callersFunctioncreateConvFunc
(op_type)
Adapter/extra_condition/model_edge.py:270
↓ 1 callersFunctioncreate_demo_canny
(run)
demo.py:62
↓ 1 callersFunctioncreate_demo_pose
(run)
demo.py:121
↓ 1 callersFunctioncreate_demo_sketch
(run)
demo.py:3
↓ 1 callersFunctionencode_prompt
(prompt_batch, text_encoders, tokenizers, proportion_empty_prompts, is_train=True)
train_sketch.py:337
↓ 1 callersMethodencode_prompt
(self, prompt_batch, proportion_empty_prompts, is_train=True)
Adapter/Sampling.py:94
↓ 1 callersFunctionexists
(x)
configs/utils.py:51
↓ 1 callersFunctionget_base_argument_parser
get the base argument parser for inference scripts
Adapter/inference_base.py:7
↓ 1 callersFunctionget_cond_model
(opt, cond_type: ExtraCondition)
Adapter/extra_condition/api.py:24
↓ 1 callersFunctionget_obj_from_str
(string, reload=False)
configs/utils.py:86
↓ 1 callersFunctionget_parameter_dtype
(parameter: torch.nn.Module)
Adapter/models/adapters.py:31
↓ 1 callersFunctionget_state_dict_from_checkpoint
(pl_sd)
configs/utils.py:109
↓ 1 callersMethodinference
(self, prompt, size, prompt_n='', adapter_features=None, guidance_scale=7.5, seed=-1, steps=50)
Adapter/Sampling.py:62
↓ 1 callersFunctionmain
(args)
train_sketch.py:383
↓ 1 callersMethodmake_loader
(self, dataset_config)
dataset/dataset_laion.py:71
↓ 1 callersFunctionparse_args
(input_args=None)
train_sketch.py:110
↓ 1 callersFunctionrandom_threshold
(edge, low_threshold=0.3, high_threshold=0.8)
train_sketch.py:377
↓ 1 callersFunctionread_state_dict
(checkpoint_file, print_global_state=False)
configs/utils.py:126
↓ 1 callersMethodreset_parameters
(self)
Adapter/extra_condition/model_edge.py:345
↓ 1 callersMethodreset_schedule
(self, timesteps)
Adapter/Sampling.py:59
↓ 1 callersMethodtrain_dataloader
(self)
dataset/dataset_laion.py:133
↓ 1 callersFunctiontransform_checkpoint_dict_key
(k)
configs/utils.py:101
FunctionHWC3
(x)
Adapter/extra_condition/openpose/util.py:174
Method__call__
(self, sample)
dataset/utils.py:18
Method__call__
(self, sample)
dataset/utils.py:33
Method__call__
(self, oriImg)
Adapter/extra_condition/openpose/hand.py:27
Method__call__
(self, oriImg)
Adapter/extra_condition/openpose/body.py:26
Method__init__
(self)
dataset/utils.py:15
Method__init__
(self, low_threshold=100, high_threshold=200, shift_range=50)
dataset/utils.py:27
Method__init__
(self, tar_base1, tar_base2, batch_size, t
dataset/dataset_laion.py:46
Method__init__
(self, model_id)
Adapter/Sampling.py:15
Method__init__
(self, in_c, out_c, down, ksize=3, sk=False, use_conv=True)
Adapter/models/adapters.py:81
Method__init__
(self, channels=[320, 640, 1280, 1280], nums_rb=3, cin=64, ksize=3, sk=False, use_conv=True)
Adapter/models/adapters.py:117
Method__init__
(self, pdc, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=False)
Adapter/extra_condition/model_edge.py:324
Method__init__
(self, in_channels, out_channels)
Adapter/extra_condition/model_edge.py:382
Method__init__
(self, channels)
Adapter/extra_condition/model_edge.py:407
Method__init__
(self, pdc, inplane, ouplane, stride=1)
Adapter/extra_condition/model_edge.py:417
Method__init__
(self, pdc, inplane, ouplane, stride=1)
Adapter/extra_condition/model_edge.py:445
Method__init__
(self, inplane, pdcs, dil=None, sa=False, convert=False)
Adapter/extra_condition/model_edge.py:471
Method__init__
(self, model_path)
Adapter/extra_condition/openpose/hand.py:18
Method__init__
(self, model_path)
Adapter/extra_condition/openpose/body.py:17
Method__init__
(self)
Adapter/extra_condition/openpose/api.py:42
Method__init__
(self)
Adapter/extra_condition/openpose/model.py:121
Functioncount_params
(model, verbose=False)
configs/utils.py:69
Functiondefault
(val, d)
configs/utils.py:55
Functiondict_collation_fn
Take a list of samples (as dictionary) and create a batch, preserving the keys. If `tensors` is True, `ndarray` objects are combined into ten
dataset/dataset_laion.py:13
Functiondraw_bodypose
(canvas, candidate, subset)
Adapter/extra_condition/openpose/util.py:40
Functiondraw_handpose
(canvas, all_hand_peaks, show_number=False)
Adapter/extra_condition/openpose/util.py:77
Methoddtype
`torch.dtype`: The dtype of the module (assuming that all the module parameters have the same dtype).
Adapter/models/adapters.py:138
Methodfilter_keys
(self, x)
dataset/dataset_laion.py:127
Methodfilter_size
(self, x)
dataset/dataset_laion.py:118
Functionfind_tensor_attributes
(module: torch.nn.Module)
Adapter/models/adapters.py:44
Methodforward
r""" The [`UNet2DConditionModel`] forward method. Args: sample (`torch.FloatTensor`): The noisy input ten
models/unet.py:6
Methodforward
(self, x)
Adapter/models/adapters.py:74
Methodforward
(self, x)
Adapter/models/adapters.py:100
Methodforward
(self, x)
Adapter/models/adapters.py:144
Methodforward
(self, input)
Adapter/extra_condition/model_edge.py:352
Methodforward
(self, x)
Adapter/extra_condition/model_edge.py:370
Methodforward
(self, x)
Adapter/extra_condition/model_edge.py:393
Methodforward
(self, x)
Adapter/extra_condition/model_edge.py:412
Methodforward
(self, x)
Adapter/extra_condition/model_edge.py:429
Methodforward
(self, x)
Adapter/extra_condition/model_edge.py:459
Methodforward
(self, x)
Adapter/extra_condition/model_edge.py:564
Methodforward
(self, x)
Adapter/extra_condition/openpose/api.py:28
Methodforward
(self, x)
Adapter/extra_condition/openpose/api.py:52
Methodforward
(self, x)
Adapter/extra_condition/openpose/model.py:89
Methodforward
(self, x)
Adapter/extra_condition/openpose/model.py:165
Functionfunc
(x, weights, bias=None, stride=1, padding=0, dilation=1, groups=1)
Adapter/extra_condition/model_edge.py:276
Functionget_adapter_feature
(inputs, adapters)
Adapter/extra_condition/api.py:303
Functionget_cond_canny
(opt, cond_image, cond_inp_type='image', cond_model=None)
Adapter/extra_condition/api.py:206
Functionget_cond_color
(opt, cond_image, cond_inp_type='image', cond_model=None)
Adapter/extra_condition/api.py:240
Functionget_cond_depth
(opt, cond_image, cond_inp_type='image', cond_model=None)
Adapter/extra_condition/api.py:163
Functionget_cond_edge
(opt, cond_image, cond_inp_type='image', cond_model=None)
Adapter/extra_condition/api.py:281
Functionget_cond_keypose
(opt, cond_image, cond_inp_type='image', cond_model=None)
Adapter/extra_condition/api.py:115
Functionget_cond_openpose
(opt, cond_image, cond_inp_type='image', cond_model=None)
Adapter/extra_condition/api.py:255
Functionget_cond_seg
(opt, cond_image, cond_inp_type='image', cond_model=None)
Adapter/extra_condition/api.py:99
Functionget_cond_sketch
(opt, cond_image, cond_inp_type, cond_model=None)
Adapter/extra_condition/api.py:75
Functionget_cond_style
(opt, cond_image, cond_inp_type='image', cond_model=None)
Adapter/extra_condition/api.py:226
Functionget_cond_zoedepth
(opt, cond_image, cond_inp_type='image', cond_model=None)
Adapter/extra_condition/api.py:184
Methodget_weights
(self)
Adapter/extra_condition/model_edge.py:550
FunctionhandDetect
(candidate, subset, oriImg)
Adapter/extra_condition/openpose/util.py:101
Functionisimage
(x)
configs/utils.py:45
Functionismap
(x)
configs/utils.py:39
Functionload_model_from_config
(config, ckpt, vae_ckpt=None, verbose=False)
configs/utils.py:140
Functionlog_txt_as_img
(wh, xc, size=10)
configs/utils.py:15
next →1–100 of 109, ranked by callers