MCPcopy Create free account

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

Functions287 in github.com/Haian-Jin/Neural_Gaffer

Method__init__
( self, start_index, end_index, output, input_dir, input_fi
scripts/Objavarse_rendering/scripts/preprocess_rendered_image.py:42
Method__init__
( self, start_index, end_index, output, input_dir, input_fi
scripts/Objavarse_rendering/scripts/preprocess_environment_map.py:152
Method__len__
(self)
dataset/dataset_relighting_training.py:125
Method__len__
(self)
dataset/dataset_relighting_eval_real.py:109
Method__len__
(self)
dataset/dataset_relighting_objaverse_3d.py:142
Method_encode_pose
(self, pose, device, num_images_per_prompt, do_classifier_free_guidance)
pipeline_neural_gaffer.py:497
Method_encode_pose
(self, pose, device, num_images_per_prompt, do_classifier_free_guidance)
neural_gaffer_3d_relighting/diffusion_relighting.py:481
Method_encode_prompt
r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*):
pipeline_neural_gaffer.py:286
Method_encode_prompt
r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*):
neural_gaffer_3d_relighting/diffusion_relighting.py:270
Method_execution_device
r""" Returns the device on which the pipeline's models will be executed. After calling `pipeline.enable_sequential_cpu_offload()` the
pipeline_neural_gaffer.py:269
Method_execution_device
r""" Returns the device on which the pipeline's models will be executed. After calling `pipeline.enable_sequential_cpu_offload()` the
neural_gaffer_3d_relighting/diffusion_relighting.py:253
Methodcompute_appfeature
(self, xyz_sampled)
neural_gaffer_3d_relighting/models/tensoRF.py:61
Methodcompute_appfeature
(self, xyz_sampled)
neural_gaffer_3d_relighting/models/tensoRF.py:234
Methodcompute_appfeature
(self, xyz_sampled)
neural_gaffer_3d_relighting/models/tensoRF.py:362
Methodcompute_densityfeature
(self, xyz_sampled)
neural_gaffer_3d_relighting/models/tensoRF.py:46
Methodcompute_densityfeature
(self, xyz_sampled)
neural_gaffer_3d_relighting/models/tensoRF.py:216
Methodcompute_densityfeature
(self, xyz_sampled)
neural_gaffer_3d_relighting/models/tensoRF.py:345
Methodcompute_features
(self, xyz_sampled)
neural_gaffer_3d_relighting/models/tensoRF.py:24
Methodcompute_features
(self, xyz_sampled)
neural_gaffer_3d_relighting/models/tensorBase.py:215
Methoddensity_L1
(self)
neural_gaffer_3d_relighting/models/tensoRF.py:198
Functiondepth2dist
(z_vals, cos_angle)
neural_gaffer_3d_relighting/dataLoader/ray_utils.py:9
Methoddisable_vae_slicing
r""" Disable sliced VAE decoding. If `enable_vae_slicing` was previously invoked, this method will go back to computing decoding in on
pipeline_neural_gaffer.py:192
Methoddisable_vae_slicing
r""" Disable sliced VAE decoding. If `enable_vae_slicing` was previously invoked, this method will go back to computing decoding in on
neural_gaffer_3d_relighting/diffusion_relighting.py:176
Methoddisable_vae_tiling
r""" Disable tiled VAE decoding. If `enable_vae_tiling` was previously invoked, this method will go back to computing decoding in one
pipeline_neural_gaffer.py:208
Methoddisable_vae_tiling
r""" Disable tiled VAE decoding. If `enable_vae_tiling` was previously invoked, this method will go back to computing decoding in one
neural_gaffer_3d_relighting/diffusion_relighting.py:192
Methodenable_model_cpu_offload
r""" Offloads all models to CPU using accelerate, reducing memory usage with a low impact on performance. Compared to `enable_sequenti
pipeline_neural_gaffer.py:240
Methodenable_model_cpu_offload
r""" Offloads all models to CPU using accelerate, reducing memory usage with a low impact on performance. Compared to `enable_sequenti
neural_gaffer_3d_relighting/diffusion_relighting.py:224
Methodenable_sequential_cpu_offload
r""" Offloads all models to CPU using accelerate, significantly reducing memory usage. When called, unet, text_encoder, vae and safety
pipeline_neural_gaffer.py:215
Methodenable_sequential_cpu_offload
r""" Offloads all models to CPU using accelerate, significantly reducing memory usage. When called, unet, text_encoder, vae and safety
neural_gaffer_3d_relighting/diffusion_relighting.py:199
Methodenable_vae_slicing
r""" Enable sliced VAE decoding. When this option is enabled, the VAE will split the input tensor in slices to compute decoding in se
pipeline_neural_gaffer.py:183
Methodenable_vae_slicing
r""" Enable sliced VAE decoding. When this option is enabled, the VAE will split the input tensor in slices to compute decoding in se
neural_gaffer_3d_relighting/diffusion_relighting.py:167
Methodenable_vae_tiling
r""" Enable tiled VAE decoding. When this option is enabled, the VAE will split the input tensor into tiles to compute decoding and e
pipeline_neural_gaffer.py:199
Methodenable_vae_tiling
r""" Enable tiled VAE decoding. When this option is enabled, the VAE will split the input tensor into tiles to compute decoding and e
neural_gaffer_3d_relighting/diffusion_relighting.py:183
Functioneval_sh
Evaluate spherical harmonics at unit directions using hardcoded SH polynomials. Works with torch/np/jnp. ... Can be 0 or more batch d
neural_gaffer_3d_relighting/models/sh.py:34
FunctionfindItem
(items, target)
neural_gaffer_3d_relighting/utils.py:79
Methodforward
(self,x)
neural_gaffer_3d_relighting/utils.py:143
Methodforward
(self, pts, viewdirs, features)
neural_gaffer_3d_relighting/models/tensorBase.py:74
Methodforward
(self, pts, viewdirs, features)
neural_gaffer_3d_relighting/models/tensorBase.py:100
Methodforward
(self, pts, viewdirs, features)
neural_gaffer_3d_relighting/models/tensorBase.py:126
Methodforward
(self, rays_chunk, white_bg=True, is_train=False, ndc_ray=False, N_samples=-1)
neural_gaffer_3d_relighting/models/tensorBase.py:418
Functionget_cond_normals_map
(normals_path, RT)
scripts/Objavarse_rendering/scripts/preprocess_rendered_image.py:27
Methodget_cond_normals_map
(self, normals_path, RT)
dataset/dataset_relighting_eval_real.py:160
Methodget_cond_normals_map
(self, normals_path, RT)
dataset/dataset_relighting_objaverse_3d.py:192
Methodget_optparam_groups
(self, lr_init_spatialxyz = 0.02, lr_init_network = 0.001)
neural_gaffer_3d_relighting/models/tensoRF.py:171
Methodget_optparam_groups
(self, lr_init_spatialxyz = 0.02, lr_init_network = 0.001)
neural_gaffer_3d_relighting/models/tensoRF.py:337
Methodget_optparam_groups
(self, lr_init_spatial = 0.02, lr_init_network = 0.001)
neural_gaffer_3d_relighting/models/tensorBase.py:227
Functionget_ray_directions_blender
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:45
Functioninit_log
(log, keys)
neural_gaffer_3d_relighting/utils.py:28
Methodinit_svd_volume
(self, res, device)
neural_gaffer_3d_relighting/models/tensoRF.py:9
Methodinit_svd_volume
(self, res, device)
neural_gaffer_3d_relighting/models/tensoRF.py:145
Methodinit_svd_volume
(self, res, device)
neural_gaffer_3d_relighting/models/tensoRF.py:322
Functionlinear2srgb_torch
(tensor_0to1)
scripts/generate_bg_and_rotate_envir_map.py:243
Methodload_envir_map
(self, path)
dataset/dataset_relighting_eval_real.py:173
Methodload_im_masked
replace background pixel with white color
dataset/dataset_relighting_training.py:183
Methodload_im_masked
replace background pixel with white color
dataset/dataset_relighting_eval_real.py:128
Methodload_im_with_mask
replace background pixel with white color; the main difference with load_im_masked is that the mask is passed as a numpy array
dataset/dataset_relighting_training.py:198
Methodload_im_with_mask
replace background pixel with white color; the main difference with load_im_masked is that the mask is passed as a numpy array
dataset/dataset_relighting_eval_real.py:143
Functionndc2dist
(ndc_pts, cos_angle)
neural_gaffer_3d_relighting/dataLoader/ray_utils.py:18
Functionndc_bbox
(all_rays)
neural_gaffer_3d_relighting/dataLoader/ray_utils.py:269
Functionndc_rays
(H, W, focal, near, rays_o, rays_d)
neural_gaffer_3d_relighting/dataLoader/ray_utils.py:109
Functionprocess_im
(im)
scripts/generate_bg_and_rotate_envir_map.py:87
Functionray_marcher
sample points along the rays Inputs: rays: () Returns:
neural_gaffer_3d_relighting/dataLoader/ray_utils.py:184
Methodread_depth
(self, filename)
neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_Relighting.py:43
Methodread_depth
(self, filename)
neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_TensoRF.py:40
Functionread_hdr
Reads an HDR map from disk. Args: path (str): Path to the .hdr file. Returns: numpy.ndarray: Loaded (float) HDR map with RGB
dataset/dataset_relighting_objaverse_3d.py:20
Methodrun
(self)
scripts/Objavarse_rendering/scripts/preprocess_environment_map.py:178
Methodrun_safety_checker
(self, image, device, dtype)
pipeline_neural_gaffer.py:533
Methodrun_safety_checker
(self, image, device, dtype)
neural_gaffer_3d_relighting/diffusion_relighting.py:517
Functionsample_pdf
(bins, weights, N_samples, det=False, pytest=False)
neural_gaffer_3d_relighting/dataLoader/ray_utils.py:129
Methodshrink
(self, new_aabb)
neural_gaffer_3d_relighting/models/tensoRF.py:278
Methodshrink
(self, new_aabb, voxel_size)
neural_gaffer_3d_relighting/models/tensorBase.py:308
Methodtrain_dataloader
(self)
dataset/dataset_relighting_eval_real.py:39
Methodtrain_dataloader
(self)
dataset/dataset_relighting_objaverse_3d.py:63
Methodtrain_step
(self, pred_rgb, image_idx:list, step_ratio=None, guidance_scale=5, as_latent=False)
neural_gaffer_3d_relighting/zero123_relighting_utils.py:191
Functiontransform_from_xyz_to_degree
input: pos, [3,] camera location, point to the origin output: elevation, azimuth, distance
scripts/Objavarse_rendering/scripts/blender_script.py:411
Methodturn_off_viewdep
(self)
neural_gaffer_3d_relighting/models/tensorBase.py:177
Methodup_sampling_VM
(self, plane_coef, line_coef, res_target)
neural_gaffer_3d_relighting/models/tensoRF.py:96
Methodupsample_volume_grid
(self, res_target)
neural_gaffer_3d_relighting/models/tensoRF.py:270
Methodupsample_volume_grid
(self, res_target)
neural_gaffer_3d_relighting/models/tensoRF.py:392
Methodval_dataloader
(self)
dataset/dataset_relighting_training.py:55
Methodval_dataloader
(self)
dataset/dataset_relighting_eval_real.py:48
Methodval_dataloader
(self)
dataset/dataset_relighting_objaverse_3d.py:73
Methodvector_comp_diffs
(self)
neural_gaffer_3d_relighting/models/tensoRF.py:195
Functionvisualize_depth
depth: (H, W)
neural_gaffer_3d_relighting/utils.py:33
Functionworker
( queue: multiprocessing.JoinableQueue, count: multiprocessing.Value, gpu: int, s3: Optional[b
scripts/Objavarse_rendering/scripts/distribute-general-rendering.py:40
Methodworld2ndc
(self,points,lindisp=None)
neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_Relighting.py:229
Methodworld2ndc
(self,points, lindisp=None)
neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_TensoRF.py:170
← previous201–287 of 287, ranked by callers