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

↓ 1 callersFunctioncalc_pose
(phis, thetas, size, radius=1.2)
elevation_estimate/utils/utils3d.py:36
↓ 1 callersFunctioncalc_pose
(phis, thetas, size, radius = 1.2, device='cuda')
utils/utils.py:80
↓ 1 callersMethodcalculate_adaptive_weight
(self, nll_loss, g_loss, last_layer=None)
ldm/modules/losses/vqperceptual.py:85
↓ 1 callersMethodcalculate_adaptive_weight
(self, nll_loss, g_loss, last_layer=None)
ldm/modules/losses/contperceptual.py:32
↓ 1 callersFunctioncanonical_to_camera
(pts, pose)
elevation_estimate/utils/utils3d.py:22
↓ 1 callersFunctioncart_to_hom
:param pts: (N, 3 or 2) :return pts_hom: (N, 4 or 3)
elevation_estimate/utils/utils3d.py:5
↓ 1 callersMethodcat_z_vals
(self, rays_o, rays_d, z_vals, new_z_vals, sdf, lod, sdf_network, gru_fusion,
reconstruction/models/sparse_neus_renderer.py:117
↓ 1 callersMethodcolor_blend
(self, pts, position, normals, view_dirs, feature_vectors, img_index, pts_pixel_color, pts
reconstruction/models/sparse_sdf_network.py:703
↓ 1 callersFunctioncolormap
Copied from Detectron codebase.
elevation_estimate/utils/plt_utils.py:196
↓ 1 callersMethodcompute
extract features of pts for rendering :param pts: :param geometryVolume: :param vol_dims: :param partial_vol_
reconstruction/models/projector.py:96
↓ 1 callersMethodcompute_angle
:param xyz: [N_rays, n_samples,3 ] :param query_c2w: [1,4,4] :param supporting_c2ws: [n,4,4] :return:
reconstruction/models/projector.py:15
↓ 1 callersMethodcompute_angle_view_independent
:param xyz: [N_rays, n_samples,3 ] :param surface_normals: [N_rays, n_samples,3 ] :param supporting_c2ws: [n,4,4] :r
reconstruction/models/projector.py:40
↓ 1 callersFunctioncompute_frechet_distance
(mu_sample,sigma_sample,mu_ref,sigma_ref)
ldm/modules/evaluate/torch_frechet_video_distance.py:25
↓ 1 callersMethodcompute_inception_score
(self, activations: np.ndarray, split_size: int = 5000)
ldm/modules/evaluate/adm_evaluator.py:201
↓ 1 callersMethodcompute_loss
(self, inp)
ldm/guidance.py:19
↓ 1 callersFunctioncompute_max_candidates
Compute the max candidates of all pairs within a batch Args: p_m0, p_m1 (torch.Tensor): padded masks
elevation_estimate/loftr/utils/coarse_matching.py:46
↓ 1 callersFunctioncompute_perceptual_similarity
(folder, pred_img, tgt_img, take_every_other)
ldm/modules/evaluate/evaluate_perceptualsim.py:353
↓ 1 callersMethodcompute_prec_recall
( self, activations_ref: np.ndarray, activations_sample: np.ndarray )
ldm/modules/evaluate/adm_evaluator.py:216
↓ 1 callersFunctioncompute_statistics
(videos_fake, videos_real, device: str='cuda', bs=32, only_ref=False,only_sample=False)
ldm/modules/evaluate/torch_frechet_video_distance.py:199
↓ 1 callersMethodcompute_statistics
(self, activations: np.ndarray)
ldm/modules/evaluate/adm_evaluator.py:196
↓ 1 callersMethodcompute_view_independent
extract features of pts for rendering :param pts: :param geometryVolume: :param vol_dims: :param partial_vol_
reconstruction/models/projector.py:231
↓ 1 callersFunctionconvert_mesh_format
(exp_dir, output_format=".obj")
utils/utils.py:31
↓ 1 callersFunctioncos_sim
(in0, in1)
ldm/modules/evaluate/evaluate_perceptualsim.py:25
↓ 1 callersMethodcreate_embedding_fn
(self)
reconstruction/models/embedder.py:12
↓ 1 callersFunctioncreate_window
(window_size, channel, std=1.5)
reconstruction/loss/ncc.py:36
↓ 1 callersMethoddata_setup
if use ddp, use setup() not prepare_data(), prepare_data() only called on 1 GPU/TPU in distributed :return:
reconstruction/exp_runner_generic_blender_train.py:163
↓ 1 callersMethoddata_setup
if use ddp, use setup() not prepare_data(), prepare_data() only called on 1 GPU/TPU in distributed :return:
reconstruction/exp_runner_generic_blender_val.py:160
↓ 1 callersMethodddim_sampling
(self, cond, shape, x_T=None, ddim_use_original_steps=False, callb
ldm/models/diffusion/ddim.py:130
↓ 1 callersMethoddefine_transforms
(self)
reconstruction/data/One2345_eval_new_data.py:102
↓ 1 callersMethoddefine_transforms
(self)
reconstruction/data/One2345_train.py:132
↓ 1 callersFunctiondownload_checkpoint
(url, save_path)
download_ckpt.py:4
↓ 1 callersFunctiondynamic_alpha
(n_matches, milestones=[0, 300, 1000, 2000], alphas=[1.0, 0.8, 0.4, 0.2])
elevation_estimate/utils/plotting.py:136
↓ 1 callersFunctionelev_est_api
(img_paths, min_elev=30, max_elev=150, K=None, dbg=False)
elevation_estimate/utils/elev_est_api.py:196
↓ 1 callersMethodembed
(self, inputs)
reconstruction/models/embedder.py:41
↓ 1 callersMethodencode
(self, x)
ldm/models/autoencoder.py:324
↓ 1 callersMethodencode_with_pretrained
(self,x)
ldm/modules/diffusionmodules/model.py:816
↓ 1 callersFunctionequals
(val)
ldm/modules/x_transformer.py:76
↓ 1 callersFunctionerror_colormap
(err, thr, alpha=1.0)
elevation_estimate/utils/plotting.py:150
↓ 1 callersFunctionexample03
()
ldm/data/laion.py:457
↓ 1 callersMethodexport_mesh
(self, idx=-1, resolution_level=-1)
reconstruction/exp_runner_generic_blender_train.py:551
↓ 1 callersMethodexport_mesh
(self, resolution=360)
reconstruction/exp_runner_generic_blender_val.py:553
↓ 1 callersMethodexport_mesh_step
(self, sample, iter_step=0, chunk_size=512,
reconstruction/models/trainer_generic.py:827
↓ 1 callersMethodextract_fields
(self, bound_min, bound_max, resolution, query_func, device, # * related to conditional
reconstruction/models/sparse_neus_renderer.py:882
↓ 1 callersMethodfile_backup
(self)
reconstruction/exp_runner_generic_blender_train.py:415
↓ 1 callersMethodfile_backup
(self)
reconstruction/exp_runner_generic_blender_val.py:420
↓ 1 callersMethodfilter_pts_by_depthmaps
Use the pred_depthmaps to remove redundant pts (pruned by sdf, sdf always have two sides, the back side is useless) :param coords: [n
reconstruction/models/sparse_neus_renderer.py:688
↓ 1 callersFunctionfilter_size
(x)
ldm/data/laion.py:468
↓ 1 callersFunctionfilter_watermark
(x)
ldm/data/laion.py:474
↓ 1 callersMethodforward
(self, x)
ldm/modules/diffusionmodules/util.py:210
↓ 1 callersMethodfreeze
(self)
ldm/modules/encoders/modules.py:194
↓ 1 callersMethodfreeze
(self)
ldm/modules/encoders/modules.py:264
↓ 1 callersFunctionfspecial_gaussian
(hsize, sigma)
ldm/modules/image_degradation/bsrgan_light.py:187
↓ 1 callersFunctionfspecial_gaussian
(hsize, sigma)
ldm/modules/image_degradation/bsrgan.py:187
↓ 1 callersFunctionfspecial_laplacian
(alpha)
ldm/modules/image_degradation/bsrgan_light.py:201
↓ 1 callersFunctionfspecial_laplacian
(alpha)
ldm/modules/image_degradation/bsrgan.py:201
↓ 1 callersFunctionfused_mean_variance
(x, weight)
reconstruction/models/rendering_network.py:20
↓ 1 callersFunctiongaussian
(window_size, sigma)
reconstruction/loss/ncc.py:31
↓ 1 callersFunctiongaussian
(window_size, sigma)
ldm/modules/evaluate/ssim.py:12
↓ 1 callersFunctiongen_box_mask
(mask, masked)
ldm/data/inpainting/synthetic_mask.py:66
↓ 1 callersMethodgen_fn
()
ldm/modules/evaluate/adm_evaluator.py:465
↓ 1 callersFunctiongen_pose_hypothesis
(center_elevation)
elevation_estimate/utils/elev_est_api.py:110
↓ 1 callersFunctiongen_round_mask
(mask, masked, radius)
ldm/data/inpainting/synthetic_mask.py:72
↓ 1 callersFunctiongen_segment_mask
(mask, start, end, brush_width)
ldm/data/inpainting/synthetic_mask.py:56
↓ 1 callersMethodget_T
(self, target_RT, cond_RT)
ldm/data/simple.py:257
↓ 1 callersMethodget_base
(self)
ldm/data/imagenet.py:379
↓ 1 callersFunctionget_blocks
(num_layers)
ldm/thirdp/psp/helpers.py:31
↓ 1 callersMethodget_coarse_match
Args: conf_matrix (torch.Tensor): [N, L, S] data (dict): with keys ['hw0_i', 'hw1_i', 'hw0_c', 'hw1_c'] Retur
elevation_estimate/loftr/utils/coarse_matching.py:151
↓ 1 callersFunctionget_elev_est
(feature_matching, min_elev=30, max_elev=150, K=None, dbg=False)
elevation_estimate/utils/elev_est_api.py:172
↓ 1 callersMethodget_feature_matcher
(cls)
elevation_estimate/utils/elev_est_api.py:23
↓ 1 callersFunctionget_feature_matching
(img_paths, dbg=False)
elevation_estimate/utils/elev_est_api.py:56
↓ 1 callersMethodget_fine_match
(self, coords_normed, data)
elevation_estimate/loftr/utils/fine_matching.py:63
↓ 1 callersFunctionget_image_paths
(dataroot)
ldm/modules/image_degradation/utils_image.py:67
↓ 1 callersMethodget_input
(self, batch, k, cond_key=None, bs=None, log_mode=False)
ldm/models/diffusion/ddpm.py:1493
↓ 1 callersMethodget_input
(self, batch, k, cond_key=None, bs=None, return_first_stage_outputs=False)
ldm/models/diffusion/ddpm.py:1674
↓ 1 callersMethodget_input
(self, batch, k, cond_key=None, bs=None, log_mode=False)
ldm/models/diffusion/ddpm.py:1815
↓ 1 callersMethodget_input
(self, batch, k, cond_key=None, bs=None, log_mode=False)
ldm/models/diffusion/ddpm.py:1907
↓ 1 callersMethodget_intersection
(self, rays_o, rays_d, near, far, sdf_network, conditional_volume)
reconstruction/models/fast_renderer.py:33
↓ 1 callersMethodget_metrics_step
(self, sample, perturb_overwrite=-1, background_rgb=None,
reconstruction/models/trainer_generic.py:625
↓ 1 callersMethodget_null_cond
(self, version, max_length)
ldm/modules/encoders/modules.py:301
↓ 1 callersFunctionget_obj_from_str
(string, reload=False)
ldm/util.py:141
↓ 1 callersFunctionget_poses
(init_elev)
utils/utils.py:106
↓ 1 callersMethodget_scales
(self)
ldm/guidance.py:49
↓ 1 callersMethodget_split
(self)
ldm/data/coco.py:175
↓ 1 callersFunctionget_timestamp
()
ldm/modules/image_degradation/utils_image.py:33
↓ 1 callersFunctionget_timestep_embedding
This matches the implementation in Denoising Diffusion Probabilistic Models: From Fairseq. Build sinusoidal embeddings. This matches
ldm/modules/diffusionmodules/model.py:12
↓ 1 callersFunctionget_view_frustum
Get corners of 3D camera view frustum of depth image
reconstruction/data/scene.py:15
↓ 1 callersFunctionget_world_size
()
reconstruction/utils/training_utils.py:100
↓ 1 callersMethodget_x_noisy
(self, x, t, noise=None)
ldm/models/diffusion/classifier.py:110
↓ 1 callersFunctiongm_blur_kernel
(mean, cov, size=15)
ldm/modules/image_degradation/bsrgan_light.py:86
↓ 1 callersFunctiongm_blur_kernel
(mean, cov, size=15)
ldm/modules/image_degradation/bsrgan.py:86
↓ 1 callersFunctionhom_to_cart
(pts)
elevation_estimate/utils/utils3d.py:18
↓ 1 callersFunctionimage_grid
A util function for plotting a grid of images. Args: images: (N, H, W, 4) array of RGBA images rows: number of rows in the gr
elevation_estimate/utils/plt_utils.py:21
↓ 1 callersFunctionimread_uint
(path, n_channels=3)
ldm/modules/image_degradation/utils_image.py:185
↓ 1 callersFunctionimssave
imgs: list, N images of size WxHxC
ldm/modules/image_degradation/utils_image.py:112
↓ 1 callersFunctioninfer_stage_2
(model, save_path_stage1, save_path_stage2, delta_x_2, delta_y_2, indices, device, ddim_steps=75, scale=3.0)
utils/zero123_utils.py:136
↓ 1 callersMethodinit_
(self)
ldm/modules/x_transformer.py:31
↓ 1 callersMethodinit_
(self)
ldm/modules/x_transformer.py:595
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=list())
ldm/models/autoencoder.py:78
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=list())
ldm/models/autoencoder.py:313
↓ 1 callersMethodinit_from_ckpt
(self, path, ignore_keys=list(), only_model=False)
ldm/models/diffusion/classifier.py:70
← previousnext →301–400 of 1,158, ranked by callers