Functions — github.com/Haian-Jin/Neural

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

Functions287 in github.com/Haian-Jin/Neural_Gaffer

Get ray directions for all pixels in camera coordinate. Reference: https://www.scratchapixel.com/lessons/3d-basic-rendering/ r

scripts/generate_bg_and_rotate_envir_map.py:164

↓ 1 callersFunctionget_rays

Get ray origin and normalized directions in world coordinate for all pixels in one image. Reference: https://www.scratchapixel.com/lessons/3d

scripts/generate_bg_and_rotate_envir_map.py:141

↓ 1 callersFunctionimage_grid

(imgs, rows, cols)

neural_gaffer_training.py:62

↓ 1 callersFunctionimage_preprocess_nosave

(input_image, lower_contrast=True, rescale=True)

scripts/segment_foreground.py:75

↓ 1 callersFunctioninit_lpips

(net_name, device)

neural_gaffer_3d_relighting/utils.py:65

↓ 1 callersMethodinit_render_func

(self, shadingMode, pos_pe, view_pe, fea_pe, featureC, device)

neural_gaffer_3d_relighting/models/tensorBase.py:181

↓ 1 callersMethodinit_svd_volume

(self, res, device)

neural_gaffer_3d_relighting/models/tensorBase.py:212

↓ 1 callersFunctioninitialize_random_lighting

()

scripts/Objavarse_rendering/scripts/blender_script.py:389

↓ 1 callersMethodload_envir_map

(self, path)

dataset/dataset_relighting_objaverse_3d.py:205

↓ 1 callersMethodload_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_objaverse_3d.py:175

↓ 1 callersFunctionload_input

(file_dir, to_relight_idx=3, device=torch.device("cuda"))

neural_gaffer_3d_relighting/train_relighitng_3d.py:52

↓ 1 callersFunctionload_object

(object_path: str)

scripts/Objavarse_rendering/scripts/blender_script.py:289

↓ 1 callersFunctionlog_validation

