MCPcopy Create free account

hub / github.com/ashawkey/RAD-NeRF / functions

Functions291 in github.com/ashawkey/RAD-NeRF

↓ 62 callersMethodto
(self, device)
data_utils/face_tracking/render_3dmm.py:54
↓ 33 callersMethodlog
(self, *args, **kwargs)
nerf/utils.py:707
↓ 11 callersMethodupdate
(self, preds, truths)
nerf/utils.py:538
↓ 8 callersFunctionforward_transform
(geometry, euler_angle, trans, focal_length, cxy)
data_utils/face_tracking/util.py:92
↓ 7 callersFunctioncustom_meshgrid
(*args)
nerf/utils.py:34
↓ 6 callersMethod__init__
(self, n_classes, *args, **kwargs)
data_utils/face_parsing/model.py:231
↓ 6 callersMethodbackward
(ctx, grad_render, grad_real)
data_utils/face_tracking/render_land.py:89
↓ 6 callersFunctioncal_lan_loss
(proj_lan, gt_lan)
data_utils/face_tracking/util.py:99
↓ 6 callersMethodclear
(self)
nerf/utils.py:526
↓ 6 callersMethoddataloader
(self)
nerf/provider.py:716
↓ 6 callersMethodget_3dlandmarks
(self, id_para, exp_para, euler_angle, trans, focal_length, cxy)
data_utils/face_tracking/facemodel.py:49
↓ 6 callersFunctionget_encoder
(encoding, input_dim=3, multires=6, degree=4, num_levels=16,
encoding.py:6
↓ 6 callersMethodrender
(self)
nerf/gui.py:553
↓ 5 callersMethodload_checkpoint
(self, checkpoint=None, model_only=False)
nerf/utils.py:1364
↓ 4 callersFunction_check_sha1
Check whether the sha1 hash of the file content matches the expected hash. Parameters ---------- filename : str Path to the
data_utils/deepspeech_features/deepspeech_store.py:148
↓ 4 callersFunctionconvert_poses
(poses)
nerf/utils.py:231
↓ 4 callersFunctioncreate_layer_basic
(in_chan, out_chan, bnum, stride=1)
data_utils/face_parsing/resnet.py:51
↓ 4 callersFunctionforward_rott
(geometry, euler_angle, trans)
data_utils/face_tracking/util.py:86
↓ 4 callersFunctionget_audio_features
(features, att_mode, index)
nerf/utils.py:42
↓ 4 callersFunctionget_rays
get rays Args: poses: [B, 4, 4], cam2world intrinsics: [4] H, W, N: int Returns: rays_o, rays_d: [B, N, 3]
nerf/utils.py:249
↓ 4 callersFunctionlinear_to_srgb
(x)
nerf/utils.py:78
↓ 4 callersFunctionset_requires_grad
(tensor_list)
data_utils/face_tracking/face_tracker.py:19
↓ 4 callersMethodwrite
(self, writer, global_step, prefix="")
nerf/utils.py:563
↓ 3 callersMethod__init__
(self, dim_in, dim_out, dim_hidden, num_layers)
nerf/network.py:70
↓ 3 callersMethodforward_geo
(self, id_para, exp_para)
data_utils/face_tracking/facemodel.py:140
↓ 3 callersFunctionget_bg_coords
(H, W, device)
nerf/utils.py:240
↓ 3 callersMethodget_landmarks
(self, img)
nerf/utils.py:500
↓ 3 callersMethodmeasure
(self)
nerf/utils.py:560
↓ 3 callersMethodrun_step
(self)
nerf/asr.py:185
↓ 3 callersMethodsave_checkpoint
(self, name=None, full=False, best=False, remove_old=True)
nerf/utils.py:1302
↓ 3 callersMethodscale
(self, delta)
nerf/gui.py:64
↓ 3 callersMethodtest
(self, loader, save_path=None, name=None, write_image=False)
nerf/utils.py:923
↓ 3 callersMethodtest_step
(self, data, bg_color=None, perturb=False)
nerf/utils.py:841
↓ 3 callersMethodtrain
(self, train_loader, valid_loader, max_epochs)
nerf/utils.py:895
↓ 2 callersMethod__init__
(self)
data_utils/face_tracking/render_land.py:116
↓ 2 callersFunction_angle_from_tan
Extract the first or third Euler angle from the two members of the matrix which are positive constant times its sine and cosine. Args:
nerf/utils.py:87
↓ 2 callersFunction_index_from_letter
(letter: str)
nerf/utils.py:120
↓ 2 callersMethodbackward
(ctx, g)
activation.py:14
↓ 2 callersFunctioncal_col_loss
(pred_img, gt_img, img_mask)
data_utils/face_tracking/util.py:103
↓ 2 callersFunctioncal_lap_loss
(tensor_list, weight_list)
data_utils/face_tracking/util.py:57
↓ 2 callersMethodclear_queue
(self)
nerf/asr.py:364
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
data_utils/face_parsing/resnet.py:14
↓ 2 callersMethodencode_audio
(self, a)
nerf/network.py:170
↓ 2 callersFunctioneuler2rot
(euler_angle)
data_utils/face_tracking/util.py:18
↓ 2 callersMethodevaluate
(self, loader, name=None)
nerf/utils.py:918
↓ 2 callersMethodevaluate_one_epoch
(self, loader, name=None)
nerf/utils.py:1220
↓ 2 callersFunctionextract_audio
Real extract audio from video file. Parameters ---------- in_video : str Path to input video file. out_audio : str
data_utils/deepspeech_features/extract_wav.py:36
↓ 2 callersFunctionextract_features
Real extract audio from video file. Parameters ---------- in_audios : list of str Paths to input audio files. out_files :
data_utils/deepspeech_features/extract_ds_features.py:46
↓ 2 callersMethodforward_geo_sub
(self, id_para, exp_para, sub_index)
data_utils/face_tracking/facemodel.py:122
↓ 2 callersMethodforward_tex
(self, tex_para)
data_utils/face_tracking/facemodel.py:150
↓ 2 callersMethodforward_torso
(self, x, poses, enc_a, c=None)
nerf/network.py:188
↓ 2 callersMethodlisten
(self)
nerf/asr.py:117
↓ 2 callersMethodmark_untrained_grid
(self, poses, intrinsic, S=64)
nerf/renderer.py:319
↓ 2 callersMethodmeasure
(self)
nerf/utils.py:429
↓ 2 callersMethodmeasure
(self)
nerf/utils.py:465
↓ 2 callersFunctionnerf_matrix_to_ngp
(pose, scale=0.33, offset=[0, 0, 0])
nerf/provider.py:19
↓ 2 callersFunctionpolygon_area
(x, y)
nerf/provider.py:47
↓ 2 callersMethodprepare_buffer
(self, outputs)
nerf/gui.py:164
↓ 2 callersFunctionpreprocess_render
(geometry, euler, trans, cam, tris, vert_tris, ori_img)
data_utils/face_tracking/render_land.py:45
↓ 2 callersMethodreport
(self)
nerf/utils.py:566
↓ 2 callersMethodreset_parameters
(self)
gridencoder/grid.py:138
↓ 2 callersFunctionrot_trans_pts
(geometry, rot, trans)
data_utils/face_tracking/util.py:52
↓ 2 callersMethodrun
(self)
nerf/asr.py:357
↓ 2 callersFunctionseed_everything
(seed)
nerf/utils.py:336
↓ 2 callersFunctionsmooth_camera_path
(poses, kernel_size=5)
nerf/provider.py:29
↓ 2 callersFunctionsrgb_to_linear
(x)
nerf/utils.py:83
↓ 2 callersMethodstop
(self)
nerf/asr.py:129
↓ 2 callersMethodtest_step
(self)
nerf/gui.py:170
↓ 2 callersMethodtrain_step
(self, data)
nerf/utils.py:718
↓ 2 callersMethodupdate_extra_state
(self, decay=0.95, S=128)
nerf/renderer.py:384
↓ 2 callersMethodupdate_pose
(self, pose)
nerf/gui.py:36
↓ 1 callersMethodIllumination_layer
(face_texture, norm, gamma)
data_utils/face_tracking/render_3dmm.py:151
↓ 1 callersMethod__init__
(self, opt, device, type='train', downscale=1)
nerf/provider.py:312
↓ 1 callersMethod__init__
(self)
data_utils/face_parsing/resnet.py:59
↓ 1 callersMethod__init__
( self, focal=1015, img_h=500, img_w=500, batch_size=1, device
data_utils/face_tracking/render_3dmm.py:80
↓ 1 callersFunction_axis_angle_rotation
Return the rotation matrices for one of the rotations about an axis of which Euler angles describe, for each value of the angle given. Ar
nerf/utils.py:172
↓ 1 callersFunction_download
Download an given URL Parameters ---------- url : str URL to download path : str, optional Destination path to s
data_utils/deepspeech_features/deepspeech_store.py:64
↓ 1 callersFunctioncal_land
(proj_geo, is_visible, lands_info, land_num)
data_utils/face_tracking/render_land.py:127
↓ 1 callersMethodcompute_normal
(self, geometry)
data_utils/face_tracking/render_3dmm.py:103
↓ 1 callersFunctioncompute_tri_normal
(geometry, tris)
data_utils/face_tracking/render_land.py:8
↓ 1 callersFunctionconv_audio_to_deepspeech_input_vector
Convert audio raw data into DeepSpeech input vector. Parameters ---------- audio : np.array Audio data. audio_sample_rat
data_utils/deepspeech_features/deepspeech_features.py:183
↓ 1 callersFunctionconv_audios_to_deepspeech
Convert list of audio files into files with DeepSpeech features. Parameters ---------- audios : list of str or list of None
data_utils/deepspeech_features/deepspeech_features.py:16
↓ 1 callersMethodcreate_file_stream
(self)
nerf/asr.py:253
↓ 1 callersMethoddensity
(self, x, enc_a, e=None)
nerf/network.py:285
↓ 1 callersMethoddensity
(self, x)
nerf/renderer.py:139
↓ 1 callersFunctioneuler2rot
(euler_angle)
data_utils/process.py:276
↓ 1 callersFunctioneuler2rot
(euler_angle)
data_utils/face_tracking/geo_transform.py:7
↓ 1 callersFunctioneuler_angles_to_matrix
Convert rotations given as Euler angles in radians to rotation matrices. Args: euler_angles: Euler angles in radians as tensor of sha
nerf/utils.py:200
↓ 1 callersMethodeval_step
(self, data)
nerf/utils.py:810
↓ 1 callersFunctionevaluate
(respth='./res/test_res', dspth='./data', cp='model_final_diss.pth')
data_utils/face_parsing/test.py:54
↓ 1 callersFunctionextract_audio
(path, out_path, sample_rate=16000)
data_utils/process.py:9
↓ 1 callersFunctionextract_audio_features
(path, mode='wav2vec')
data_utils/process.py:17
↓ 1 callersFunctionextract_background
(base_dir, ori_imgs_dir)
data_utils/process.py:63
↓ 1 callersFunctionextract_fields
(bound_min, bound_max, resolution, query_func, S=128)
nerf/utils.py:369
↓ 1 callersFunctionextract_geometry
(bound_min, bound_max, resolution, threshold, query_func)
nerf/utils.py:387
↓ 1 callersFunctionextract_images
(path, out_path, fps=25)
data_utils/process.py:29
↓ 1 callersFunctionextract_landmarks
(ori_imgs_dir)
data_utils/process.py:45
↓ 1 callersFunctionextract_semantics
(ori_imgs_dir, parsing_dir)
data_utils/process.py:37
↓ 1 callersFunctionextract_torso_and_gt
(base_dir, ori_imgs_dir)
data_utils/process.py:120
↓ 1 callersFunctionface_tracking
(ori_imgs_dir)
data_utils/process.py:242
next →1–100 of 291, ranked by callers