Functions — github.com/Fannovel16/comfyui_controlnet

hub / github.com/Fannovel16/comfyui_controlnet_aux / functions

Functions4,192 in github.com/Fannovel16/comfyui_controlnet_aux

⨍ Functions 4,192◇ Types & classes 975↳ Endpoints 1

↓ 2 callersFunctionrot_aa

Rotate axis angle parameters.

src/custom_mesh_graphormer/utils/image_ops.py:160

↓ 2 callersMethodrun_iter

(self, data_batch, train_mode, **kwargs)

src/custom_mmpkg/custom_mmcv/runner/epoch_based_runner.py:24

↓ 2 callersFunctionrun_validate

(args, val_loader, Graphormer_model, criterion, criterion_vertices, epoch, smpl, mesh_sampler)

src/custom_mesh_graphormer/tools/run_gphmer_bodymesh.py:390

↓ 2 callersFunctionsafer_memory

(x)

src/custom_controlnet_aux/util.py:141

↓ 2 callersFunctionsafer_memory

(x)

src/custom_controlnet_aux/dsine/__init__.py:42

↓ 2 callersFunctionsafer_memory

(x)

src/custom_controlnet_aux/zoe/transformers.py:33

↓ 2 callersMethodsample_fn

(self, shape, up_scale=1, unnormalize=True, cond=None, denoise=False)

src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/ddm_const_sde.py:880

↓ 2 callersFunctionsave_checkpoint

Save checkpoint to file. The checkpoint will have 3 fields: ``meta``, ``state_dict`` and ``optimizer``. By default ``meta`` will contain vers

src/custom_mmpkg/custom_mmcv/runner/checkpoint.py:634

↓ 2 callersFunctionsave_checkpoint

(model, args, epoch, iteration, num_trial=10)

src/custom_mesh_graphormer/tools/run_gphmer_bodymesh.py:46

↓ 2 callersMethodsave_checkpoint

