↓ 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
↓ 12 callersMethoddecode(self, x_latent, cond, t_start, unconditional_guidance_scale=1.0, unconditional_conditioning=None,
ldm/models/diffusion/ddim.py:308
↓ 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
↓ 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
↓ 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 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 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 callersMethodvalidate_mesh(self, density_or_sdf_network, func_extract_geometry, world_space=True, resolution=360,
reconstruction/models/trainer_generic.py:1272
↓ 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
↓ 4 callersFunctioncalculate_weights_indices(in_length, out_length, scale, kernel, kernel_width, antialiasing)
ldm/modules/image_degradation/utils_image.py:708
↓ 4 callersMethodencode(self, x0, c, t_enc, use_original_steps=False, return_intermediates=None,
unconditional_guidanc
ldm/models/diffusion/ddim.py:246
↓ 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 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 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