MCPcopy Create free account

hub / github.com/Haian-Jin/Neural_Gaffer / functions

Functions287 in github.com/Haian-Jin/Neural_Gaffer

⨍ Functions 287◇ Types & classes 25
↓ 30 callersMethodload
(self, ckpt)
neural_gaffer_3d_relighting/models/tensorBase.py:267
↓ 22 callersMethodsave
(self, path, extras=None)
neural_gaffer_3d_relighting/models/tensorBase.py:254
↓ 12 callersFunctionevaluation
(test_dataset,tensorf, args, renderer, savePath=None, N_vis=5, prtx='', N_samples=-1, white_bg=
neural_gaffer_3d_relighting/renderer.py:24
↓ 9 callersFunctionget_rays
Get ray origin and normalized directions in world coordinate for all pixels in one image. Reference: https://www.scratchapixel.com/lessons/3d
neural_gaffer_3d_relighting/dataLoader/ray_utils.py:66
↓ 7 callersMethodload_im
replace background pixel with random color in rendering
dataset/dataset_relighting_training.py:168
↓ 7 callersMethodprocess_im
(self, im)
dataset/dataset_relighting_training.py:352
↓ 5 callersMethodencode_imgs
(self, imgs, mode=False)
neural_gaffer_3d_relighting/zero123_relighting_utils.py:270
↓ 5 callersFunctionlog_validation
(validation_dataloader, vae, image_encoder, feature_extractor, unet, args, accelerator, weight_dtype, split="v
neural_gaffer_training.py:72
↓ 5 callersFunctionpositional_encoding
(positions, freqs)
neural_gaffer_3d_relighting/models/tensorBase.py:9
↓ 5 callersMethodupdate_stepSize
(self, gridSize)
neural_gaffer_3d_relighting/models/tensorBase.py:199
↓ 4 callersMethod__init__
(self,inChanel, viewpe=6, featureC=128)
neural_gaffer_3d_relighting/models/tensorBase.py:113
↓ 4 callersFunctionevaluation_path
(test_dataset,tensorf, c2ws, renderer, savePath=None, N_vis=5, prtx='', N_samples=-1, whit
neural_gaffer_3d_relighting/renderer.py:90
↓ 4 callersMethodfiltering_rays
(self, all_rays, all_rgbs, N_samples=256, chunk=10240*5, bbox_only=False)
neural_gaffer_3d_relighting/models/tensorBase.py:356
↓ 4 callersFunctionget_ray_directions
Get ray directions for all pixels in camera coordinate. Reference: https://www.scratchapixel.com/lessons/3d-basic-rendering/ r
neural_gaffer_3d_relighting/dataLoader/ray_utils.py:24
↓ 4 callersMethodprocess_im
(self, im)
dataset/dataset_relighting_objaverse_3d.py:372
↓ 3 callersFunctionN_to_reso
(n_voxels, bbox)
neural_gaffer_3d_relighting/utils.py:53
↓ 3 callersMethodgetDenseAlpha
(self,gridSize=None)
neural_gaffer_3d_relighting/models/tensorBase.py:312
↓ 3 callersMethodget_optparam_groups
(self, lr_init_spatialxyz = 0.02, lr_init_network = 0.001)
neural_gaffer_3d_relighting/models/tensoRF.py:17
↓ 3 callersMethodload_im
replace background pixel with random color in rendering
dataset/dataset_relighting_eval_real.py:113
↓ 3 callersFunctionparse_args
(input_args=None)
parse_args.py:3
↓ 3 callersMethodprepare_img_latents
(self, image, batch_size, dtype, device, generator=None, do_classifier_free_guidance=False)
pipeline_neural_gaffer.py:607
↓ 3 callersMethodprepare_img_latents
(self, image, batch_size, dtype, device, generator=None, do_classifier_free_guidance=False)
neural_gaffer_3d_relighting/diffusion_relighting.py:591
↓ 3 callersFunctionprint_model_info
(model)
neural_gaffer_training.py:419
↓ 3 callersMethodprocess_im
(self, im)
dataset/dataset_relighting_eval_real.py:210
↓ 2 callersMethodTV_loss_app
(self, reg)
neural_gaffer_3d_relighting/models/tensoRF.py:441
↓ 2 callersMethod__init__
(self, aabb, gridSize, device, **kargs)
neural_gaffer_3d_relighting/models/tensoRF.py:5
↓ 2 callersFunction_encode_image
(image_encoder, image, device, dtype, do_classifier_free_guidance)
neural_gaffer_training.py:257
↓ 2 callersMethod_tensor_size
(self,t)
neural_gaffer_3d_relighting/utils.py:153
↓ 2 callersFunctionadd_light_env
r"""Adds environment lighting. Args: env (tuple(float) or str, optional): Environment map. If tuple, it's RGB or RGBA, each el
scripts/Objavarse_rendering/scripts/blender_script.py:143
↓ 2 callersFunctioncal_n_samples
(reso, step_ratio=0.5)
neural_gaffer_3d_relighting/utils.py:58
↓ 2 callersMethodcartesian_to_spherical
(self, xyz)
dataset/dataset_relighting_training.py:139
↓ 2 callersMethodcartesian_to_spherical
(self, xyz)
dataset/dataset_relighting_objaverse_3d.py:145
↓ 2 callersMethodcompute_densityfeature
(self, xyz_sampled)
neural_gaffer_3d_relighting/models/tensorBase.py:218
↓ 2 callersFunctioncompute_spherical
(elevation, azimuth, distance)
scripts/Objavarse_rendering/scripts/blender_script.py:102
↓ 2 callersFunctionconfig_parser
(cmd=None)
neural_gaffer_3d_relighting/opt.py:3
↓ 2 callersFunctionconvert_sdf_samples_to_ply
Convert sdf samples to .ply :param pytorch_3d_sdf_tensor: a torch.FloatTensor of shape (n,n,n) :voxel_grid_origin: a list of three float
neural_gaffer_3d_relighting/utils.py:158
↓ 2 callersFunctionconvolve2d
(z, f)
neural_gaffer_3d_relighting/utils.py:107
↓ 2 callersMethoddecode_latents
(self, latents)
pipeline_neural_gaffer.py:543
↓ 2 callersMethoddecode_latents
(self, latents)
neural_gaffer_3d_relighting/diffusion_relighting.py:527
↓ 2 callersMethodfeature2density
(self, density_features)
neural_gaffer_3d_relighting/models/tensorBase.py:388
↓ 2 callersFunctiongenerate_envir_map_dir
(envmap_h, envmap_w)
scripts/Objavarse_rendering/scripts/preprocess_environment_map.py:24
↓ 2 callersMethodget_img_embeds
(self, image_list)
neural_gaffer_3d_relighting/zero123_relighting_utils.py:110
↓ 2 callersMethodget_object_id
(self)
dataset/dataset_relighting_training.py:128
↓ 2 callersMethodinit_one_svd
(self, n_component, gridSize, scale, device)
neural_gaffer_3d_relighting/models/tensoRF.py:151
↓ 2 callersMethodinit_one_svd
(self, n_component, gridSize, scale, device)
neural_gaffer_3d_relighting/models/tensoRF.py:328
↓ 2 callersMethodnormalize_coord
(self, xyz_sampled)
neural_gaffer_3d_relighting/models/tensorBase.py:224
↓ 2 callersFunctionread_pfm
(filename)
neural_gaffer_3d_relighting/dataLoader/ray_utils.py:231
↓ 2 callersMethodrefine
(self, pred_rgb, image_idx, guidance_scale=3, steps=50, strength=0.8, )
neural_gaffer_3d_relighting/zero123_relighting_utils.py:133
↓ 2 callersFunctionrgb_lpips
(np_gt, np_im, net_name, device)
neural_gaffer_3d_relighting/utils.py:71
↓ 2 callersMethodsample_alpha
(self, xyz_sampled)
neural_gaffer_3d_relighting/models/tensorBase.py:50
↓ 2 callersMethodsample_ray
(self, rays_o, rays_d, is_train=True, N_samples=-1)
neural_gaffer_3d_relighting/models/tensorBase.py:286
↓ 2 callersFunctionscene_bbox
(single_obj=None, ignore_matrix=False)
scripts/Objavarse_rendering/scripts/blender_script.py:303
↓ 2 callersFunctionscene_root_objects
()
scripts/Objavarse_rendering/scripts/blender_script.py:320
↓ 2 callersMethodup_sampling_VM
(self, plane_coef, line_coef, res_target)
neural_gaffer_3d_relighting/models/tensoRF.py:255
↓ 2 callersMethodvectorDiffs
(self, vector_comps)
neural_gaffer_3d_relighting/models/tensoRF.py:76
↓ 2 callersMethodvectorDiffs
(self, vector_comps)
neural_gaffer_3d_relighting/models/tensoRF.py:184
↓ 2 callersFunctionvisualize_depth_numpy
depth: (H, W)
neural_gaffer_3d_relighting/utils.py:11
↓ 1 callersFunctionCLIP_preprocess
(x)
neural_gaffer_training.py:245
↓ 1 callersMethodCLIP_preprocess
(self, x)
pipeline_neural_gaffer.py:424
↓ 1 callersMethodCLIP_preprocess
(self, x)
neural_gaffer_3d_relighting/diffusion_relighting.py:408
↓ 1 callersMethodCLIP_preprocess
(self, x)
neural_gaffer_3d_relighting/zero123_relighting_utils.py:95
↓ 1 callersMethodTV_loss_density
(self, reg)
neural_gaffer_3d_relighting/models/tensoRF.py:435
↓ 1 callersFunctionVAE_encode_decode
(input_image_path)
scripts/diffusion_test.py:8
↓ 1 callersMethod__len__
(self)
neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_Relighting.py:233
↓ 1 callersMethod__len__
(self)
neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_TensoRF.py:174
↓ 1 callersFunction_clip_0to1_warn_torch
Enforces [0, 1] on a tensor/array that should be already [0, 1].
scripts/generate_bg_and_rotate_envir_map.py:228
↓ 1 callersMethod_encode_image
(self, image, device, num_images_per_prompt, do_classifier_free_guidance)
pipeline_neural_gaffer.py:440
↓ 1 callersMethod_encode_image
(self, image, device, num_images_per_prompt, do_classifier_free_guidance)
neural_gaffer_3d_relighting/diffusion_relighting.py:424
↓ 1 callersFunction_encode_image_without_pose
(image_encoder, image, device, dtype, do_classifier_free_guidance)
neural_gaffer_training.py:274
↓ 1 callersMethod_encode_image_without_pose
(self, image, device, num_images_per_prompt, do_classifier_free_guidance)
pipeline_neural_gaffer.py:524
↓ 1 callersMethod_encode_image_without_pose
(self, image, device, num_images_per_prompt, do_classifier_free_guidance)
neural_gaffer_3d_relighting/diffusion_relighting.py:508
↓ 1 callersMethodcheck_inputs
(self, image, height, width, callback_steps)
pipeline_neural_gaffer.py:568
↓ 1 callersMethodcheck_inputs
(self, image, height, width, callback_steps)
neural_gaffer_3d_relighting/diffusion_relighting.py:552
↓ 1 callersMethodcompute_alpha
(self, xyz_locs, length=1)
neural_gaffer_3d_relighting/models/tensorBase.py:395
↓ 1 callersMethodcompute_appfeature
(self, xyz_sampled)
neural_gaffer_3d_relighting/models/tensorBase.py:221
↓ 1 callersMethodcompute_spherical
(self, elevation, azimuth, distance)
neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_Relighting.py:47
↓ 1 callersMethodcompute_spherical
(self, elevation, azimuth, distance)
neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_TensoRF.py:44
↓ 1 callersFunctiondda
(rays_o, rays_d, bbox_3D)
neural_gaffer_3d_relighting/dataLoader/ray_utils.py:174
↓ 1 callersMethoddecode_latents
(self, latents)
neural_gaffer_3d_relighting/zero123_relighting_utils.py:262
↓ 1 callersMethoddefine_proj_mat
(self)
neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_Relighting.py:226
↓ 1 callersMethoddefine_proj_mat
(self)
neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_TensoRF.py:167
↓ 1 callersMethoddefine_transforms
(self)
neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_Relighting.py:223
↓ 1 callersMethoddefine_transforms
(self)
neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_TensoRF.py:164
↓ 1 callersMethoddensity_L1
(self)
neural_gaffer_3d_relighting/models/tensoRF.py:429
↓ 1 callersFunctiondownload_object
Download the object and return the path.
scripts/Objavarse_rendering/scripts/blender_script.py:531
↓ 1 callersFunctioneval_sh_bases
Evaluate spherical harmonics bases at unit directions, without taking linear combination. At each point, the final result may the be
neural_gaffer_3d_relighting/models/sh.py:87
↓ 1 callersFunctionexport_mesh
(args)
neural_gaffer_3d_relighting/train.py:36
↓ 1 callersFunctionexport_mesh
(args)
neural_gaffer_3d_relighting/train_relighitng_3d.py:40
↓ 1 callersMethodfreeze_density
(self)
neural_gaffer_3d_relighting/models/tensoRF.py:164
↓ 1 callersFunctiongenerate_envir_map_dir
(envmap_h, envmap_w)
scripts/generate_bg_and_rotate_envir_map.py:40
↓ 1 callersMethodgenerate_envir_map_dir
(self, envmap_h, envmap_w)
dataset/dataset_relighting_objaverse_3d.py:212
↓ 1 callersFunctionget_3x4_RT_matrix_from_blender
(cam)
scripts/Objavarse_rendering/scripts/blender_script.py:332
↓ 1 callersMethodget_T
(self, target_RT, cond_RT)
dataset/dataset_relighting_training.py:147
↓ 1 callersMethodget_T
(self, target_RT, cond_RT)
dataset/dataset_relighting_objaverse_3d.py:154
↓ 1 callersFunctionget_envir_map_light
(envir_map, incident_dir)
scripts/generate_bg_and_rotate_envir_map.py:215
↓ 1 callersMethodget_kwargs
(self)
neural_gaffer_3d_relighting/models/tensorBase.py:230
↓ 1 callersFunctionget_light
(hdr_rgb, incident_dir, hdr_weight=None)
scripts/generate_bg_and_rotate_envir_map.py:59
↓ 1 callersFunctionget_light
(hdr_rgb, incident_dir, hdr_weight=None, if_weighted=False)
scripts/Objavarse_rendering/scripts/preprocess_environment_map.py:63
↓ 1 callersMethodget_light
(self, hdr_rgb, incident_dir, hdr_weight=None, if_weighted=False)
dataset/dataset_relighting_objaverse_3d.py:231
↓ 1 callersFunctionget_ray_d
(input_RT)
scripts/generate_bg_and_rotate_envir_map.py:184
next →1–100 of 287, ranked by callers