(self, out_dir, filename_tmpl, save_op

src/custom_mmpkg/custom_mmcv/runner/base_runner.py:197

↓ 2 callersFunctionscandir

Scan a directory to find the interested files. Args: dir_path (str | obj:`Path`): Path of the directory. suffix (str | tuple(str)

src/custom_mmpkg/custom_mmcv/utils/path.py:39

↓ 2 callersFunctionscatter

Scatter inputs to target gpus. The only difference from original :func:`scatter` is to add support for :type:`~mmcv.parallel.DataContainer`.

src/custom_mmpkg/custom_mmcv/parallel/scatter_gather.py:9

↓ 2 callersMethodscatter

(self, inputs, kwargs, device_ids)

src/custom_mmpkg/custom_mmcv/parallel/distributed.py:26

↓ 2 callersMethodset_input

Unpack input data from the dataloader and perform necessary pre-processing steps. Parameters: input (dict): includes the data its

src/custom_controlnet_aux/leres/pix2pix/models/base_model.py:64

↓ 2 callersMethodset_requires_grad

Set requies_grad=Fasle for all the networks to avoid unnecessary computations Parameters: nets (network list) -- a list of netwo

src/custom_controlnet_aux/leres/pix2pix/models/base_model.py:233

↓ 2 callersMethodset_updated_estimate

(self, est)

src/custom_controlnet_aux/leres/leres/depthmap.py:302

↓ 2 callersMethodsetup_category_id_and_number

(self)

src/custom_controlnet_aux/diffusion_edge/taming/data/annotated_objects_dataset.py:142

↓ 2 callersFunctionsetup_logger

Initialize the detectron2 logger and set its verbosity level to "DEBUG". Args: output (str): a file name or a directory to save log.

src/custom_controlnet_aux/metric3d/mono/utils/logger.py:30

↓ 2 callersMethodsince_last_check

Time since the last checking. Either :func:`since_start` or :func:`since_last_check` is a checking operation. Returns (float

src/custom_mmpkg/custom_mmcv/utils/timer.py:74

↓ 2 callersFunctionsingle_gpu_test

Test with single GPU. Args: model (nn.Module): Model to be tested. data_loader (utils.data.Dataloader): Pytorch data loader.

src/custom_mmpkg/custom_mmseg/apis/test.py:34

↓ 2 callersFunctionsingle_head_full_attention_1d

(q, k, v, h=None, w=None,

src/custom_controlnet_aux/unimatch/unimatch/attention.py:19

↓ 2 callersFunctionsingleestimate

(img, msize, model, net_type)

src/custom_controlnet_aux/leres/leres/depthmap.py:156

↓ 2 callersMethodstart

(self)

src/custom_mmpkg/custom_mmcv/utils/progressbar.py:26

↓ 2 callersFunctionstr_to_indices

Expects a string in the format '32-123, 256, 280-321

src/custom_controlnet_aux/diffusion_edge/taming/data/imagenet.py:25

↓ 2 callersFunctionsubtract_flat_id

(rot_mats)

src/custom_manopth/tensutils.py:34

↓ 2 callersFunctionsynchronize

Helper function to synchronize (barrier) among all processes when using distributed training

src/custom_mesh_graphormer/utils/comm.py:39

↓ 2 callersMethodtest

Forward function used in test time. This function wraps <forward> function in no_grad() so we don't save intermediate steps for backprop

src/custom_controlnet_aux/leres/pix2pix/models/base_model.py:102

↓ 2 callersFunctionth_posemap_axisang

(pose_vectors)

src/custom_manopth/tensutils.py:6

↓ 2 callersFunctionthin_one_time

src/custom_controlnet_aux/mlsd/__init__.py:28

↓ 2 callersMethodto

(self, device)

src/custom_controlnet_aux/pidi/__init__.py:29

↓ 2 callersMethodto

(self, device)

src/custom_controlnet_aux/normalbae/__init__.py:53

↓ 2 callersMethodto

(self, device)

src/custom_controlnet_aux/midas/transformers.py:51

↓ 2 callersFunctionto8U

(img)

src/custom_controlnet_aux/tile/guided_filter.py:21

↓ 2 callersMethodtoken_pair_from_bbox

(self, bbox: BoundingBox)

src/custom_controlnet_aux/diffusion_edge/taming/data/conditional_builder/objects_center_points.py:68

↓ 2 callersFunctiontorch_gc

()

src/custom_controlnet_aux/util.py:200

↓ 2 callersFunctiontrack_progress

Track the progress of tasks execution with a progress bar. Tasks are done with a simple for-loop. Args: func (callable): The functio

src/custom_mmpkg/custom_mmcv/utils/progressbar.py:64

↓ 2 callersMethodtrain_step

(self, *inputs, **kwargs)

src/custom_mmpkg/custom_mmcv/parallel/data_parallel.py:47

↓ 2 callersFunctiontransform_test_data_scalecano

Pre-process the input for forwarding. Employ `label scale canonical transformation.' Args: rgb: input rgb image. [H, W, 3]

src/custom_controlnet_aux/metric3d/mono/utils/do_test.py:157

↓ 2 callersFunctiontrunc_normal_init

(module: nn.Module, mean: float = 0, std: float = 1,

src/custom_mmpkg/custom_mmcv/cnn/utils/weight_init.py:73

↓ 2 callersFunctiontry_delete

(f)

src/custom_mesh_graphormer/utils/miscellaneous.py:145

↓ 2 callersFunctiontsv_reader

(tsv_file, sep='\t')

src/custom_mesh_graphormer/utils/tsv_file_ops.py:56

↓ 2 callersFunctiontsv_writer

(values, tsv_file, sep='\t')

src/custom_mesh_graphormer/utils/tsv_file_ops.py:38

↓ 2 callersFunctionuncrop_boxes_xyxy

(boxes: torch.Tensor, crop_box: List[int])

src/custom_controlnet_aux/sam/utils/amg.py:237

↓ 2 callersFunctionuniform_init

(module, a=0, b=1, bias=0)

src/custom_mmpkg/custom_mmcv/cnn/utils/weight_init.py:85

↓ 2 callersFunctionuniform_quant

(x, b)

src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/quantization.py:7

↓ 2 callersMethodunnormalize_image

(self, img)

src/custom_controlnet_aux/unimatch/utils/logger.py:73

↓ 2 callersMethodupdate_scale

update the current loss scale value when overflow happens.

src/custom_mmpkg/custom_mmcv/runner/fp16_utils.py:372

↓ 2 callersFunctionupsample_flow_with_mask

(flow, up_mask, upsample_factor, is_depth=False)

src/custom_controlnet_aux/unimatch/unimatch/utils.py:134

↓ 2 callersMethodval_step

(self, *inputs, **kwargs)

src/custom_mmpkg/custom_mmcv/parallel/data_parallel.py:69

↓ 2 callersFunctionvertices_loss

Compute per-vertex loss if vertex annotations are available.

src/custom_mesh_graphormer/tools/run_gphmer_handmesh.py:102

↓ 2 callersFunctionvgg16

VGG-16 from `Very Deep Convolutional Networks for Large-Scale Image Recognition <https://arxiv.org/abs/1409.1556>`__. Args: weights (:cla

src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/vgg.py:403

↓ 2 callersFunctionvisualize_mesh

Tensorboard logging.

src/custom_mesh_graphormer/tools/run_gphmer_handmesh.py:415

↓ 2 callersFunctionvisualize_reconstruction

Overlays gt_kp and pred_kp on img. Draws vert with text. Renderer is an instance of SMPLRenderer.

src/custom_mesh_graphormer/utils/renderer.py:259

↓ 2 callersFunctionvisualize_reconstruction_and_att_local

Overlays gt_kp and pred_kp on img. Draws vert with text. Renderer is an instance of SMPLRenderer.

src/custom_mesh_graphormer/utils/renderer.py:397

↓ 2 callersFunctionvisualize_reconstruction_test

Overlays gt_kp and pred_kp on img. Draws vert with text. Renderer is an instance of SMPLRenderer.

src/custom_mesh_graphormer/utils/renderer.py:286

↓ 2 callersFunctionvit_large

(patch_size=14, checkpoint=None, **kwargs)

src/custom_controlnet_aux/metric3d/mono/model/backbones/ViT_DINO.py:1349

↓ 2 callersFunctionvit_large_reg

(patch_size=14, num_register_tokens=4, checkpoint=None, tuning_mode=None, **kwargs)

src/custom_controlnet_aux/metric3d/mono/model/backbones/ViT_DINO_reg.py:1236

↓ 2 callersFunctionviz_depth_tensor

(disp, return_numpy=False, colormap='plasma')

src/custom_controlnet_aux/unimatch/utils/visualization.py:92

↓ 2 callersFunctionweight_quantization

(b)

src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/quantization.py:6

↓ 2 callersFunctionwrap_fp16_model

Wrap the FP32 model to FP16. If you are using PyTorch >= 1.6, torch.cuda.amp is used as the backend, otherwise, original mmcv implementation

src/custom_mmpkg/custom_mmcv/runner/fp16_utils.py:234

↓ 2 callersFunctionwrite_frame

(file_idx)

src/custom_mmpkg/custom_mmcv/video/io.py:308

↓ 2 callersFunctionwt

(x, filters, in_size, level)

src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/wavelet.py:32

↓ 1 callersFunctionDINOv2

(model_name)

src/custom_controlnet_aux/depth_anything_v2/dinov2.py:398

↓ 1 callersMethodDice_Loss

(self, pred, label)

src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/ddm_const_sde.py:391

↓ 1 callersFunctionDownsample

(dim, dim_out = None)

src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/uncond_unet.py:65

↓ 1 callersFunctionDownsample

(dim, dim_out = None)

src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/mask_cond_unet.py:346

↓ 1 callersMethodINPUT_TYPES

(s)

__init__.py:86

↓ 1 callersMethodMASK

()

utils.py:105

↓ 1 callersMethodMHA_GCN

(self, hidden_states, attention_mask, head_mask=None, history_state=None)

src/custom_mesh_graphormer/modeling/bert/modeling_graphormer.py:136

↓ 1 callersFunctionNEG_INF_DIAG

Returns a diagonal matrix of size [n, n]. The diagonal are all "-inf". This is for avoiding calculating the overlapped element in the Criss-C

src/custom_mmpkg/custom_mmcv/ops/cc_attention.py:9

↓ 1 callersMethodSEED

(default=0)

utils.py:107

↓ 1 callersMethodSTRING

(default='', multiline=False)

utils.py:115

↓ 1 callersFunctionUpsample

(dim, dim_out = None)

src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/uncond_unet.py:59

↓ 1 callersMethod__call__

(self, module)

src/custom_mmpkg/custom_mmcv/cnn/utils/weight_init.py:222

↓ 1 callersMethod__enter__

(self)

src/custom_mmpkg/custom_mmcv/utils/timer.py:49

↓ 1 callersMethod__getattr__

(self, name)

src/custom_mmpkg/custom_mmcv/utils/config.py:506

↓ 1 callersMethod__init__

(self, in_channels, out_channels, kernel_size,

src/custom_mmpkg/custom_mmcv/ops/modulated_deform_conv.py:160

↓ 1 callersMethod__init__

(self, in_channels: int, out_channels: int, kernel_size: Un

src/custom_mmpkg/custom_mmcv/ops/deform_conv.py:228

↓ 1 callersMethod__init__

(self, gamma, alpha, weight=None, reduction='mean')

src/custom_mmpkg/custom_mmcv/ops/focal_loss.py:90

↓ 1 callersMethod__init__

(self, use_xyz: bool = True)

src/custom_mmpkg/custom_mmcv/ops/group_points.py:141

↓ 1 callersMethod__init__

(self, start=True, print_tmpl=None)

src/custom_mmpkg/custom_mmcv/utils/timer.py:38

↓ 1 callersMethod__init__

(self, dataloader, start=None, interval=1,

src/custom_mmpkg/custom_mmcv/runner/hooks/evaluation.py:85

↓ 1 callersMethod__init__

(self, drop_prob=0.5, inplace=False)

src/custom_mmpkg/custom_mmcv/cnn/bricks/drop.py:59

↓ 1 callersMethod__init__

(self, in_channels, out_channels, kernel_size,

src/custom_mmpkg/custom_mmcv/cnn/bricks/conv_ws.py:28

↓ 1 callersMethod__init__

(self, *args, by_epoch=False, efficient_test=False, **kwargs)

src/custom_mmpkg/custom_mmseg/core/evaluation/eval_hooks.py:22

↓ 1 callersMethod__init__

(self, split, **kwargs)

src/custom_mmpkg/custom_mmseg/datasets/pascal_context.py:47

↓ 1 callersMethod__init__

(self, groups=1, base_width=4, **kwargs)

src/custom_mmpkg/custom_mmseg/models/backbones/resnext.py:134

↓ 1 callersMethod__init__

(self, extra, in_channels=3, conv_cfg=None,

src/custom_mmpkg/custom_mmseg/models/backbones/hrnet.py:273

↓ 1 callersMethod__init__

(self, in_channels, out_channels, stride,

src/custom_mmpkg/custom_mmseg/models/utils/inverted_residual.py:31

↓ 1 callersMethod__init__

(self, c1_in_channels, c1_channels, **kwargs)

src/custom_mmpkg/custom_mmseg/models/decode_heads/sep_aspp_head.py:42

↓ 1 callersMethod__init__

(self, dilations=(1, 6, 12, 18), **kwargs)

src/custom_mmpkg/custom_mmseg/models/decode_heads/aspp_head.py:64

↓ 1 callersMethod__init__

(self, reduction=2, use_scale=True, mode='embedded_gaussian

src/custom_mmpkg/custom_mmseg/models/decode_heads/dnl_head.py:102

↓ 1 callersMethod__init__

(self, pool_scales, in_channels, channels, conv_cfg, norm_cfg, act_cfg, align_corners)

src/custom_mmpkg/custom_mmseg/models/decode_heads/psp_head.py:24

↓ 1 callersMethod__init__

(self, num_codes=32, use_se_loss=True, add_lateral=False,

src/custom_mmpkg/custom_mmseg/models/decode_heads/enc_head.py:78

↓ 1 callersMethod__init__

(self, filter_size, fusion, in_channels, channels, conv_cfg, norm_cfg, act_cfg)

src/custom_mmpkg/custom_mmseg/models/decode_heads/dm_head.py:24

↓ 1 callersMethod__init__

(self, ema_channels, num_bases, num_stages,

src/custom_mmpkg/custom_mmseg/models/decode_heads/ema_head.py:94

↓ 1 callersMethod__init__

(self, pool_scale, fusion, in_channels, channels, conv_cfg, norm_cfg, act_cfg)

src/custom_mmpkg/custom_mmseg/models/decode_heads/apc_head.py:25

↓ 1 callersMethod__init__

(self, img_file, label_file=None, hw_file=None, linelist_file=None, is_train=True, cv2_output

src/custom_mesh_graphormer/datasets/human_mesh_tsv.py:23

↓ 1 callersMethod__init__

(self, args, img_file, label_file=None, hw_file=None, linelist_file=None, is_train=True, cv2_

src/custom_mesh_graphormer/datasets/hand_mesh_tsv.py:24

← previous next →701–800 of 4,192, ranked by callers