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

↓ 9 callersFunctionNormalize

(in_channels, num_groups=32)

src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/encoder_decoder.py:56

↓ 9 callersFunctionNormalize

(in_channels)

src/custom_controlnet_aux/diffusion_edge/taming/modules/diffusionmodules/model.py:34

↓ 9 callersMethod__init__

( self, dim, init_dim = None, out_dim = None, dim_mults=(1, 2, 4, 8),

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

↓ 9 callersMethod__init__

(self, *, ch, out_ch, ch_mult=(1,2,4,8), num_res_blocks, attn_resolutions, dropout=0.0, resam

src/custom_controlnet_aux/diffusion_edge/taming/modules/diffusionmodules/model.py:196

↓ 9 callersFunctioncast_tensor_type

Recursively convert Tensor in inputs from src_type to dst_type. Args: inputs: Inputs that to be casted. src_type (torch.dtype): S

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

↓ 9 callersFunctionfunc

(x, weights, bias=None, stride=1, padding=0, dilation=1, groups=1)

src/custom_controlnet_aux/pidi/model.py:304

↓ 9 callersMethodinfer_client

Infer a suitable file client based on the URI and arguments. Args: file_client_args (dict, optional): Arguments to instantiate a

src/custom_mmpkg/custom_mmcv/fileio/file_client.py:858

↓ 9 callersFunctionis_module_wrapper

Check if a module is a module wrapper. The following 3 modules in MMCV (and their subclasses) are regarded as module wrappers: DataParallel,

src/custom_mmpkg/custom_mmcv/parallel/utils.py:5

↓ 9 callersFunctionnormalize_vector

(v)

src/custom_manopth/rot6d.py:54

↓ 9 callersMethodnum_rows

(self)

src/custom_mesh_graphormer/utils/tsv_file.py:62

↓ 9 callersFunctionorthographic_projection

Perform orthographic projection of 3D points X using the camera parameters Args: X: size = [B, N, 3] camera: size = [B, 3] Ret

src/custom_mesh_graphormer/utils/geometric_layers.py:46

↓ 9 callersMethodto

(self, device)

src/custom_controlnet_aux/leres/__init__.py:42

↓ 9 callersMethodto

(self, device)

src/custom_controlnet_aux/sam/sam.py:51

↓ 9 callersFunctiontrunc_normal_

r"""Fills the input Tensor with values drawn from a truncated normal distribution. The values are effectively drawn from the normal distributi

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

↓ 9 callersMethodupdate

(self)

src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/ema.py:141

↓ 8 callersMethod__init__

(self, in_ch=3, mid_ch=12, out_ch=3, img_size=512)

src/custom_controlnet_aux/anime_face_segment/isnet.py:93

↓ 8 callersMethod__init__

(self)

src/custom_controlnet_aux/manga_line/model_torch.py:151

↓ 8 callersMethod__init__

(self, inchannels, outchannels, upfactor=2)

src/custom_controlnet_aux/leres/leres/network_auxi.py:240

↓ 8 callersMethod__init__

(self)

src/custom_controlnet_aux/teed/ted.py:193

↓ 8 callersFunction_vgg

(cfg: str, batch_norm: bool, weights: Optional[WeightsEnum], progress: bool, **kwargs: Any)

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

↓ 8 callersFunctionadd_prefix

Add prefix for dict. Args: inputs (dict): The input dict with str keys. prefix (str): The prefix to add. Returns: d

src/custom_mmpkg/custom_mmseg/core/utils/misc.py:1

↓ 8 callersFunctioncoords_grid

(b, h, w, homogeneous=False, device=None)

src/custom_controlnet_aux/unimatch/unimatch/geometry.py:5

↓ 8 callersMethodevery_n_iters

(self, runner, n)

src/custom_mmpkg/custom_mmcv/runner/hooks/hook.py:61

↓ 8 callersMethodget_3d_joints

This method is used to get the joint locations from the SMPL mesh Input: vertices: size = (B, 778, 3) Output:

src/custom_mesh_graphormer/modeling/_mano.py:56

↓ 8 callersFunctionget_root_logger

Get the root logger. The logger will be initialized if it has not been initialized. By default a StreamHandler will be added. If `log_file` i

src/custom_mmpkg/custom_mmseg/utils/logger.py:6

↓ 8 callersFunctionmkdir

(path)

src/custom_mesh_graphormer/utils/miscellaneous.py:15

↓ 8 callersFunctionreturn_pil

(name, processor, img)

tests/test_controlnet_aux.py:29

↓ 8 callersMethodscatter

(self, inputs, kwargs, device_ids)

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

↓ 8 callersFunctionunnormalize_to_zero_to_one

(t)

src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/utils.py:61

↓ 8 callersFunctionupdate_init_info

Update the `_params_init_info` in the module if the value of parameters are changed. Args: module (obj:`nn.Module`): The module of Py

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

↓ 8 callersMethodupsample

(self, h, pred_norm, uv_8)

src/custom_controlnet_aux/dsine/models/dsine_arch.py:108

↓ 8 callersMethodupsample_flow

(self, flow, feature, bilinear=False, upsample_factor=8, is_depth=False)

src/custom_controlnet_aux/unimatch/unimatch/unimatch.py:81

↓ 7 callersMethod__init__

(self, *, bias=0, bias_prob=None, layer=None)

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

↓ 7 callersMethod__init__

(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)

src/custom_controlnet_aux/uniformer/uniformer.py:24

↓ 7 callersMethod__init__

(self, config=None)

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

↓ 7 callersMethod__setattr__

(self, name, value)

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

↓ 7 callersMethodbuild

(self, *args, **kwargs)

src/custom_mmpkg/custom_mmcv/utils/registry.py:211

↓ 7 callersMethodclear

(self)

src/custom_mmpkg/custom_mmcv/runner/log_buffer.py:15

↓ 7 callersMethodconvert

Multiple with alpha and add beat with clip.

src/custom_mmpkg/custom_mmseg/datasets/pipelines/transforms.py:804

↓ 7 callersMethodencode

(self, c)

src/custom_controlnet_aux/diffusion_edge/taming/modules/util.py:110

↓ 7 callersMethodget_h36m_joints

This method is used to get the joint locations from the SMPL mesh Input: vertices: size = (B, 6890, 3) Output:

src/custom_mesh_graphormer/modeling/_smpl.py:133

↓ 7 callersFunctionget_prediction_head

(input_dim, hidden_dim, output_dim)

src/custom_controlnet_aux/dsine/models/submodules/__init__.py:169

↓ 7 callersMethodjoin_path

Concatenate all file paths. Join one or more filepath components intelligently. The return value is the concatenation of filepath and

src/custom_mmpkg/custom_mmcv/fileio/file_client.py:1079

↓ 7 callersFunctionmake_attn

(in_channels, attn_type="vanilla")

src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/encoder_decoder.py:223

↓ 7 callersFunctionnorm_normalize

(norm_out)

src/custom_controlnet_aux/normalbae/nets/submodules/submodules.py:64

↓ 7 callersFunctionobsolete_torch_version

(torch_version, version_threshold)

src/custom_mmpkg/custom_mmcv/cnn/bricks/wrappers.py:24

↓ 7 callersFunctionsparse_to_dense

(sparse_tensor)

src/wrapper_for_mps/__init__.py:6

↓ 7 callersFunctionxavier_init

(module, gain=1, bias=0, distribution='normal')

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

↓ 6 callersFunctionHWC3

(x)

src/custom_controlnet_aux/dsine/__init__.py:21

↓ 6 callersMethod__init__

(self, in_c1, in_c2, out_c1, out_c2, upscale = True)

src/custom_controlnet_aux/mlsd/models/mbv2_mlsd_tiny.py:10

↓ 6 callersMethod__init__

(self, in_c1, in_c2, out_c1, out_c2, upscale = True)

src/custom_controlnet_aux/mlsd/models/mbv2_mlsd_large.py:10

↓ 6 callersMethod__init__

(self, channels)

src/custom_controlnet_aux/pidi/model.py:386

↓ 6 callersMethod__init__

Initialize the GANLoss class. Parameters: gan_mode (str) - - the type of GAN objective. It currently supports vanilla, lsgan, an

src/custom_controlnet_aux/leres/pix2pix/models/networks.py:224

↓ 6 callersFunction_get_bases_name

(m)

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

↓ 6 callersMethodbackward

(ctx, a=None)

src/custom_mmpkg/custom_mmcv/ops/knn.py:73

↓ 6 callersMethodclip_grads

(self, params)

src/custom_mmpkg/custom_mmcv/runner/hooks/optimizer.py:27

↓ 6 callersMethodconstrain_to_multiple_of

(self, x, min_val=0, max_val=None)

src/custom_controlnet_aux/depth_anything_v2/util/transform.py:51

↓ 6 callersFunctionconvert_color_factory

(src, dst)

src/custom_mmpkg/custom_mmcv/image/colorspace.py:275

↓ 6 callersMethodcrop

Crop from ``img``

src/custom_mmpkg/custom_mmseg/datasets/pipelines/transforms.py:523

↓ 6 callersMethodfilter

(self, keep: torch.Tensor)

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

↓ 6 callersFunctionformat_param

(name, optim, param)

src/custom_mmpkg/custom_mmcv/runner/hooks/lr_updater.py:659

↓ 6 callersMethodforward

(self, x)

src/custom_mmpkg/custom_mmcv/cnn/bricks/wrappers.py:44

↓ 6 callersMethodforward_plugin

Forward function for plugins.

src/custom_mmpkg/custom_mmseg/models/backbones/resnet.py:242

↓ 6 callersMethodget_loggable_tags

(self, runner, allow_scalar=True,

src/custom_mmpkg/custom_mmcv/runner/hooks/logger/base.py:112

↓ 6 callersFunctionget_world_size

()

src/custom_mesh_graphormer/utils/comm.py:19

↓ 6 callersFunctionhas_method

Check whether the object has a method. Args: method (str): The method name to check. obj (object): The object to check. Retu

src/custom_mmpkg/custom_mmcv/utils/misc.py:367

↓ 6 callersFunctionis_main_process

()

src/custom_mesh_graphormer/utils/comm.py:35

↓ 6 callersFunctionis_supported_instance

(module)

src/custom_mmpkg/custom_mmcv/cnn/utils/flops_counter.py:540

↓ 6 callersMethodload_checkpoint

(self, filename, map_location='cpu', s

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

↓ 6 callersFunctionnormalize_to_neg_one_to_one

(img)

src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/utils.py:58

↓ 6 callersMethodparse

Parse our options, create checkpoints directory suffix, and set up gpu device.

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

↓ 6 callersFunctionpoint_sample

A wrapper around :func:`grid_sample` to support 3D point_coords tensors Unlike :func:`torch.nn.functional.grid_sample` it assumes point_coords to

src/custom_mmpkg/custom_mmcv/ops/point_sample.py:242

↓ 6 callersMethodrandom_sample

Randomly sample an img_scale when ``multiscale_mode=='range'``. Args: img_scales (list[tuple]): Images scale range for sampling.

src/custom_mmpkg/custom_mmseg/datasets/pipelines/transforms.py:86

↓ 6 callersFunctionresize_image_with_pad

(input_image, resolution, upscale_method="INTER_CUBIC", skip_hwc3=False, mode='edge')

src/custom_controlnet_aux/dsine/__init__.py:45

↓ 6 callersFunctionrotateY

(points, angle)

src/custom_mesh_graphormer/utils/renderer.py:21

↓ 6 callersFunctionrun_script

(cmd, cwd='.')

utils.py:221

↓ 6 callersFunctionsplit_feature

(feature, num_splits=2, channel_last=False, )

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

↓ 6 callersMethodto

(self, device)

src/custom_controlnet_aux/depth_anything_v2/__init__.py:36

↓ 6 callersFunctionto32F

(img)

src/custom_controlnet_aux/tile/guided_filter.py:15

↓ 5 callersMethod__init__

(self, config)

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

↓ 5 callersMethod__init__

(self, inp_ch=32, output_chal=1, down_sample_times=3, channel_mode='v0')

src/custom_controlnet_aux/metric3d/mono/model/decode_heads/HourGlassDecoder.py:98

↓ 5 callersMethod_set_lr

(self, runner, lr_groups)

src/custom_mmpkg/custom_mmcv/runner/hooks/lr_updater.py:58

↓ 5 callersMethod_set_momentum

(self, runner, momentum_groups)

src/custom_mmpkg/custom_mmcv/runner/hooks/momentum_updater.py:35

↓ 5 callersFunctionannealing_cos

Calculate annealing cos learning rate. Cosine anneal from `weight * start + (1 - weight) * end` to `end` as percentage goes from 0.0 to 1.0.

src/custom_mmpkg/custom_mmcv/runner/hooks/lr_updater.py:627

↓ 5 callersMethodaverage

Average latest n values or all values.

src/custom_mmpkg/custom_mmcv/runner/log_buffer.py:33

↓ 5 callersFunctionbuild_activation_layer

Build activation layer. Args: cfg (dict): The activation layer config, which should contain: - type (str): Layer type.

src/custom_mmpkg/custom_mmcv/cnn/bricks/activation.py:81

↓ 5 callersFunctioncommon_input_validate

(input_image, output_type, **kwargs)

src/custom_controlnet_aux/dsine/__init__.py:72

↓ 5 callersMethoddecode

(self, z)

src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/encoder_decoder.py:958

↓ 5 callersFunctiondecompress_keypoints

( numbers: Optional[List[float]], )

src/custom_controlnet_aux/dwpose/__init__.py:149

↓ 5 callersFunctiondefault

(val, d)

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

↓ 5 callersFunctiondefault

(val, d)

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

↓ 5 callersFunctiondownload

(url, local_path, chunk_size=1024)

src/custom_controlnet_aux/diffusion_edge/taming/util.py:18

↓ 5 callersFunctionexists

(val)

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

↓ 5 callersFunctionexists

(x)

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

↓ 5 callersFunctionflow_to_image

Convert flow into middlebury color code image :param flow: optical flow map :return: optical flow image in middlebury color

src/custom_controlnet_aux/unimatch/utils/flow_viz.py:240

↓ 5 callersMethodforward_test

Args: imgs (List[Tensor]): the outer list indicates test-time augmentations and inner Tensor should have a shape

src/custom_mmpkg/custom_mmseg/models/segmentors/base.py:76

↓ 5 callersMethodget

Read data from a given ``filepath`` with 'rb' mode. Note: There are two types of return values for ``get``, one is ``bytes``

src/custom_mmpkg/custom_mmcv/fileio/file_client.py:976

↓ 5 callersMethodget_key

(self, idx)

src/custom_mesh_graphormer/utils/tsv_file.py:84

↓ 5 callersFunctionget_rank

()

src/custom_mesh_graphormer/utils/comm.py:27

↓ 5 callersMethodinit

(m)

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

← previous next →101–200 of 4,192, ranked by callers