MCPcopy Create free account

hub / github.com/ali-vilab/videocomposer / functions

Functions561 in github.com/ali-vilab/videocomposer

⨍ Functions 561◇ Types & classes 102
Functionpidinet_bsd_small
(pretrained=False, vanilla_cnn=True)
tools/annotator/sketch/pidinet.py:719
Functionpidinet_bsd_tiny
(pretrained=False, vanilla_cnn=True)
tools/annotator/sketch/pidinet.py:706
Functionpidinet_multicue
(pretrained=False, vanilla_cnn=True)
tools/annotator/sketch/pidinet.py:761
Functionpidinet_nyud
(pretrained=False, vanilla_cnn=True)
tools/annotator/sketch/pidinet.py:748
Methodplms_sample_loop
(self, noise, model, model_kwargs={}, clamp=None, percentile=None, condition_fn=None, guide_scale=None, plms_t
artist/ops/diffusion.py:335
Methodplms_sample_loop
(self, noise, model, model_kwargs={}, clamp=None, percentile=None, condition_fn=None, guide_scale=None, plms_t
artist/ops/diffusion.py:789
Functionpre_caption
(caption, max_words)
tools/videocomposer/datasets.py:30
Functionprint_pass
(*objects, sep=" ", end="\n", file=sys.stdout, flush=False)
utils/logging.py:23
Functionput_object
(bucket, oss_key, data, retry=5)
artist/ops/utils.py:194
Functionput_object_from_file
(bucket, oss_key, filename, retry=5)
artist/ops/utils.py:218
Functionput_torch_object
(bucket, oss_key, data, retry=5)
artist/ops/utils.py:205
Methodquantize
(self, x, *args, **kwargs)
tools/videocomposer/autoencoder.py:623
Methodquantize_image
(self, rgb_img)
tools/annotator/histogram/palette.py:100
Functionrandom_crop
(lq, hq, sf=4, lq_patchsize=64)
artist/ops/degration.py:795
Functionrandom_resize
(img, size)
gen_sketch.py:17
Functionrandom_resize
(img, size)
tools/videocomposer/inference_single.py:180
Functionrandom_resize
(img, size)
tools/videocomposer/datasets.py:52
Functionrandom_resize
(img, size)
tools/videocomposer/inference_multi.py:180
Functionread_gzip
(filename, retry=5)
artist/ops/utils.py:157
Functionread_image
(filename, retry=5)
artist/ops/utils.py:132
Functionread_tfs
(tfs, retry=5)
artist/ops/utils.py:721
Methodrecur
(key, elem)
utils/config.py:230
Functionrecv
(tensor, src=None, group=None, **kwargs)
artist/ops/distributed.py:254
Functionreduce
(tensor, dst, op=dist.ReduceOp.SUM, group=None, **kwargs)
artist/ops/distributed.py:71
Functionreduce_dict
(input_dict, group=None, reduction='mean', **kwargs)
artist/ops/distributed.py:120
Functionreduce_scatter
(output, input_list, op=dist.ReduceOp.SUM, group=None, **kwargs)
artist/ops/distributed.py:245
Functionresize_image
(input_image, resolution)
tools/annotator/util.py:26
Functionrgb2ycbcr
same as matlab rgb2ycbcr only_y: only return Y channel Input: uint8, [0, 255] float, [0, 1]
artist/ops/degration.py:52
Functionrope
r"""Apply rotary position embedding on x of shape [B, *(spatial dimensions), C].
artist/ops/utils.py:739
Functionsave_caps
(bucket, oss_key, caps, retry=5)
artist/ops/utils.py:589
Functionsave_image
(bucket, oss_key, tensor, nrow=8, normalize=True, range=(-1, 1), retry=5)
artist/ops/utils.py:260
Functionsave_text
(bucket, oss_key, tensor, nrow=8, retry=5)
artist/ops/utils.py:565
Functionsave_video
(bucket, oss_key, tensor, mean=[0.5,0.5,0.5],std=[0.5,0.5,0.5],nrow=8, retry=5)
artist/ops/utils.py:287
Functionsave_video_grid_mp4
(bucket, oss_key, tensor, mean=[0.5,0.5,0.5],std=[0.5,0.5,0.5], nrow=None, fps=5, retry=5)
artist/ops/utils.py:524
Functionsave_video_multiple_conditions
(oss_key, video_tensor, model_kwargs, source_imgs, palette, mean=[0.5, 0.5, 0.5], std=[0
artist/ops/utils.py:312
Functionsave_video_multiple_conditions_with_data
(bucket, video_save_key, gt_video_save_key, vis_oss_key, video_tensor, model_kwargs, source_imgs, palette,
artist/ops/utils.py:382
Functionsave_video_vs_conditions
(bucket, oss_key, video_tensor, conditions, source_imgs, mean=[0.5,0.5,0.5],std=[0.5,0.5,0.5],nrow=8, retry=5)
artist/ops/utils.py:484
Functionscatter
r"""NOTE: only supports CPU tensor communication.
artist/ops/distributed.py:239
Functionsend
(tensor, dst, group=None, **kwargs)
artist/ops/distributed.py:249
Methodserver
(self)
artist/ops/utils.py:711
Functionsetup_logging
Sets up the logging for multiple processes. Only enable the logging for the master process, and suppress logging for the non-master processes
utils/logging.py:34
Functionsingle2uint16
(img)
artist/ops/degration.py:49
Functionsinkhorn
(Q, eps=0.5, num_iters=3)
artist/ops/distributed.py:405
Functionsketch_simplification_mse
(pretrained=False)
tools/annotator/sketch/sketch_simplification.py:89
Functionsketch_to_pencil_v1
(pretrained=False)
tools/annotator/sketch/sketch_simplification.py:97
Functionsketch_to_pencil_v2
(pretrained=False)
tools/annotator/sketch/sketch_simplification.py:105
Functionspherical_kmeans
(feats, num_clusters, num_iters=10)
artist/ops/distributed.py:367
Functionsrmd_degradation
blur + bicubic downsampling Args: x: HxWxC image, [0, 1] k: hxw, double sf: down-scale factor Return: d
artist/ops/degration.py:630
Methodstep
Performs a single optimization step Arguments: closure (callable, optional): A closure that reevaluates the model
artist/optim/adafactor.py:140
Methodsymbolic
(graph, input)
artist/ops/distributed.py:303
Methodsymbolic
(graph, input)
artist/ops/distributed.py:318
Methodsymbolic
(graph, input)
artist/ops/distributed.py:333
Methodsymbolic
(graph, input)
artist/ops/distributed.py:348
Functionsynchronize
Helper function to synchronize (barrier) among all processes when using distributed training
utils/distributed.py:126
Functionto_device
(batch, device, non_blocking=False)
artist/ops/utils.py:181
Functionto_fp16
(m)
artist/models/clip.py:11
Functionuint162single
(img)
artist/ops/degration.py:46
Functionunzip
(filename, dst_dir=None)
artist/ops/utils.py:650
Methodvariational_lower_bound_loop
r"""Compute the entire variational lower bound, measured in bits-per-dim.
artist/ops/diffusion.py:441
Methodvariational_lower_bound_loop
r"""Compute the entire variational lower bound, measured in bits-per-dim.
artist/ops/diffusion.py:972
Functionycbcr2rgb
same as matlab ycbcr2rgb Input: uint8, [0, 255] float, [0, 1]
artist/ops/degration.py:75
← previous501–561 of 561, ranked by callers