↓ 3 callersMethodextract_depth_maps(self, rays_o, rays_d, near, far, sdf_network, conditional_volume)
reconstruction/models/fast_renderer.py:23
↓ 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 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 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
↓ 2 callersMethod__init__(self, in_channels, out_channels,
kernel_size=3, stride=1, pad=1,
norm_act=I
reconstruction/models/featurenet.py:13
↓ 2 callersFunction_ssim(
img1, img2, window, window_size, channel, mask=None, size_average=True
)
ldm/modules/evaluate/ssim.py:31
↓ 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