MCPcopy Create free account

hub / github.com/ShihaoZhaoZSH/Uni-ControlNet / functions

Functions2,265 in github.com/ShihaoZhaoZSH/Uni-ControlNet

↓ 3 callersFunction_scale_channel
Scale the data in the corresponding channel.
annotator/uniformer/mmcv/image/photometric.py:145
↓ 3 callersMethod_sync_params
(self)
annotator/uniformer/mmcv/parallel/distributed_deprecated.py:37
↓ 3 callersFunctionaccuracy
Calculate accuracy according to the prediction and target. Args: pred (torch.Tensor): The model prediction, shape (N, num_class, ...)
annotator/uniformer/mmseg/models/losses/accuracy.py:4
↓ 3 callersFunctionadd_blur
(img, sf=4)
ldm/modules/image_degradation/bsrgan_light.py:324
↓ 3 callersFunctionallreduce_params
Allreduce parameters. Args: params (list[torch.Parameters]): List of parameters or buffers of a model. coalesce (bool
annotator/uniformer/mmcv/runner/dist_utils.py:99
↓ 3 callersFunctioncolor_val
Convert various input to color tuples. Args: color (:obj:`Color`/str/tuple/int/ndarray): Color inputs Returns: tuple[int]: A
annotator/uniformer/mmcv/visualization/color.py:24
↓ 3 callersFunctionconvert_video
Convert a video with ffmpeg. This provides a general api to ffmpeg, the executed command is:: `ffmpeg -y <pre_options> -i <in_file> <opt
annotator/uniformer/mmcv/video/processing.py:11
↓ 3 callersFunctioncount_params
(model, verbose=False)
ldm/util.py:65
↓ 3 callersMethodcrop
Crop from ``img``
annotator/uniformer/mmseg/datasets/pipelines/transforms.py:523
↓ 3 callersMethoddecode
(self, z)
ldm/models/autoencoder.py:88
↓ 3 callersMethoddpm_solver_first_update
DPM-Solver-1 (equivalent to DDIM) from time `s` to time `t`. Args: x: A pytorch tensor. The initial value at time `s`.
ldm/models/diffusion/dpm_solver/dpm_solver.py:469
↓ 3 callersMethodencode_decode
Encode images with backbone and decode into a semantic segmentation map of the same size as input.
annotator/uniformer/mmseg/models/segmentors/encoder_decoder.py:84
↓ 3 callersMethodencode_first_stage
(self, x)
ldm/models/diffusion/ddpm.py:831
↓ 3 callersMethodevery_n_epochs
(self, runner, n)
annotator/uniformer/mmcv/runner/hooks/hook.py:55
↓ 3 callersMethodforward_train
Placeholder for Forward function for training.
annotator/uniformer/mmseg/models/segmentors/base.py:51
↓ 3 callersMethodforward_train
Forward function for training. Args: img (Tensor): Input images. img_metas (list[dict]): List of image info dict wher
annotator/uniformer/mmseg/models/segmentors/encoder_decoder.py:136
↓ 3 callersFunctionfspecial
python code from: https://github.com/ronaldosena/imagens-medicas-2/blob/40171a6c259edec7827a6693a93955de2bd39e76/Aulas/aula_2_-_uniform_filte
ldm/modules/image_degradation/bsrgan_light.py:209
↓ 3 callersFunctionfspecial
python code from: https://github.com/ronaldosena/imagens-medicas-2/blob/40171a6c259edec7827a6693a93955de2bd39e76/Aulas/aula_2_-_uniform_filte
ldm/modules/image_degradation/bsrgan.py:210
↓ 3 callersFunctionget_class_weight
Get class weight for loss function. Args: class_weight (list[float] | str | None): If class_weight is a str, take it as a fil
annotator/uniformer/mmseg/models/losses/utils.py:8
↓ 3 callersMethodget_first_stage_encoding
(self, encoder_posterior)
ldm/models/diffusion/ddpm.py:655
↓ 3 callersMethodget_input
(self, batch, k)
ldm/models/autoencoder.py:102
↓ 3 callersMethodget_loss
(self, pred, target, mean=True)
ldm/models/diffusion/ddpm.py:367
↓ 3 callersMethodget_text
Read data from a given ``filepath`` with 'r' mode. Args: filepath (str or Path): Path to read data. encoding (str): T
annotator/uniformer/mmcv/fileio/file_client.py:994
↓ 3 callersMethodget_unconditional_conditioning
(self, batch_size, null_label=None)
ldm/models/diffusion/ddpm.py:1124
↓ 3 callersMethodget_weighting
(self, h, w, Ly, Lx, device)
ldm/models/diffusion/ddpm.py:698
↓ 3 callersFunctionimresize
Resize image to a given size. Args: img (ndarray): The input image. size (tuple[int]): Target size (w, h). return_scale (
annotator/uniformer/mmcv/image/geometric.py:51
↓ 3 callersFunctionimshow
Show an image. Args: img (str or ndarray): The image to be displayed. win_name (str): The window name. wait_time (int): V
annotator/uniformer/mmcv/visualization/image.py:9
↓ 3 callersFunctionimwrite
Write image to file. Args: img (ndarray): Image array to be written. file_path (str): Image file path. params (None or li
annotator/uniformer/mmcv/image/io.py:242
↓ 3 callersMethodinference
Inference with slide/whole style. Args: img (Tensor): The input image of shape (N, 3, H, W). img_meta (dict): Image i
annotator/uniformer/mmseg/models/segmentors/encoder_decoder.py:233
↓ 3 callersFunctionisimage
(x)
ldm/util.py:41
↓ 3 callersFunctionismap
(x)
ldm/util.py:35
↓ 3 callersFunctionload_from_http
load checkpoint through HTTP or HTTPS scheme path. In distributed setting, this function only download checkpoint at local rank 0. Args:
annotator/uniformer/mmcv/runner/checkpoint.py:267
↓ 3 callersMethodlog_images
(self, batch, N=8, n_row=2, sample=True, return_keys=None, **kwargs)
ldm/models/diffusion/ddpm.py:477
↓ 3 callersMethodlow_vram_shift
(self, is_diffusing)
models/uni_controlnet.py:157
↓ 3 callersMethodmake_block_plugins
make plugins for block. Args: in_channels (int): Input channels of plugin. plugins (list[dict]): List of plugins cfg
annotator/uniformer/mmseg/models/backbones/resnet.py:219
↓ 3 callersFunctionmake_ddim_sampling_parameters
(alphacums, ddim_timesteps, eta, verbose=True)
ldm/modules/diffusionmodules/util.py:63
↓ 3 callersFunctionmake_ddim_timesteps
(ddim_discr_method, num_ddim_timesteps, num_ddpm_timesteps, verbose=True)
ldm/modules/diffusionmodules/util.py:46
↓ 3 callersFunctionmake_layers
(block, no_relu_layers)
annotator/openpose/model.py:7
↓ 3 callersMethodmarginal_alpha
Compute alpha_t of a given continuous-time label t in [0, T].
ldm/models/diffusion/dpm_solver/dpm_solver.py:120
↓ 3 callersMethodput
(self, key, val)
annotator/uniformer/mmcv/video/io.py:30
↓ 3 callersMethodscatter
(self, inputs, kwargs, device_ids)
annotator/uniformer/mmcv/parallel/distributed_deprecated.py:52
↓ 3 callersMethodshared_step
(self, batch)
ldm/models/diffusion/ddpm.py:427
↓ 3 callersMethodsinglestep_dpm_solver_second_update
Singlestep solver DPM-Solver-2 from time `s` to time `t`. Args: x: A pytorch tensor. The initial value at time `s`.
ldm/models/diffusion/dpm_solver/dpm_solver.py:515
↓ 3 callersFunctionssim
(img1, img2)
ldm/modules/image_degradation/utils_image.py:669
↓ 3 callersFunctiontimestep_embedding
Create sinusoidal timestep embeddings. :param timesteps: a 1-D Tensor of N indices, one per batch element. These may be
ldm/modules/diffusionmodules/util.py:154
↓ 3 callersMethodto_rgb
(self, x)
ldm/models/autoencoder.py:192
↓ 3 callersMethodto_rgb
(self, x)
ldm/models/diffusion/ddpm.py:1303
↓ 2 callersFunctionNormalize
(in_channels)
ldm/modules/attention.py:88
↓ 2 callersMethod__exit__
(self, type, value, traceback)
annotator/uniformer/mmcv/utils/timer.py:53
↓ 2 callersMethod__init__
(self, ocr_channels, scale=1, **kwargs)
annotator/uniformer/mmseg/models/decode_heads/ocr_head.py:97
↓ 2 callersMethod__init__
(self, in_channels, channels)
annotator/uniformer/mmseg/models/decode_heads/da_head.py:20
↓ 2 callersMethod__init__
(self, kernel_size, group_size, scale_factor)
annotator/uniformer/mmcv/ops/carafe.py:194
↓ 2 callersMethod__init__
(self, output_size, spatial_scale=1.0, sampling_ratio=0,
annotator/uniformer/mmcv/ops/deform_roi_pool.py:94
↓ 2 callersMethod__init__
(self, depth, num_stages=4, strides=(1, 2, 2, 2),
annotator/uniformer/mmcv/cnn/resnet.py:210
↓ 2 callersMethod__init__
(self)
ldm/modules/diffusionmodules/upscaling.py:58
↓ 2 callersFunction_allreduce_coalesced
(tensors, world_size, bucket_size_mb=-1)
annotator/uniformer/mmcv/runner/dist_utils.py:145
↓ 2 callersFunction_any
Since built-in ``any`` works only when the element of iterable is not iterable, implement the function.
annotator/uniformer/mmcv/utils/testing.py:25
↓ 2 callersMethod_build_input_conv
(self, channel, conv_cfg, norm_cfg)
annotator/uniformer/mmcv/ops/merge_cells.py:78
↓ 2 callersMethod_dist_broadcast_coalesced
(self, tensors, buffer_size)
annotator/uniformer/mmcv/parallel/distributed_deprecated.py:29
↓ 2 callersMethod_do_evaluate
perform evaluation and save ckpt.
annotator/uniformer/mmcv/runner/hooks/evaluation.py:269
↓ 2 callersMethod_dump_log
(self, log_dict, runner)
annotator/uniformer/mmcv/runner/hooks/logger/text.py:185
↓ 2 callersMethod_freeze_stages
Freeze stages param and norm stats.
annotator/uniformer/mmseg/models/backbones/resnet.py:581
↓ 2 callersMethod_get_coarse_point_feats
Sample from fine grained features. Args: prev_output (list[Tensor]): Prediction of previous decode head. points (Tens
annotator/uniformer/mmseg/models/decode_heads/point_head.py:147
↓ 2 callersMethod_get_fine_grained_point_feats
Sample from fine grained features. Args: x (list[Tensor]): Feature pyramid from by neck or backbone. points (Tensor):
annotator/uniformer/mmseg/models/decode_heads/point_head.py:123
↓ 2 callersFunction_get_mmcv_home
()
annotator/uniformer/mmcv/runner/checkpoint.py:30
↓ 2 callersFunction_get_mmcv_home
()
annotator/uniformer/mmcv_custom/checkpoint.py:30
↓ 2 callersMethod_get_real_position
(self)
annotator/uniformer/mmcv/video/io.py:124
↓ 2 callersMethod_get_rows_from_list
(self, samples)
ldm/models/diffusion/ddpm.py:469
↓ 2 callersMethod_get_weight
(self, weight)
annotator/uniformer/mmcv/cnn/bricks/conv_ws.py:101
↓ 2 callersMethod_init_rule
Initialize rule, key_indicator, comparison_func, and best score. Here is the rule to determine which rule is used for key indicator w
annotator/uniformer/mmcv/runner/hooks/evaluation.py:153
↓ 2 callersFunction_initialize
(module, cfg, wholemodule=False)
annotator/uniformer/mmcv/cnn/utils/weight_init.py:511
↓ 2 callersFunction_jpegflag
(flag='color', channel_order='bgr')
annotator/uniformer/mmcv/image/io.py:69
↓ 2 callersMethod_load_from_state_dict
Override default load function. AWS overrides the function _load_from_state_dict to recover weight_gamma and weight_beta if they are
annotator/uniformer/mmcv/cnn/bricks/conv_ws.py:114
↓ 2 callersMethod_merge_a_into_b
merge dict ``a`` into dict ``b`` (non-inplace). Values in ``a`` will overwrite ``b``. ``b`` is copied first to avoid in-place modific
annotator/uniformer/mmcv/utils/config.py:274
↓ 2 callersFunction_pillow2array
Convert a pillow image to numpy array. Args: img (:obj:`PIL.Image.Image`): The image loaded using PIL flag (str): Flags specifyin
annotator/uniformer/mmcv/image/io.py:85
↓ 2 callersMethod_register_backend
(cls, name, backend, force=False, prefixes=None)
annotator/uniformer/mmcv/fileio/file_client.py:886
↓ 2 callersMethod_register_scheme
(cls, prefixes, loader, force=False)
annotator/uniformer/mmcv/runner/checkpoint.py:168
↓ 2 callersMethod_resize
(self, x, size)
annotator/uniformer/mmcv/ops/merge_cells.py:92
↓ 2 callersMethod_save_checkpoint
Save the current checkpoint and delete unwanted checkpoint.
annotator/uniformer/mmcv/runner/hooks/checkpoint.py:119
↓ 2 callersMethod_save_ckpt
Save the best checkpoint. It will compare the score according to the compare function, write related information (best score, best ch
annotator/uniformer/mmcv/runner/hooks/evaluation.py:314
↓ 2 callersFunction_scale_size
Rescale a size by a ratio. Args: size (tuple[int]): (w, h). scale (float | tuple(float)): Scaling factor. Returns: t
annotator/uniformer/mmcv/image/geometric.py:16
↓ 2 callersMethod_should_evaluate
Judge whether to perform evaluation. Here is the rule to judge whether to perform evaluation: 1. It will not perform evaluation durin
annotator/uniformer/mmcv/runner/hooks/evaluation.py:279
↓ 2 callersMethod_swap_ema_parameters
Swap the parameter of model with parameter in ema_buffer.
annotator/uniformer/mmcv/runner/hooks/ema.py:83
↓ 2 callersMethod_validation_step
(self, batch, batch_idx, postfix="")
ldm/models/autoencoder.py:136
↓ 2 callersFunctionadd_Gaussian_noise
(img, noise_level1=2, noise_level2=25)
ldm/modules/image_degradation/bsrgan_light.py:372
↓ 2 callersFunctionadd_Poisson_noise
(img)
ldm/modules/image_degradation/bsrgan.py:404
↓ 2 callersFunctionadd_resize
(img, sf=4)
ldm/modules/image_degradation/bsrgan.py:339
↓ 2 callersFunctionadd_speckle_noise
(img, noise_level1=2, noise_level2=25)
ldm/modules/image_degradation/bsrgan.py:386
↓ 2 callersMethodafter_epoch
(self, runner)
annotator/uniformer/mmcv/runner/hooks/hook.py:22
↓ 2 callersMethodafter_iter
(self, runner)
annotator/uniformer/mmcv/runner/hooks/hook.py:28
↓ 2 callersFunctionallreduce_grads
Allreduce gradients. Args: params (list[torch.Parameters]): List of parameters of a model coalesce (bool, optional): Whether allr
annotator/uniformer/mmcv/runner/dist_utils.py:121
↓ 2 callersFunctionbbox_clip
Clip bboxes to fit the image shape. Args: bboxes (ndarray): Shape (..., 4*k) img_shape (tuple[int]): (height, width) of the image
annotator/uniformer/mmcv/image/geometric.py:342
↓ 2 callersMethodbefore_epoch
(self, runner)
annotator/uniformer/mmcv/runner/hooks/hook.py:19
↓ 2 callersMethodbefore_iter
(self, runner)
annotator/uniformer/mmcv/runner/hooks/hook.py:25
↓ 2 callersFunctionbgr2gray
Convert a BGR image to grayscale image. Args: img (ndarray): The input image. keepdim (bool): If False (by default), then return
annotator/uniformer/mmcv/image/colorspace.py:22
↓ 2 callersFunctionbicubic_degradation
Args: x: HxWxC image, [0, 1] sf: down-scale factor Return: bicubicly downsampled LR image
ldm/modules/image_degradation/bsrgan_light.py:227
↓ 2 callersFunctionbicubic_degradation
Args: x: HxWxC image, [0, 1] sf: down-scale factor Return: bicubicly downsampled LR image
ldm/modules/image_degradation/bsrgan.py:228
↓ 2 callersFunctionbuild_dataloader
Build PyTorch DataLoader. In distributed training, each GPU/process has a dataloader. In non-distributed training, there is only one dataload
annotator/uniformer/mmseg/datasets/builder.py:78
↓ 2 callersFunctionbuild_dataset
Build datasets.
annotator/uniformer/mmseg/datasets/builder.py:61
↓ 2 callersFunctionbuild_dropout
Builder for drop out layers.
annotator/uniformer/mmcv/cnn/bricks/drop.py:63
← previousnext →201–300 of 2,265, ranked by callers