MCPcopy Create free account

hub / github.com/One-2-3-45/One-2-3-45 / functions

Functions1,158 in github.com/One-2-3-45/One-2-3-45

↓ 4 callersFunctionzero_module
Zero out the parameters of a module and return it.
ldm/modules/diffusionmodules/util.py:174
↓ 3 callersMethod__init__
(self)
reconstruction/loss/color_loss.py:7
↓ 3 callersMethod__init__
(self, factor, output_size, initial_size=None, image_key="jpg")
ldm/data/laion.py:225
↓ 3 callersMethod__init__
(self, ddconfig, lossconfig, n_embed, embe
ldm/models/autoencoder.py:15
↓ 3 callersMethod_make_layer
(self, block, dim, stride=1)
elevation_estimate/loftr/backbone/resnet_fpn.py:92
↓ 3 callersFunctionadd_blur
(img, sf=4)
ldm/modules/image_degradation/bsrgan_light.py:325
↓ 3 callersFunctioncheckpoint
Evaluate a function without caching intermediate activations, allowing for reduced memory at the expense of extra compute in the backward pas
ldm/modules/diffusionmodules/util.py:102
↓ 3 callersFunctioncount_params
(model, verbose=False)
ldm/util.py:124
↓ 3 callersFunctioncreate_window
(window_size, channel)
ldm/modules/evaluate/ssim.py:22
↓ 3 callersFunctiondefault
(val, d)
ldm/modules/attention.py:19
↓ 3 callersMethodema_scope
(self, context=None)
ldm/models/autoencoder.py:64
↓ 3 callersMethodencode_image
:param raw_image (H, W, 3) array of uint8 in [0, 255].
demo/app.py:123
↓ 3 callersFunctionexists
(val)
ldm/modules/attention.py:11
↓ 3 callersMethodextract_depth_maps
(self, rays_o, rays_d, near, far, sdf_network, conditional_volume)
reconstruction/models/fast_renderer.py:23
↓ 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:210
↓ 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 callersFunctiongen_random_rays_from_single_image
generate random rays in world space, for a single image :param H: :param W: :param N_rays: :param image: [3, H, W] :param int
reconstruction/models/rays.py:57
↓ 3 callersMethodget_fold_unfold
:param x: img of size (bs, c, h, w) :return: n img crops of size (n, bs, c, kernel_size[0], kernel_size[1])
ldm/models/diffusion/ddpm.py:669
↓ 3 callersMethodget_input
(self, batch, k)
ldm/models/autoencoder.py:124
↓ 3 callersMethodget_input
(self, batch, k)
ldm/models/autoencoder.py:344
↓ 3 callersMethodget_loss
(self, pred, target, mean=True)
ldm/models/diffusion/ddpm.py:329
↓ 3 callersFunctionget_shifts
(x)
reconstruction/ops/grid_sampler.py:359
↓ 3 callersFunctionget_weight
(s, a=-0.5)
reconstruction/ops/grid_sampler.py:220
↓ 3 callersMethodget_weighting
(self, h, w, Ly, Lx, device)
ldm/models/diffusion/ddpm.py:653
↓ 3 callersMethodmake_loader
(self, dataset_config, train=True)
ldm/data/laion.py:125
↓ 3 callersFunctionmake_tranforms
(image_transforms)
ldm/data/simple.py:103
↓ 3 callersMethodmodify_score
(self, model, e_t, x, t, c)
ldm/guidance.py:57
↓ 3 callersFunctionnormalize
(vectors)
elevation_estimate/utils/utils3d.py:38
↓ 3 callersFunctionnormalize
(vectors)
utils/utils.py:82
↓ 3 callersFunctionperceptual_sim
(img1, img2, vgg16)
ldm/modules/evaluate/evaluate_perceptualsim.py:328
↓ 3 callersMethodpredict_start_from_noise
(self, x_t, t, noise)
ldm/models/diffusion/ddpm.py:266
↓ 3 callersFunctionpsnr
(img1, img2, mask=None,reshape=False)
ldm/modules/evaluate/evaluate_perceptualsim.py:304
↓ 3 callersFunctionsample_model_batch
(model, sampler, input_im, xs, ys, n_samples=4, precision='autocast', ddim_eta=1.0, ddim_steps=75, scale=3.0,
utils/zero123_utils.py:60
↓ 3 callersMethodshared_step
(self, batch)
ldm/models/diffusion/ddpm.py:387
↓ 3 callersFunctionsparse_to_dense_volume
convert the sparse volume into dense volume to enable trilinear sampling to save GPU memory; :param coords: [num_pts, 3] :param featu
reconstruction/models/sparse_sdf_network.py:502
↓ 3 callersFunctionssim
(img1, img2)
ldm/modules/image_degradation/utils_image.py:669
↓ 3 callersFunctionssim_metric
(img1, img2, mask=None)
ldm/modules/evaluate/evaluate_perceptualsim.py:299
↓ 3 callersMethodto_rgb
(self, x)
ldm/models/autoencoder.py:255
↓ 3 callersMethodupsample
:param pre_feat: (Tensor), features from last level, (N, C) :param pre_coords: (Tensor), coordinates from last level, (N, 4) (4 : Ba
reconstruction/models/sparse_sdf_network.py:198
↓ 3 callersFunctionvisualize_depth_numpy
depth: (H, W)
reconstruction/utils/misc_utils.py:28
↓ 2 callersFunctionNormalize
(in_channels)
ldm/modules/attention.py:76
↓ 2 callersMethod__init__
(self, in_planes, planes, stride=1)
elevation_estimate/loftr/backbone/resnet_fpn.py:16
↓ 2 callersMethod__init__
(self, in_channels, out_channels, kernel_size=3, stride=1, pad=1, norm_act=I
reconstruction/models/featurenet.py:13
↓ 2 callersMethod__init__
Returns only captions with dummy images
ldm/data/simple.py:474
↓ 2 callersMethod__init__
(self, channels, reduction)
ldm/thirdp/psp/helpers.py:59
↓ 2 callersFunction_batch_pairwise_distances
Compute pairwise distances between two batches of feature vectors.
ldm/modules/evaluate/adm_evaluator.py:436
↓ 2 callersFunction_download_inception_model
()
ldm/modules/evaluate/adm_evaluator.py:595
↓ 2 callersMethod_get_rows_from_list
(self, samples)
ldm/models/diffusion/ddpm.py:429
↓ 2 callersMethod_load_im
(self, filename)
ldm/data/simple.py:164
↓ 2 callersFunction_ssim
( img1, img2, window, window_size, channel, mask=None, size_average=True )
ldm/modules/evaluate/ssim.py:31
↓ 2 callersMethod_upsample_add
(self, x, y)
reconstruction/tsparse/modules.py:70
↓ 2 callersMethod_upsample_add
(self, x, y)
reconstruction/models/featurenet.py:70
↓ 2 callersMethod_validation_step
(self, batch, batch_idx, suffix="")
ldm/models/autoencoder.py:170
↓ 2 callersFunctionadd_Gaussian_noise
(img, noise_level1=2, noise_level2=25)
ldm/modules/image_degradation/bsrgan_light.py:373
↓ 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 callersMethodadjust_learning_rate
(self)
reconstruction/exp_runner_generic_blender_train.py:400
↓ 2 callersMethodadjust_learning_rate
(self)
reconstruction/exp_runner_generic_blender_val.py:405
↓ 2 callersFunctionalways
(val)
ldm/modules/x_transformer.py:64
↓ 2 callersFunctionback_project_sparse_type
Unproject the image fetures to form a 3D (sparse) feature volume :param coords: coordinates of voxels, dim: (num of voxels, 4) (4 : batc
reconstruction/ops/back_project.py:5
↓ 2 callersMethodbackward
(ctx, *output_grads)
ldm/modules/diffusionmodules/util.py:131
↓ 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:228
↓ 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 callersFunctioncam2pixel
Transform coordinates in the camera frame to the pixel frame. Args: cam_coords: pixel coordinates defined in the first camera coordinates
reconstruction/ops/back_project.py:89
↓ 2 callersMethodcartesian_to_spherical
(self, xyz)
ldm/data/simple.py:248
↓ 2 callersMethodcompute_activations
Compute image features for downstream evals. :param batches: a iterator over NHWC numpy arrays in [0, 255]. :return: a tuple
ldm/modules/evaluate/adm_evaluator.py:163
↓ 2 callersFunctioncompute_stats
(feats: np.ndarray)
ldm/modules/evaluate/torch_frechet_video_distance.py:34
↓ 2 callersMethodcompute_z_diff
compute the depth difference of query pts projected on the image and the predicted depth values of the image :param xyz: [N_rays, n_
reconstruction/models/projector.py:65
↓ 2 callersMethodcopy_to
(self, model)
ldm/modules/ema.py:46
↓ 2 callersFunctioncount_flops_attn
A counter for the `thop` package to count the operations in an attention operation. Meant to be used like: macs, params = thop.pr
ldm/modules/diffusionmodules/openaimodel.py:328
↓ 2 callersFunctioncubic
(x)
ldm/modules/image_degradation/utils_image.py:700
↓ 2 callersFunctioncubic_interpolate_batch
one dimensional cubic interpolation :param p: [B, N, 4] (4) should be in order :param x: [B, N] :return:
reconstruction/ops/grid_sampler.py:297
↓ 2 callersMethoddecode
(self, text)
ldm/modules/encoders/modules.py:146
↓ 2 callersMethoddecode
(self, z)
ldm/models/autoencoder.py:330
↓ 2 callersMethoddelta_border
:param h: height :param w: width :return: normalized distance to image border, wtith min distance = 0 at border and
ldm/models/diffusion/ddpm.py:639
↓ 2 callersMethodema_scope
(self, context=None)
ldm/models/diffusion/ddpm.py:181
↓ 2 callersMethodencode
(self, x)
ldm/models/autoencoder.py:96
↓ 2 callersFunctionestimate_elev
(root_dir)
elevation_estimate/estimate_wild_imgs.py:4
↓ 2 callersMethodevaluate_pr
Evaluate precision and recall efficiently. :param features_1: [N1 x D] feature vectors for reference batch. :param radii_1:
ldm/modules/evaluate/adm_evaluator.py:347
↓ 2 callersMethodfind_in_interval
(self, n)
ldm/lr_scheduler.py:52
↓ 2 callersFunctionfind_optim_elev
(elevs, nimgs, matches, K, dbg=False)
elevation_estimate/utils/elev_est_api.py:148
↓ 2 callersMethodforward
(self, inputs)
reconstruction/models/fields.py:74
↓ 2 callersMethodforward
(self, inputs)
reconstruction/models/fields.py:140
↓ 2 callersMethodfrechet_distance
Compute the Frechet distance between two sets of statistics.
ldm/modules/evaluate/adm_evaluator.py:93
↓ 2 callersFunctiongen_poses
(shape_dir, pose_est)
utils/utils.py:130
↓ 2 callersFunctiongen_rays_from_single_image
generate rays in world space, for image image :param H: :param W: :param intrinsics: [3,3] :param c2ws: [4,4] :return:
reconstruction/models/rays.py:11
↓ 2 callersFunctionget_boundingbox
# get the minimum bounding box of all visual hulls :param img_hw: :param intrinsics: :param extrinsics: :param near_fars: :re
reconstruction/data/scene.py:48
↓ 2 callersMethodget_conditioning
(self, batch, k=None)
ldm/models/diffusion/classifier.py:133
↓ 2 callersFunctionget_data_from_str
(input_str,nprc = None)
ldm/modules/evaluate/torch_frechet_video_distance.py:119
↓ 2 callersMethodget_input
(self, batch, k)
ldm/models/diffusion/classifier.py:124
↓ 2 callersMethodget_input
(self, batch, k, return_first_stage_outputs=False, force_c_encode=False, cond_key=None, retu
ldm/models/diffusion/ddpm.py:723
↓ 2 callersFunctionget_ray_directions
Get ray directions for all pixels in camera coordinate. Reference: https://www.scratchapixel.com/lessons/3d-basic-rendering/ r
reconstruction/data/One2345_train.py:13
↓ 2 callersMethodget_trainable_params
(self)
reconstruction/models/trainer_generic.py:133
↓ 2 callersMethodgradient
(self, x)
reconstruction/models/fields.py:98
↓ 2 callersFunctiongroup_dict_by_key
(cond, d)
ldm/modules/x_transformer.py:93
↓ 2 callersFunctiongroupby_prefix_and_trim
(prefix, d)
ldm/modules/x_transformer.py:110
↓ 2 callersFunctionimage_preprocess_nosave
(input_image, lower_contrast=True, rescale=True)
utils/utils.py:50
↓ 2 callersMethodinit_from_ckpt
(self, path, ignore_keys=list(), only_model=False)
ldm/models/diffusion/ddpm.py:196
↓ 2 callersFunctioninit_model
(device, ckpt, half_precision=False)
utils/zero123_utils.py:37
← previousnext →101–200 of 1,158, ranked by callers