MCPcopy Create free account

hub / github.com/Fannovel16/comfyui_controlnet_aux / functions

Functions4,192 in github.com/Fannovel16/comfyui_controlnet_aux

↓ 1 callersMethod__init__
(self, face_model_path, inference_size=None, gaussian_sigma=None,
src/custom_controlnet_aux/open_pose/face.py:317
↓ 1 callersMethod__init__
(self)
src/custom_controlnet_aux/open_pose/model.py:25
↓ 1 callersMethod__init__
(self, input_nc, output_nc, n_residual_blocks=9, sigmoid=True)
src/custom_controlnet_aux/lineart/__init__.py:36
↓ 1 callersMethod__init__
Init. Args: features (int): number of features
src/custom_controlnet_aux/depth_anything_v2/util/blocks.py:33
↓ 1 callersMethod__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
src/custom_controlnet_aux/depth_anything_v2/dinov2_layers/swiglu_ffn.py:14
↓ 1 callersMethod__init__
Construct a Unet generator Parameters: input_nc (int) -- the number of channels in input images output_nc (int) -- th
src/custom_controlnet_aux/lineart_anime/__init__.py:19
↓ 1 callersMethod__init__
(self)
src/custom_controlnet_aux/dsine/models/dsine_arch.py:49
↓ 1 callersMethod__init__
(self, encoder)
src/custom_controlnet_aux/leres/leres/multi_depth_model_woauxi.py:26
↓ 1 callersMethod__init__
(self, face_model_path, inference_size=None, gaussian_sigma=None,
src/custom_controlnet_aux/dwpose/face.py:317
↓ 1 callersMethod__init__
(self)
src/custom_controlnet_aux/dwpose/model.py:25
↓ 1 callersMethod__init__
(self, output_dim=128, norm_layer=nn.InstanceNorm2d, num_output_scales=1,
src/custom_controlnet_aux/unimatch/unimatch/backbone.py:40
↓ 1 callersMethod__init__
( self, input_dim: int, hidden_dim: int, output_dim: int, num_layers:
src/custom_controlnet_aux/sam/modeling/mask_decoder.py:155
↓ 1 callersMethod__init__
( self, embedding_dim: int, mlp_dim: int, act: Type[nn.Module] = nn.GELU,
src/custom_controlnet_aux/sam/modeling/common.py:14
↓ 1 callersMethod__init__
Encodes prompts for input to SAM's mask decoder. Arguments: embed_dim (int): The prompts' embedding dimension im
src/custom_controlnet_aux/sam/modeling/prompt_encoder.py:17
↓ 1 callersMethod__init__
(self, bit_a, signed=False)
src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/quantization.py:85
↓ 1 callersMethod__init__
( self, model, *, image_size, timesteps = 1000, sampling_times
src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/ddm_const_sde.py:44
↓ 1 callersMethod__init__
Construct a PatchGAN discriminator Parameters: input_nc (int) -- the number of channels in input images ndf (int)
src/custom_controlnet_aux/diffusion_edge/taming/modules/discriminator/model.py:21
↓ 1 callersMethod__init__
(self)
src/custom_controlnet_aux/diffusion_edge/taming/modules/losses/vqperceptual.py:10
↓ 1 callersMethod__init__
(self, size=256, random_crop=False, interpolation="bicubic")
src/custom_controlnet_aux/diffusion_edge/taming/data/ade20k.py:12
↓ 1 callersMethod__patch_instance_norm_state_dict
Fix InstanceNorm checkpoints incompatibility (prior to 0.4)
src/custom_controlnet_aux/leres/pix2pix/models/base_model.py:176
↓ 1 callersMethod_add_children
Add children for a registry. The ``registry`` will be added as children based on its scope. The parent registry could build objects f
src/custom_mmpkg/custom_mmcv/utils/registry.py:214
↓ 1 callersMethod_auxiliary_head_forward_train
Run forward function and calculate loss for auxiliary head in training.
src/custom_mmpkg/custom_mmseg/models/segmentors/encoder_decoder.py:113
↓ 1 callersMethod_binary_op
(self, x1, x2)
src/custom_mmpkg/custom_mmcv/ops/merge_cells.py:89
↓ 1 callersFunction_cached_log_stream
(filename)
src/custom_controlnet_aux/metric3d/mono/utils/logger.py:97
↓ 1 callersMethod_check_branches
Check branches configuration.
src/custom_mmpkg/custom_mmseg/models/backbones/hrnet.py:46
↓ 1 callersMethod_check_branches
(self, num_branches, blocks, num_blocks, num_inchannels, num_channels)
src/custom_mesh_graphormer/modeling/hrnet/hrnet_cls_net_gridfeat.py:127
↓ 1 callersMethod_check_branches
(self, num_branches, blocks, num_blocks, num_inchannels, num_channels)
src/custom_mesh_graphormer/modeling/hrnet/hrnet_cls_net.py:127
↓ 1 callersMethod_check_image
(self, image_rgb)
src/custom_controlnet_aux/densepose/densepose.py:188
↓ 1 callersMethod_check_input_divisible
(self, x)
src/custom_mmpkg/custom_mmseg/models/backbones/unet.py:399
↓ 1 callersMethod_check_mask_matrix
(self, mask, matrix)
src/custom_controlnet_aux/densepose/densepose.py:193
↓ 1 callersFunction_concat_dataset
Build :obj:`ConcatDataset by.
src/custom_mmpkg/custom_mmseg/datasets/builder.py:17
↓ 1 callersMethod_convert_to_label_id
Convert trainId to id for cityscapes.
src/custom_mmpkg/custom_mmseg/datasets/cityscapes.py:39
↓ 1 callersMethod_convert_to_onehot_labels
Convert segmentation label to onehot. Args: seg_label (Tensor): Segmentation label of shape (N, H, W). num_classes (i
src/custom_mmpkg/custom_mmseg/models/decode_heads/enc_head.py:159
↓ 1 callersMethod_crop_encoder
(self, crop_coordinates: BoundingBox)
src/custom_controlnet_aux/diffusion_edge/taming/data/conditional_builder/objects_center_points.py:129
↓ 1 callersMethod_decode_head_forward_test
Run forward function and calculate loss for decode head in inference.
src/custom_mmpkg/custom_mmseg/models/segmentors/encoder_decoder.py:107
↓ 1 callersMethod_decode_head_forward_train
Run forward function and calculate loss for decode head in training.
src/custom_mmpkg/custom_mmseg/models/segmentors/encoder_decoder.py:96
↓ 1 callersMethod_dump_init_info
Dump the initialization information to a file named `initialization.log.json` in workdir. Args: logger_name (str): The na
src/custom_mmpkg/custom_mmcv/runner/base_module.py:137
↓ 1 callersMethod_embed_boxes
Embeds box prompts.
src/custom_controlnet_aux/sam/modeling/prompt_encoder.py:93
↓ 1 callersMethod_embed_masks
Embeds mask inputs.
src/custom_controlnet_aux/sam/modeling/prompt_encoder.py:102
↓ 1 callersMethod_embed_masks
(self, input_mask: torch.Tensor, has_mask_input: torch.Tensor)
src/custom_controlnet_aux/sam/utils/onnx.py:69
↓ 1 callersMethod_embed_points
Embeds point prompts.
src/custom_controlnet_aux/sam/modeling/prompt_encoder.py:73
↓ 1 callersMethod_embed_points
(self, point_coords: torch.Tensor, point_labels: torch.Tensor)
src/custom_controlnet_aux/sam/utils/onnx.py:51
↓ 1 callersMethod_evaluate_cityscapes
Evaluation in Cityscapes protocol. Args: results (list): Testing results of the dataset. logger (logging.Logger | str
src/custom_mmpkg/custom_mmseg/datasets/cityscapes.py:164
↓ 1 callersFunction_expand_onehot_labels
Expand onehot labels to match the size of prediction.
src/custom_mmpkg/custom_mmseg/models/losses/cross_entropy_loss.py:35
↓ 1 callersMethod_file2dict
(filename, use_predefined_variables=True)
src/custom_mmpkg/custom_mmcv/utils/config.py:179
↓ 1 callersMethod_filter_relpaths
(self, relpaths)
src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/imagenet.py:48
↓ 1 callersMethod_filter_relpaths
(self, relpaths)
src/custom_controlnet_aux/diffusion_edge/taming/data/imagenet.py:60
↓ 1 callersFunction_find_free_port
()
src/custom_controlnet_aux/metric3d/mono/utils/comm.py:39
↓ 1 callersFunction_fix_aspect_ratio
Extend the scale to match the given aspect ratio. Args: scale (np.ndarray): The image scale (w, h) in shape (2, ) aspect_ratio (f
src/custom_controlnet_aux/dwpose/dw_torchscript/jit_pose.py:153
↓ 1 callersFunction_fix_aspect_ratio
Extend the scale to match the given aspect ratio. Args: scale (np.ndarray): The image scale (w, h) in shape (2, ) aspect_ratio (f
src/custom_controlnet_aux/dwpose/dw_onnx/cv_ox_pose.py:154
↓ 1 callersMethod_forward_impl
(self, x)
src/custom_controlnet_aux/mlsd/models/mbv2_mlsd_tiny.py:218
↓ 1 callersMethod_forward_impl
(self, x)
src/custom_controlnet_aux/mlsd/models/mbv2_mlsd_large.py:218
↓ 1 callersMethod_forward_impl
(self, x)
src/custom_controlnet_aux/leres/leres/Resnext_torch.py:196
↓ 1 callersMethod_forward_impl
(self, x: Tensor)
src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/efficientnet.py:355
↓ 1 callersMethod_forward_impl
(self, x: Tensor)
src/custom_controlnet_aux/diffusion_edge/denoising_diffusion_pytorch/resnet.py:266
↓ 1 callersMethod_freeze_stages
(self)
src/custom_mmpkg/custom_mmseg/models/backbones/mobilenet_v3.py:242
↓ 1 callersMethod_freeze_stages
(self)
src/custom_mmpkg/custom_mmseg/models/backbones/mobilenet_v2.py:164
↓ 1 callersFunction_fuse_conv_bn
Fuse conv and bn into one module. Args: conv (nn.Module): Conv to be fused. bn (nn.Module): BN to be fused. Returns:
src/custom_mmpkg/custom_mmcv/cnn/utils/fuse_conv_bn.py:6
↓ 1 callersMethod_generate_detectron2_style_palette
Generate professional color palette with good visual separation.
src/custom_controlnet_aux/oneformer/transformers.py:111
↓ 1 callersMethod_generate_masks
(self, image: np.ndarray)
src/custom_controlnet_aux/sam/automatic_mask_generator.py:197
↓ 1 callersMethod_generate_professional_colormap
Generate professional colormap for segmentation visualization.
src/custom_controlnet_aux/oneformer/transformers.py:96
↓ 1 callersMethod_get_batch_size
Gets the batch size of the output given the batch size of the input prompts.
src/custom_controlnet_aux/sam/modeling/prompt_encoder.py:107
↓ 1 callersMethod_get_checkpoint_loader
Finds a loader that supports the given path. Falls back to the local loader if no other loader is found. Args: path (str)
src/custom_mmpkg/custom_mmcv/runner/checkpoint.py:211
↓ 1 callersFunction_get_conv
()
src/custom_mmpkg/custom_mmcv/utils/parrots_wrapper.py:41
↓ 1 callersFunction_get_cuda_home
()
src/custom_mmpkg/custom_mmcv/utils/parrots_wrapper.py:21
↓ 1 callersFunction_get_dataloader
()
src/custom_mmpkg/custom_mmcv/utils/parrots_wrapper.py:49
↓ 1 callersMethod_get_device
(self)
src/custom_controlnet_aux/sam/modeling/prompt_encoder.py:125
↓ 1 callersFunction_get_extension
()
src/custom_mmpkg/custom_mmcv/utils/parrots_wrapper.py:58
↓ 1 callersMethod_get_init_info
(self)
src/custom_mmpkg/custom_mmcv/cnn/utils/weight_init.py:194
↓ 1 callersMethod_get_init_info
(self)
src/custom_mmpkg/custom_mmcv/cnn/utils/weight_init.py:237
↓ 1 callersMethod_get_init_info
(self)
src/custom_mmpkg/custom_mmcv/cnn/utils/weight_init.py:280
↓ 1 callersMethod_get_init_info
(self)
src/custom_mmpkg/custom_mmcv/cnn/utils/weight_init.py:335
↓ 1 callersMethod_get_init_info
(self)
src/custom_mmpkg/custom_mmcv/cnn/utils/weight_init.py:378
↓ 1 callersMethod_get_init_info
(self)
src/custom_mmpkg/custom_mmcv/cnn/utils/weight_init.py:440
↓ 1 callersMethod_get_init_info
(self)
src/custom_mmpkg/custom_mmcv/cnn/utils/weight_init.py:506
↓ 1 callersMethod_get_intermediate_layers_chunked
(self, x, n=1)
src/custom_controlnet_aux/depth_anything_v2/dinov2.py:283
↓ 1 callersMethod_get_intermediate_layers_chunked
(self, x, n=1)
src/custom_controlnet_aux/metric3d/mono/model/backbones/ViT_DINO.py:791
↓ 1 callersMethod_get_intermediate_layers_chunked
(self, x, n=1)
src/custom_controlnet_aux/metric3d/mono/model/backbones/ViT_DINO.py:1264
↓ 1 callersMethod_get_intermediate_layers_chunked
(self, x, n=1)
src/custom_controlnet_aux/metric3d/mono/model/backbones/ViT_DINO_reg.py:1035
↓ 1 callersMethod_get_intermediate_layers_not_chunked
(self, x, n=1)
src/custom_controlnet_aux/depth_anything_v2/dinov2.py:271
↓ 1 callersMethod_get_intermediate_layers_not_chunked
(self, x, n=1)
src/custom_controlnet_aux/metric3d/mono/model/backbones/ViT_DINO.py:779
↓ 1 callersMethod_get_intermediate_layers_not_chunked
(self, x, n=1)
src/custom_controlnet_aux/metric3d/mono/model/backbones/ViT_DINO.py:1252
↓ 1 callersMethod_get_intermediate_layers_not_chunked
(self, x, n=1)
src/custom_controlnet_aux/metric3d/mono/model/backbones/ViT_DINO_reg.py:1023
↓ 1 callersMethod_get_max_memory
(self, runner)
src/custom_mmpkg/custom_mmcv/runner/hooks/logger/text.py:109
↓ 1 callersFunction_get_norm
()
src/custom_mmpkg/custom_mmcv/utils/parrots_wrapper.py:81
↓ 1 callersFunction_get_pool
()
src/custom_mmpkg/custom_mmcv/utils/parrots_wrapper.py:69
↓ 1 callersMethod_get_scale_ratio
(self, intrinsic)
src/custom_controlnet_aux/metric3d/mono/utils/transform.py:119
↓ 1 callersFunction_get_shear_matrix
Generate the shear matrix for transformation. Args: magnitude (int | float): The magnitude used for shear. direction (str): The f
src/custom_mmpkg/custom_mmcv/image/geometric.py:593
↓ 1 callersFunction_get_translate_matrix
Generate the translate matrix. Args: offset (int | float): The offset used for translate. direction (str): The translate directio
src/custom_mmpkg/custom_mmcv/image/geometric.py:662
↓ 1 callersMethod_has_inf_or_nan
Check if params contain NaN.
src/custom_mmpkg/custom_mmcv/runner/fp16_utils.py:358
↓ 1 callersMethod_inference
(self, img)
src/custom_controlnet_aux/uniformer/__init__.py:34
↓ 1 callersMethod_initFilter
(self)
src/custom_controlnet_aux/tile/guided_filter.py:165
↓ 1 callersMethod_initFilter
(self)
src/custom_controlnet_aux/tile/guided_filter.py:204
↓ 1 callersMethod_init_auxiliary_head
Initialize ``auxiliary_head``
src/custom_mmpkg/custom_mmseg/models/segmentors/encoder_decoder.py:49
↓ 1 callersMethod_init_decode_head
Initialize ``decode_head``
src/custom_mmpkg/custom_mmseg/models/segmentors/encoder_decoder.py:43
↓ 1 callersFunction_init_dist_mpi
(backend, **kwargs)
src/custom_mmpkg/custom_mmcv/runner/dist_utils.py:35
↓ 1 callersFunction_init_dist_mpi
(backend, **kwargs)
src/custom_controlnet_aux/unimatch/utils/dist_utils.py:33
↓ 1 callersFunction_init_dist_pytorch
(backend, **kwargs)
src/custom_mmpkg/custom_mmcv/runner/dist_utils.py:27
↓ 1 callersFunction_init_dist_pytorch
(backend, **kwargs)
src/custom_controlnet_aux/unimatch/utils/dist_utils.py:25
↓ 1 callersFunction_init_dist_ror
(cfg)
src/custom_controlnet_aux/metric3d/mono/utils/comm.py:87
← previousnext →801–900 of 4,192, ranked by callers