(validation_dataloader, vae, image_encoder, feature_extractor, unet, args, accelerator, weight_dtype, img_per_

neural_gaffer_inference_real_data.py:33

↓ 1 callersFunctionlog_validation

(validation_dataloader, vae, image_encoder, feature_extractor, unet, args, accelerator, weight_dtype, split="v

neural_gaffer_inference_objaverse_3d.py:35

↓ 1 callersFunctionmain

(args)

neural_gaffer_inference_real_data.py:145

↓ 1 callersFunctionmain

(args)

neural_gaffer_inference_objaverse_3d.py:191

↓ 1 callersFunctionmain

(args)

neural_gaffer_training.py:283

↓ 1 callersFunctionndc_rays_blender

(H, W, focal, near, rays_o, rays_d)

neural_gaffer_3d_relighting/dataLoader/ray_utils.py:90

↓ 1 callersMethodnextids

(self)

neural_gaffer_3d_relighting/train.py:27

↓ 1 callersMethodnextids

(self)

neural_gaffer_3d_relighting/train_relighitng_3d.py:31

↓ 1 callersMethodnormalize_coord

(self, xyz_sampled)

neural_gaffer_3d_relighting/models/tensorBase.py:56

↓ 1 callersFunctionnormalize_scene

()

scripts/Objavarse_rendering/scripts/blender_script.py:365

↓ 1 callersFunctionpred_bbox

(image)

scripts/segment_foreground.py:63

↓ 1 callersMethodprepare_extra_step_kwargs

(self, generator, eta)

pipeline_neural_gaffer.py:551

↓ 1 callersMethodprepare_extra_step_kwargs

(self, generator, eta)

neural_gaffer_3d_relighting/diffusion_relighting.py:535

↓ 1 callersMethodprepare_latents

(self, batch_size, num_channels_latents, height, width, dtype, device, generator, latents=None)

pipeline_neural_gaffer.py:590

↓ 1 callersMethodprepare_latents

(self, batch_size, num_channels_latents, height, width, dtype, device, generator, latents=None)

neural_gaffer_3d_relighting/diffusion_relighting.py:574

↓ 1 callersFunctionpreprocess

(predictor, raw_im, lower_contrast=False)

scripts/segment_foreground.py:106

↓ 1 callersMethodpreprocess_data

(self, start_index, end_index)

scripts/Objavarse_rendering/scripts/preprocess_rendered_image.py:70

↓ 1 callersMethodpreprocess_data

(self, start_index, end_index)

scripts/Objavarse_rendering/scripts/preprocess_environment_map.py:182

↓ 1 callersFunctionrandomize_lighting

()

scripts/Objavarse_rendering/scripts/blender_script.py:381

↓ 1 callersFunctionraw2alpha

(sigma, dist)

neural_gaffer_3d_relighting/models/tensorBase.py:17

↓ 1 callersFunctionread_hdr

Reads an HDR map from disk. Args: path (str): Path to the .hdr file. Returns: numpy.ndarray: Loaded (float) HDR map with RGB

scripts/generate_bg_and_rotate_envir_map.py:19

↓ 1 callersFunctionread_hdr

Reads an HDR map from disk. Args: path (str): Path to the .hdr file. Returns: numpy.ndarray: Loaded (float) HDR map with RGB

scripts/Objavarse_rendering/scripts/preprocess_environment_map.py:43

↓ 1 callersMethodread_meta

(self)

neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_Relighting.py:59

↓ 1 callersMethodread_meta

(self)

neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_TensoRF.py:56

↓ 1 callersFunctionreconstruction

(args)

neural_gaffer_3d_relighting/train.py:85

↓ 1 callersFunctionreconstruction

(args)

neural_gaffer_3d_relighting/train_relighitng_3d.py:142

↓ 1 callersFunctionrelighting_test

(args)

neural_gaffer_3d_relighting/train_relighitng_3d.py:118

↓ 1 callersFunctionremove_unwanted_objects

Remove unwanted objects from the scene, such as lights and background plane objects.

scripts/Objavarse_rendering/scripts/blender_script.py:193

↓ 1 callersFunctionrender_test

(args)

neural_gaffer_3d_relighting/train.py:49

↓ 1 callersFunctionrender_test

(args)

neural_gaffer_3d_relighting/train_relighitng_3d.py:82

↓ 1 callersFunctionreset_area_lighting

()

scripts/Objavarse_rendering/scripts/blender_script.py:406

↓ 1 callersFunctionreset_scene

Resets the scene to a clean state.

scripts/Objavarse_rendering/scripts/blender_script.py:213

↓ 1 callersFunctionrgb_ssim

(img0, img1, max_val, filter_size=11, filter_sigma=1.5, k1=0.01,

neural_gaffer_3d_relighting/utils.py:88

↓ 1 callersFunctionrotate_and_preprcess_envir_map

(envir_map, aligned_RT, rotation_idx=0, total_view=120)

scripts/generate_bg_and_rotate_envir_map.py:100

↓ 1 callersFunctionrotate_and_preprcess_envir_map

(envir_map, aligned_RT, light_area_weight=None, view_dirs=None)

scripts/Objavarse_rendering/scripts/preprocess_environment_map.py:101

↓ 1 callersMethodrotate_and_preprcess_envir_map

(self, envir_map, aligned_RT, rotation_idx=0)

dataset/dataset_relighting_objaverse_3d.py:261

↓ 1 callersMethodrun

(self)

scripts/Objavarse_rendering/scripts/preprocess_rendered_image.py:66

↓ 1 callersFunctionrun_dir

(args)

scripts/segment_foreground.py:113

↓ 1 callersFunctionsafe_l2_normalize_numpy

(x, dim=-1, eps=1e-6)

scripts/Objavarse_rendering/scripts/preprocess_rendered_image.py:24

↓ 1 callersMethodsafe_l2_normalize_numpy

(self, x, dim=-1, eps=1e-6)

dataset/dataset_relighting_eval_real.py:157

↓ 1 callersMethodsafe_l2_normalize_numpy

(self, x, dim=-1, eps=1e-6)

dataset/dataset_relighting_objaverse_3d.py:189

↓ 1 callersFunctionsam_init

(path="ckpts/sam_vit_h_4b8939.pth", device_id=0)

scripts/segment_foreground.py:32

↓ 1 callersFunctionsam_out_nosave

(predictor, input_image, *bbox_sliders)

scripts/segment_foreground.py:42

↓ 1 callersMethodsample_ray_ndc

(self, rays_o, rays_d, is_train=True, N_samples=-1)

neural_gaffer_3d_relighting/models/tensorBase.py:275

↓ 1 callersFunctionsave_images

(object_file: str)

scripts/Objavarse_rendering/scripts/blender_script.py:422

↓ 1 callersFunctionsave_model_card

(repo_id: str, image_logs=None, base_model=str, repo_folder=None)

neural_gaffer_training.py:208

↓ 1 callersFunctionscene_meshes

()

scripts/Objavarse_rendering/scripts/blender_script.py:326

↓ 1 callersFunctionset_camera_location

(elevation, azimuth, distance)

scripts/Objavarse_rendering/scripts/blender_script.py:116

↓ 1 callersFunctionset_camera_location_with_xyz

(xyz)

scripts/Objavarse_rendering/scripts/blender_script.py:128

↓ 1 callersMethodshrink

(self, new_aabb)

neural_gaffer_3d_relighting/models/tensoRF.py:399

↓ 1 callersMethodtrain_dataloader

(self)

dataset/dataset_relighting_training.py:44

↓ 1 callersMethodturn_on_viewdep

(self)

neural_gaffer_3d_relighting/models/tensorBase.py:179

↓ 1 callersMethodup_sampling_Vector

(self, density_line_coef, app_line_coef, res_target)

neural_gaffer_3d_relighting/models/tensoRF.py:380

↓ 1 callersMethodupdateAlphaMask

(self, gridSize=(200,200,200))

neural_gaffer_3d_relighting/models/tensorBase.py:330

↓ 1 callersMethodupsample_volume_grid

(self, res_target)

neural_gaffer_3d_relighting/models/tensoRF.py:127

↓ 1 callersMethodvector_comp_diffs

(self)

neural_gaffer_3d_relighting/models/tensoRF.py:90

FunctionOctreeRender_trilinear_fast

(rays, tensorf, chunk=4096, N_samples=-1, ndc_ray=False, white_bg=True, is_train=False, device='cuda')

neural_gaffer_3d_relighting/renderer.py:9

FunctionRGBRender

(xyz_sampled, viewdirs, features)

neural_gaffer_3d_relighting/models/tensorBase.py:34

FunctionSHRender

(xyz_sampled, viewdirs, features)

neural_gaffer_3d_relighting/models/tensorBase.py:27

MethodTV_loss_app

(self, reg)

neural_gaffer_3d_relighting/models/tensoRF.py:210

MethodTV_loss_density

(self, reg)

neural_gaffer_3d_relighting/models/tensoRF.py:204

Method__call__

r""" Function invoked when calling the pipeline for generation. Args: input_imgs (`PIL` or `List[PIL]`, *optional*):

pipeline_neural_gaffer.py:673

Method__call__

r""" Function invoked when calling the pipeline for generation. Args: input_imgs (`PIL` or `List[PIL]`, *optional*):

neural_gaffer_3d_relighting/diffusion_relighting.py:658

Method__getitem__

(self, idx)

neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_Relighting.py:241

Method__getitem__

(self, idx)

neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_TensoRF.py:177

Method__getitem__

(self, index)

dataset/dataset_relighting_training.py:215

Method__getitem__

(self, index)

dataset/dataset_relighting_eval_real.py:182

Method__getitem__

(self, index)

dataset/dataset_relighting_objaverse_3d.py:307

Method__init__

( self, vae: AutoencoderKL, image_encoder: CLIPVisionModelWithProjection, unet

pipeline_neural_gaffer.py:95

Method__init__

(self, total, batch)

neural_gaffer_3d_relighting/train.py:21

Method__init__

(self,TVLoss_weight=1)

neural_gaffer_3d_relighting/utils.py:139

Method__init__

(self, total, batch)

neural_gaffer_3d_relighting/train_relighitng_3d.py:25

Method__init__

( self, vae: AutoencoderKL, image_encoder: CLIPVisionModelWithProjection, unet

neural_gaffer_3d_relighting/diffusion_relighting.py:79

Method__init__

(self, device, fp16=True, t_range=[0.02, 0.98], model_key="kxic/zero123-xl", checkpoint_path

neural_gaffer_3d_relighting/zero123_relighting_utils.py:23

Method__init__

(self, datadir, split='train', downsample=1.0, is_stack=False, N_vis=-1, to_relight_idx=4)

neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_Relighting.py:14

Method__init__

(self, datadir, split='train', downsample=1.0, is_stack=False, N_vis=-1)

neural_gaffer_3d_relighting/dataLoader/Gaffer_3D_TensoRF.py:14

Method__init__

(self, aabb, gridSize, device, **kargs)

neural_gaffer_3d_relighting/models/tensoRF.py:140

Method__init__

(self, aabb, gridSize, device, **kargs)

neural_gaffer_3d_relighting/models/tensoRF.py:318

Method__init__

(self, device, aabb, alpha_volume)

neural_gaffer_3d_relighting/models/tensorBase.py:40

Method__init__

(self,inChanel, viewpe=6, feape=6, featureC=128)

neural_gaffer_3d_relighting/models/tensorBase.py:61

Method__init__

(self,inChanel, viewpe=6, pospe=6, featureC=128)

neural_gaffer_3d_relighting/models/tensorBase.py:87

Method__init__

(self, aabb, gridSize, device, density_n_comp = 8, appearance_n_comp = 24, app_dim = 27, s

neural_gaffer_3d_relighting/models/tensorBase.py:139

Method__init__

(self, lighting_dir_train, img_dir_train, lighting_dir_val, img_dir_val,

dataset/dataset_relighting_training.py:22

Method__init__

Create a dataset from a folder of images. If you pass in a root directory it will be searched for images ending in ext (ext can be a l

dataset/dataset_relighting_training.py:68

Method__init__

(self, lighting_dir_train, img_dir_train, lighting_dir_val, img_dir_val,

dataset/dataset_relighting_eval_real.py:17

Method__init__

Create a dataset from a folder of images. If you pass in a root directory it will be searched for images ending in ext (ext can be a l

dataset/dataset_relighting_eval_real.py:59

Method__init__

(self, lighting_dir_train, img_dir_train, lighting_dir_val, img_dir_val,

dataset/dataset_relighting_objaverse_3d.py:41

Method__init__

Create a dataset from a folder of images. If you pass in a root directory it will be searched for images ending in ext (ext can be a l

dataset/dataset_relighting_objaverse_3d.py:85

← previous next →101–200 of 287, ranked by callers