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

↓ 255 callersMethodto
Same as to in torch module Don't really underestand why this isn't a module in the first place
ldm/models/diffusion/ddim.py:21
↓ 53 callersMethodregister_buffer
(self, name, attr, device=None)
ldm/models/diffusion/ddim.py:30
↓ 42 callersMethodload
(cls, path: str, arr_name: str)
ldm/modules/evaluate/adm_evaluator.py:521
↓ 22 callersFunctioninstantiate_from_config
(config)
ldm/util.py:131
↓ 20 callersMethoddecode_first_stage
(self, z, predict_cids=False, force_not_quantize=False)
ldm/models/diffusion/ddpm.py:763
↓ 17 callersFunctionextract_into_tensor
(a, t, x_shape)
ldm/modules/diffusionmodules/util.py:96
↓ 16 callersFunctionexists
(val)
ldm/modules/x_transformer.py:54
↓ 15 callersMethod__init__
(self, *, ch, out_ch, ch_mult=(1,2,4,8), num_res_blocks, attn_resolutions, dropout=0.0, resam
ldm/modules/diffusionmodules/model.py:217
↓ 15 callersMethod__init__
(self, n_embed, n_layer, vocab_size=30522, max_seq_len=77, device="cuda",use_tokenizer=True,
ldm/modules/encoders/modules.py:152
↓ 15 callersFunctionconv_nd
Create a 1D, 2D, or 3D convolution module.
ldm/modules/diffusionmodules/util.py:218
↓ 15 callersMethodmeshgrid
(self, h, w)
ldm/models/diffusion/ddpm.py:632
↓ 14 callersMethodq_sample
(self, x_start, t, noise=None)
ldm/models/diffusion/ddpm.py:324
↓ 13 callersMethodregister_buffer
(self, name, attr)
ldm/models/diffusion/plms.py:19
↓ 13 callersMethodsample_log
(self, cond, batch_size, ddim, ddim_steps, **kwargs)
ldm/models/diffusion/ddpm.py:1227
↓ 12 callersMethod__init__
(self, value, fn)
ldm/modules/x_transformer.py:118
↓ 12 callersMethoddecode
(self, quant)
ldm/models/autoencoder.py:107
↓ 12 callersMethoddecode
(self, x_latent, cond, t_start, unconditional_guidance_scale=1.0, unconditional_conditioning=None,
ldm/models/diffusion/ddim.py:308
↓ 12 callersFunctionexists
(x)
ldm/util.py:106
↓ 12 callersFunctionget_block
(in_channel, depth, num_units, stride=2)
ldm/thirdp/psp/helpers.py:27
↓ 12 callersMethodget_input
(self, batch, k)
ldm/models/diffusion/ddpm.py:379
↓ 12 callersMethodload_state_dict
(self, state_dict, *args, **kwargs)
elevation_estimate/loftr/loftr.py:77
↓ 11 callersFunctionlog_txt_as_img
(wh, xc, size=10)
ldm/util.py:70
↓ 10 callersFunctionconv3x3
3x3 convolution with padding
elevation_estimate/loftr/backbone/resnet_fpn.py:10
↓ 10 callersFunctioncubic_interpolate
one dimensional cubic interpolation :param p: [N, 4] (4) should be in order :param x: [N] :return:
reconstruction/ops/grid_sampler.py:241
↓ 10 callersMethodget_conditional_volume
:param feature_maps: pyramid features (B,V,C0+C1+C2,H,W) fused pyramid features :param partial_vol_origin: [B, 3] the world coordin
reconstruction/models/sparse_sdf_network.py:286
↓ 10 callersFunctionnonlinearity
(x)
ldm/modules/diffusionmodules/model.py:33
↓ 10 callersMethodobtain_pyramid_feature_maps
get feature maps of all conditional images :param imgs: :return:
reconstruction/models/trainer_generic.py:1104
↓ 9 callersFunctionNormalize
(in_channels, num_groups=32)
ldm/modules/diffusionmodules/model.py:38
↓ 9 callersMethod__init__
(self, channels, use_conv, dims=2, out_channels=None, padding=1)
ldm/modules/diffusionmodules/openaimodel.py:101
↓ 9 callersFunctionsample_ptsFeatures_from_featureMaps
sample features of pts from 2d feature maps :param pts: [N_rays, N_samples, 3] :param featureMaps: [N_views, C, H, W] :param w2cs: [N
reconstruction/models/render_utils.py:88
↓ 8 callersMethod__init__
(self, **kwargs)
reconstruction/tsparse/modules.py:157
↓ 8 callersMethodapply_model
(self, x_noisy, t, cond, return_ids=False)
ldm/models/diffusion/ddpm.py:888
↓ 8 callersFunctionnormalization
Make a standard normalization layer. :param channels: number of input channels. :return: an nn.Module for normalization.
ldm/modules/diffusionmodules/util.py:199
↓ 8 callersMethodsample
(self)
ldm/modules/distributions/distributions.py:17
↓ 8 callersMethodyear
(self)
ldm/data/coco.py:95
↓ 7 callersMethod__init__
(self, unet_config, timesteps=1000, beta_schedule="linear",
ldm/models/diffusion/ddpm.py:49
↓ 7 callersFunctionconv1x1
1x1 convolution without padding
elevation_estimate/loftr/backbone/resnet_fpn.py:5
↓ 7 callersMethodencode
(self, text)
ldm/modules/encoders/modules.py:171
↓ 7 callersMethodencode_first_stage
(self, x)
ldm/models/diffusion/ddpm.py:823
↓ 7 callersMethodget_learned_conditioning
(self, c)
ldm/models/diffusion/ddpm.py:619
↓ 7 callersFunctionload_img
(img_name, size=None)
ldm/modules/evaluate/evaluate_perceptualsim.py:334
↓ 7 callersFunctionmake_attn
(in_channels, attn_type="vanilla")
ldm/modules/diffusionmodules/model.py:205
↓ 7 callersFunctionpredict_stage1_gradio
(model, raw_im, save_path = "", adjust_set=[], device="cuda", ddim_steps=75, scale=3.0)
utils/zero123_utils.py:101
↓ 7 callersMethodrender
(self, rays_o, rays_d, near, far, sdf_network, rendering_network, perturb_overwrite=-1,
reconstruction/models/sparse_neus_renderer.py:457
↓ 7 callersMethodto_rgb
(self, x)
ldm/models/diffusion/ddpm.py:1432
↓ 7 callersFunctionzero123_infer
(model, input_dir_path, start_idx=0, end_idx=12, indices=None, device="cuda", ddim_steps=75, scale=3.0)
utils/zero123_utils.py:162
↓ 6 callersMethod__init__
(self, dim_in, dim_out)
ldm/modules/attention.py:38
↓ 6 callersMethod__init__
(self, txt_file, data_root, size=None, int
ldm/data/lsun.py:10
↓ 6 callersFunctionadd_JPEG_noise
(img)
ldm/modules/image_degradation/bsrgan.py:418
↓ 6 callersFunctionadd_blur
(img, sf=4)
ldm/modules/image_degradation/bsrgan.py:325
↓ 6 callersFunctiondefault
(val, d)
ldm/util.py:110
↓ 6 callersFunctiondefault
(val, d)
ldm/modules/x_transformer.py:58
↓ 6 callersFunctiongenerate_grid
generate grid if 3D volume, grid[:,:,x,y,z] = (x,y,z) :param n_vox: :param interval: :return:
reconstruction/ops/generate_grids.py:4
↓ 6 callersMethodget_alpha_inter_ratio
(self, start, end)
reconstruction/exp_runner_generic_blender_train.py:407
↓ 6 callersMethodget_alpha_inter_ratio
(self, start, end)
reconstruction/exp_runner_generic_blender_val.py:412
↓ 6 callersFunctionismap
(x)
ldm/util.py:94
↓ 6 callersMethodsample
(self, batch_size=16, return_intermediates=False)
ldm/models/diffusion/ddpm.py:318
↓ 6 callersMethodsdf
(self, x)
reconstruction/models/fields.py:92
↓ 6 callersMethodvalidate_mesh
(self, density_or_sdf_network, func_extract_geometry, world_space=True, resolution=360,
reconstruction/models/trainer_generic.py:1272
↓ 5 callersMethod__init__
reconstruction/models/fields.py:262
↓ 5 callersMethod_get_denoise_row_from_list
(self, samples, desc='', force_no_decoder_quantization=False)
ldm/models/diffusion/ddpm.py:598
↓ 5 callersFunctiongen_large_mask
img_h: int, an image height img_w: int, an image width marg: int, a margin for a box starting coordinate p_irr: float, 0 <= p_irr <=
ldm/data/inpainting/synthetic_mask.py:85
↓ 5 callersFunctionget_embedder
(multires, normalize=False, input_dims=3)
reconstruction/models/embedder.py:45
↓ 5 callersMethodget_first_stage_encoding
(self, encoder_posterior)
ldm/models/diffusion/ddpm.py:610
↓ 5 callersMethodget_pts_mask_for_conditional_volume
:param pts: [N, 3] :param mask_volume: [1, 1, X, Y, Z] :return:
reconstruction/models/sparse_neus_renderer.py:154
↓ 5 callersMethodget_unconditional_conditioning
(self, batch_size, null_label=None, image_size=512)
ldm/models/diffusion/ddpm.py:1241
↓ 5 callersFunctionisimage
(x)
ldm/util.py:100
↓ 5 callersFunctionlinear
Create a linear module.
ldm/modules/diffusionmodules/util.py:231
↓ 5 callersMethodmode
(self)
ldm/modules/distributions/distributions.py:20
↓ 5 callersMethodquantize
(self, x, *args, **kwargs)
ldm/models/autoencoder.py:437
↓ 5 callersFunctionsparse_to_dense_channel
(locs, values, dim, c, default_val, device)
reconstruction/tsparse/torchsparse_utils.py:125
↓ 4 callersMethod__init__
(self, volume, coords=None, type='dense')
reconstruction/models/sparse_sdf_network.py:538
↓ 4 callersMethod__init__
(self, pnet_type="vgg", pnet_rand=False, use_gpu=True)
ldm/modules/evaluate/evaluate_perceptualsim.py:237
↓ 4 callersMethod__init__
Imagenet Superresolution Dataloader Performs following ops in order: 1. crops a crop of size s from image either as random o
ldm/data/imagenet.py:273
↓ 4 callersMethod__init__
(self, size=None, dataroot="", datajson="", onehot_segmentation=False, use_stuffthing=False,
ldm/data/coco.py:13
↓ 4 callersMethod_make_layer
(self, block, dim, stride=1)
elevation_estimate/loftr/backbone/resnet_fpn.py:172
↓ 4 callersFunctionadd_Gaussian_noise
(img, noise_level1=2, noise_level2=25)
ldm/modules/image_degradation/bsrgan.py:369
↓ 4 callersFunctionadd_JPEG_noise
(img)
ldm/modules/image_degradation/bsrgan_light.py:422
↓ 4 callersFunctionadopt_weight
(weight, global_step, threshold=0, value=0.)
ldm/modules/losses/vqperceptual.py:20
↓ 4 callersFunctionbicubic_interpolate
two dimensional cubic interpolation :param p: [N, 4, 4] :param x: [N] :param y: [N] :return:
reconstruction/ops/grid_sampler.py:253
↓ 4 callersMethodcal_losses_sdf
(self, render_out, sample_rays, iter_step=-1, lod=0)
reconstruction/models/trainer_generic.py:1127
↓ 4 callersFunctioncalc_cam_cone_pts_3d
:param polar_deg (float). :param azimuth_deg (float). :param radius_m (float). :param fov_deg (float). :return (5, 3) array of fl
demo/app.py:48
↓ 4 callersFunctioncalculate_weights_indices
(in_length, out_length, scale, kernel, kernel_width, antialiasing)
ldm/modules/image_degradation/utils_image.py:708
↓ 4 callersFunctioncartesian_to_spherical
(xyz)
ldm/data/nerf_like.py:11
↓ 4 callersMethodcompute_top_k
(self, logits, labels, k, reduction="mean")
ldm/models/diffusion/classifier.py:150
↓ 4 callersMethodencode
(self, x0, c, t_enc, use_original_steps=False, return_intermediates=None, unconditional_guidanc
ldm/models/diffusion/ddim.py:246
↓ 4 callersMethodforward
(self, img1, img2, mask=None)
ldm/modules/evaluate/ssim.py:87
↓ 4 callersMethodget_last_layer
(self)
ldm/models/autoencoder.py:230
↓ 4 callersMethodget_last_layer
(self)
ldm/models/autoencoder.py:397
↓ 4 callersMethodget_sdf_volume
:param conditional_volume: [1,C, dX,dY,dZ] :param mask_volume: [1,1, dX,dY,dZ] :param coords_volume: [1,3, dX,dY,dZ]
reconstruction/models/sparse_sdf_network.py:441
↓ 4 callersFunctionget_stats
(stats)
ldm/modules/evaluate/torch_frechet_video_distance.py:142
↓ 4 callersMethodget_valid_sparse_coords_by_sdf
assume batch size == 1, from the first lod to get sparse voxels :param sdf_volume: [num_pts, 1] :param coords_volume: [3, X,
reconstruction/models/sparse_neus_renderer.py:823
↓ 4 callersMethodget_valid_sparse_coords_by_sdf_depthfilter
assume batch size == 1, from the first lod to get sparse voxels :param sdf_volume: [1, X, Y, Z] :param coords_volume: [3, X,
reconstruction/models/sparse_neus_renderer.py:746
↓ 4 callersFunctiongrid_sample_3d
bilinear sampling cannot guarantee continuous first-order gradient mimic pytorch grid_sample function The 8 corner points of a volume not
reconstruction/ops/grid_sampler.py:64
↓ 4 callersMethodload_im
replace background pixel with random color in rendering
ldm/data/simple.py:274
↓ 4 callersFunctionnoise_like
(shape, device, repeat=False)
ldm/modules/diffusionmodules/util.py:264
↓ 4 callersMethodprocess_im
(self, im)
ldm/data/simple.py:328
↓ 4 callersFunctionsample_ptsFeatures_from_featureVolume
sample feature of pts_wrd from featureVolume, all in world space :param pts: [N_rays, n_samples, 3] :param featureVolume: [C,wX,wY,wZ]
reconstruction/models/render_utils.py:54
↓ 4 callersMethodshared_step
(self, batch, t=None)
ldm/models/diffusion/classifier.py:179
↓ 4 callersFunctionvoxel_to_point
(x, z, nearest=False)
reconstruction/tsparse/torchsparse_utils.py:70
next →1–100 of 1,158, ranked by callers