MCPcopy Create free account

hub / github.com/Picsart-AI-Research/Text2Video-Zero / functions

Functions1,788 in github.com/Picsart-AI-Research/Text2Video-Zero

↓ 1 callersMethodregister_timer_hook
(self, timer_config)
annotator/uniformer/mmcv/runner/base_runner.py:470
↓ 1 callersFunctionregister_torch_optimizers
()
annotator/uniformer/mmcv/runner/optimizer/builder.py:13
↓ 1 callersMethodregister_training_hooks
Register default and custom hooks for training. Default and custom hooks include: +----------------------+-------------------------+
annotator/uniformer/mmcv/runner/base_runner.py:503
↓ 1 callersFunctionrel_roi_point_to_abs_img_point
Convert roi based relative point coordinates to image based absolute point coordinates. Args: rois (Tensor): RoIs or BBoxes, shape (N
annotator/uniformer/mmcv/ops/point_sample.py:137
↓ 1 callersFunctionremove_batch_counter_hook_function
(module)
annotator/uniformer/mmcv/cnn/utils/flops_counter.py:524
↓ 1 callersMethodreset_parameters
(self)
annotator/uniformer/mmcv/ops/deform_conv.py:269
↓ 1 callersMethodreset_parameters
(self)
annotator/uniformer/mmcv/ops/sync_bn.py:236
↓ 1 callersMethodreset_parameters
(self)
annotator/uniformer/mmcv/cnn/bricks/context_block.py:75
↓ 1 callersMethodreset_running_stats
(self)
annotator/uniformer/mmcv/ops/sync_bn.py:230
↓ 1 callersMethodresults2img
Write the segmentation results to images. Args: results (list[list | tuple | ndarray]): Testing results of the da
annotator/uniformer/mmseg/datasets/cityscapes.py:50
↓ 1 callersMethodrun
(self, data_loaders, workflow, **kwargs)
annotator/uniformer/mmcv/runner/base_runner.py:193
↓ 1 callersMethodsample
Sample pixels that have high loss or with low prediction confidence. Args: seg_logit (torch.Tensor): segmentation logits, shape (
annotator/uniformer/mmseg/core/seg/sampler/ohem_pixel_sampler.py:30
↓ 1 callersMethodsaturation
Saturation distortion.
annotator/uniformer/mmseg/datasets/pipelines/transforms.py:827
↓ 1 callersMethodscaled_l2
(x, codewords, scale)
annotator/uniformer/mmseg/ops/encoding.py:33
↓ 1 callersFunctionscatter_map
(obj)
annotator/uniformer/mmcv/parallel/scatter_gather.py:16
↓ 1 callersFunctionshow_result_pyplot
Visualize the segmentation results on the image. Args: model (nn.Module): The loaded segmentor. img (str or np.ndarray): Image fi
annotator/uniformer/mmseg/apis/inference.py:101
↓ 1 callersMethodsimple_test
Placeholder for single image test.
annotator/uniformer/mmseg/models/segmentors/base.py:56
↓ 1 callersMethodsince_start
Total time since the timer is started. Returns (float): Time in seconds.
annotator/uniformer/mmcv/utils/timer.py:64
↓ 1 callersMethodslide_inference
Inference by sliding-window with overlap. If h_crop > h_img or w_crop > w_img, the small patch will be used to decode without padding
annotator/uniformer/mmseg/models/segmentors/encoder_decoder.py:169
↓ 1 callersMethodspatial_pool
(self, x)
annotator/uniformer/mmcv/cnn/bricks/context_block.py:85
↓ 1 callersMethodsplit_scope_key
Split scope and key. The first scope will be split from key. Examples: >>> Registry.split_scope_key('mmdet.ResNet')
annotator/uniformer/mmcv/utils/registry.py:149
↓ 1 callersFunctionsynchronize_stream
(output, devices, streams)
annotator/uniformer/mmcv/parallel/_functions.py:34
↓ 1 callersFunctiontensor2imgs
Convert tensor to 3-channel images. Args: tensor (torch.Tensor): Tensor that contains multiple images, shape ( N, C, H, W).
annotator/uniformer/mmcv/image/misc.py:12
↓ 1 callersFunctiontotal_intersect_and_union
Calculate Total Intersection and Union. Args: results (list[ndarray] | list[str]): List of prediction segmentation maps or li
annotator/uniformer/mmseg/core/evaluation/metrics.py:88
↓ 1 callersFunctiontrunc_normal_
r"""Fills the input Tensor with values drawn from a truncated normal distribution. The values are effectively drawn from the normal distributi
annotator/uniformer/mmcv/cnn/utils/weight_init.py:662
↓ 1 callersFunctionupfirdn2d_native
(input, kernel, up_x, up_y, down_x, down_y, pad_x0, pad_x1, pad_y0, pad_y1)
annotator/uniformer/mmcv/ops/upfirdn2d.py:290
↓ 1 callersFunctionweights_to_cpu
Copy a model state_dict to cpu. Args: state_dict (OrderedDict): Model weights on GPU. Returns: OrderedDict: Model weights on
annotator/uniformer/mmcv/runner/checkpoint.py:553
↓ 1 callersFunctionweights_to_cpu
Copy a model state_dict to cpu. Args: state_dict (OrderedDict): Model weights on GPU. Returns: OrderedDict: Model weights on
annotator/uniformer/mmcv_custom/checkpoint.py:359
↓ 1 callersMethodwhole_inference
Inference with full image.
annotator/uniformer/mmseg/models/segmentors/encoder_decoder.py:214
↓ 1 callersFunctionwindow_partition
Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)
annotator/uniformer/mmseg/models/backbones/uniformer.py:138
↓ 1 callersFunctionwindow_reverse
Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image
annotator/uniformer/mmseg/models/backbones/uniformer.py:152
↓ 1 callersFunctionwrite_pfm
Write pfm file. Args: path (str): pathto file image (array): data scale (int, optional): Scale. Defaults to 1.
annotator/midas/utils.py:58
Method__call__
( self, attn, hidden_states, encoder_hidden_states=None,
utils.py:184
Method__call__
( self, prompt: Union[str, List[str]], video_length: Optional[int], height: Op
text_to_video_pipeline.py:225
Method__call__
(self, img, low_threshold, high_threshold)
annotator/canny/__init__.py:5
Method__call__
(self, img)
annotator/uniformer/__init__.py:20
Method__call__
Call function to load images into results. Args: results (dict): A result dict contains the file name of the imag
annotator/uniformer/mmseg/apis/inference.py:45
Method__call__
Call function to apply test time augment transforms on results. Args: results (dict): Result dict contains the data to transform.
annotator/uniformer/mmseg/datasets/pipelines/test_time_aug.py:93
Method__call__
Call function to resize images, bounding boxes, masks, semantic segmentation map. Args: results (dict): Result dict from
annotator/uniformer/mmseg/datasets/pipelines/transforms.py:210
Method__call__
Call function to flip bounding boxes, masks, semantic segmentation maps. Args: results (dict): Result dict from loading p
annotator/uniformer/mmseg/datasets/pipelines/transforms.py:259
Method__call__
Call function to pad images, masks, semantic segmentation maps. Args: results (dict): Result dict from loading pipeline.
annotator/uniformer/mmseg/datasets/pipelines/transforms.py:342
Method__call__
Call function to normalize images. Args: results (dict): Result dict from loading pipeline. Returns: dict: N
annotator/uniformer/mmseg/datasets/pipelines/transforms.py:381
Method__call__
Call function to rerange images. Args: results (dict): Result dict from loading pipeline. Returns: dict: Rera
annotator/uniformer/mmseg/datasets/pipelines/transforms.py:423
Method__call__
Call function to Use CLAHE method process images. Args: results (dict): Result dict from loading pipeline. Returns:
annotator/uniformer/mmseg/datasets/pipelines/transforms.py:472
Method__call__
Call function to randomly crop images, semantic segmentation maps. Args: results (dict): Result dict from loading pipeline.
annotator/uniformer/mmseg/datasets/pipelines/transforms.py:529
Method__call__
Call function to rotate image, semantic segmentation maps. Args: results (dict): Result dict from loading pipeline. Retu
annotator/uniformer/mmseg/datasets/pipelines/transforms.py:609
Method__call__
Call function to convert RGB image to grayscale image. Args: results (dict): Result dict from loading pipeline. Returns:
annotator/uniformer/mmseg/datasets/pipelines/transforms.py:676
Method__call__
Call function to process the image with gamma correction. Args: results (dict): Result dict from loading pipeline. Retur
annotator/uniformer/mmseg/datasets/pipelines/transforms.py:724
Method__call__
Call function to scale the semantic segmentation map. Args: results (dict): Result dict from loading pipeline. Returns:
annotator/uniformer/mmseg/datasets/pipelines/transforms.py:754
Method__call__
Call function to perform photometric distortion on images. Args: results (dict): Result dict from loading pipeline. Retu
annotator/uniformer/mmseg/datasets/pipelines/transforms.py:848
Method__call__
Call functions to load image and get image meta information. Args: results (dict): Result dict from :obj:`mmseg.CustomDataset`.
annotator/uniformer/mmseg/datasets/pipelines/loading.py:42
Method__call__
Call function to load multiple types annotations. Args: results (dict): Result dict from :obj:`mmseg.CustomDataset`. Ret
annotator/uniformer/mmseg/datasets/pipelines/loading.py:113
Method__call__
Call function to apply transforms sequentially. Args: data (dict): A result dict contains the data to transform. Returns
annotator/uniformer/mmseg/datasets/pipelines/compose.py:29
Method__call__
Call function to convert data in results to :obj:`torch.Tensor`. Args: results (dict): Result dict contains the data to convert.
annotator/uniformer/mmseg/datasets/pipelines/formating.py:47
Method__call__
Call function to convert image in results to :obj:`torch.Tensor` and transpose the channel order. Args: results (dict): R
annotator/uniformer/mmseg/datasets/pipelines/formating.py:81
Method__call__
Call function to convert image in results to :obj:`torch.Tensor` and transpose the channel order. Args: results (dict): R
annotator/uniformer/mmseg/datasets/pipelines/formating.py:117
Method__call__
Call function to convert data in results to :obj:`mmcv.DataContainer`. Args: results (dict): Result dict contains the dat
annotator/uniformer/mmseg/datasets/pipelines/formating.py:155
Method__call__
Call function to transform and format common fields in results. Args: results (dict): Result dict contains the data to convert.
annotator/uniformer/mmseg/datasets/pipelines/formating.py:189
Method__call__
Call function to collect keys in results. The keys in ``meta_keys`` will be converted to :obj:mmcv.DataContainer. Args: r
annotator/uniformer/mmseg/datasets/pipelines/formating.py:264
Method__call__
(self, parser, namespace, values, option_string=None)
annotator/uniformer/mmcv/utils/config.py:683
Method__call__
(self)
annotator/uniformer/mmcv/runner/default_constructor.py:43
Method__call__
(self, model)
annotator/uniformer/mmcv/runner/optimizer/default_constructor.py:234
Method__call__
(self, module)
annotator/uniformer/mmcv/cnn/utils/weight_init.py:179
Method__call__
(self, module)
annotator/uniformer/mmcv/cnn/utils/weight_init.py:265
Method__call__
(self, module: nn.Module)
annotator/uniformer/mmcv/cnn/utils/weight_init.py:318
Method__call__
(self, module)
annotator/uniformer/mmcv/cnn/utils/weight_init.py:363
Method__call__
(self, module)
annotator/uniformer/mmcv/cnn/utils/weight_init.py:423
Method__call__
(self, module)
annotator/uniformer/mmcv/cnn/utils/weight_init.py:459
Method__call__
(self, module)
annotator/uniformer/mmcv/cnn/utils/weight_init.py:483
Method__call__
(self, oriImg)
annotator/openpose/hand.py:25
Method__call__
(self, oriImg)
annotator/openpose/body.py:24
Method__call__
(self, oriImg, hand=False)
annotator/openpose/__init__.py:29
Method__call__
(self, input_image, a=np.pi * 2.0, bg_th=0.1)
annotator/midas/__init__.py:13
Method__call__
(self, sample)
annotator/midas/midas/transforms.py:162
Method__call__
(self, sample)
annotator/midas/midas/transforms.py:205
Method__call__
(self, sample)
annotator/midas/midas/transforms.py:218
Method__contains__
(self, key)
annotator/uniformer/mmcv/utils/registry.py:115
Method__enter__
(self)
annotator/uniformer/mmcv/video/io.py:269
Method__exit__
(self, exc_type, exc_value, traceback)
annotator/uniformer/mmcv/video/io.py:272
Method__getattr__
(self, name)
annotator/uniformer/mmcv/utils/config.py:38
Method__getitem__
Get training/test data after pipeline. Args: idx (int): Index of data. Returns: dict: Training/test data (wi
annotator/uniformer/mmseg/datasets/custom.py:178
Method__getitem__
Get item from original dataset.
annotator/uniformer/mmseg/datasets/dataset_wrappers.py:44
Method__getitem__
(self, name)
annotator/uniformer/mmcv/utils/config.py:509
Method__getitem__
(self, index)
annotator/uniformer/mmcv/video/io.py:243
Method__getstate__
(self)
annotator/uniformer/mmcv/utils/config.py:525
Method__init__
(self, unet_chunk_size=2)
utils.py:181
Method__init__
( self, vae: AutoencoderKL, text_encoder: CLIPTextModel, tokenizer: CLIPTokeni
text_to_video_pipeline.py:36
Method__init__
(self, device, dtype, **kwargs)
model.py:27
Method__init__
(self)
annotator/uniformer/__init__.py:12
Method__init__
(self, channels, num_codes)
annotator/uniformer/mmseg/ops/encoding.py:17
Method__init__
(self, size=None, scale_factor=None, mode='nearest',
annotator/uniformer/mmseg/ops/wrappers.py:31
Method__init__
(self, context, thresh=None, min_kept=100000)
annotator/uniformer/mmseg/core/seg/sampler/ohem_pixel_sampler.py:23
Method__init__
(self, **kwargs)
annotator/uniformer/mmseg/core/seg/sampler/base_pixel_sampler.py:7
Method__init__
(self, *args, by_epoch=False, efficient_test=False, **kwargs)
annotator/uniformer/mmseg/core/evaluation/eval_hooks.py:70
Method__init__
(self, split, **kwargs)
annotator/uniformer/mmseg/datasets/pascal_context.py:96
Method__init__
(self, **kwargs)
annotator/uniformer/mmseg/datasets/drive.py:21
Method__init__
(self, pipeline, img_dir, img_suffix='.jpg',
annotator/uniformer/mmseg/datasets/custom.py:75
Method__init__
(self, **kwargs)
annotator/uniformer/mmseg/datasets/hrf.py:21
Method__init__
(self, datasets)
annotator/uniformer/mmseg/datasets/dataset_wrappers.py:17
Method__init__
(self, dataset, times)
annotator/uniformer/mmseg/datasets/dataset_wrappers.py:37
← previousnext →601–700 of 1,788, ranked by callers