MCPcopy Create free account

hub / github.com/VITA-Group/GNT / functions

Functions215 in github.com/VITA-Group/GNT

↓ 13 callersFunctionread_next_bytes
Read and unpack the next bytes from a binary file. :param fid: :param num_bytes: Sum of combination of {2, 4, 8}, e.g. 2, 6, 16, 30, etc.
gnt/data_loaders/colmap_read_model.py:39
↓ 12 callersFunctionnormalize
(x)
gnt/data_loaders/llff_data_utils.py:168
↓ 12 callersFunctionnormalize
(x)
gnt/data_loaders/shiny_data_utils.py:140
↓ 11 callersFunctionget_nearest_pose_ids
Args: tar_pose: target pose [3, 3] ref_poses: reference poses [N, 3, 3] num_select: the number of nearest views to select
gnt/data_loaders/data_utils.py:225
↓ 8 callersFunctionbatch_parse_llff_poses
(poses)
gnt/data_loaders/llff_data_utils.py:27
↓ 7 callersFunctionload_llff_data
( basedir, factor=8, recenter=True, bd_factor=0.75, spherify=False, path_zflat=False,
gnt/data_loaders/llff_data_utils.py:298
↓ 6 callersMethod__init__
(self, args, in_feat_ch=32, posenc_dim=3, viewenc_dim=3, ret_alpha=False)
gnt/transformer_network.py:206
↓ 6 callersFunctioncolorize
(x, cmap_name="jet", mask=None, range=None, append_cbar=False, cbar_in_image=False)
utils.py:190
↓ 5 callersFunctionrandom_crop
(rgb, camera, src_rgbs, src_cameras, size=(400, 600), center=None)
gnt/data_loaders/data_utils.py:85
↓ 5 callersFunctionrectify_inplane_rotation
(src_pose, tar_pose, src_img, th=40)
gnt/data_loaders/data_utils.py:60
↓ 5 callersFunctionview_obj2camera_rgb
(view)
gnt/data_loaders/spaces_dataset.py:14
↓ 4 callersMethod__init__
(self, num_in_layers, num_out_layers, kernel_size, stride)
gnt/feature_network.py:143
↓ 4 callersFunctionimg2mse
:param x: img 1, [(...), 3] :param y: img 2, [(...), 3] :param mask: optional, [(...)] :return: mse score
utils.py:56
↓ 4 callersFunctionposes_avg
(poses)
gnt/data_loaders/shiny_data_utils.py:158
↓ 4 callersFunctionrandom_flip
(rgb, camera, src_rgbs, src_cameras)
gnt/data_loaders/data_utils.py:120
↓ 3 callersMethod_make_layer
(self, block, planes, blocks, stride=1, dilate=False)
gnt/feature_network.py:245
↓ 3 callersFunction_minify
(basedir, factors=[], resolutions=[])
gnt/data_loaders/llff_data_utils.py:39
↓ 3 callersFunction_minify
(basedir, factors=[], resolutions=[])
gnt/data_loaders/shiny_data_utils.py:40
↓ 3 callersFunctionconv1x1
1x1 convolution
gnt/feature_network.py:28
↓ 3 callersFunctionconv3x3
3x3 convolution with padding
gnt/feature_network.py:13
↓ 3 callersFunctioncreate_window
(window_size, channel)
utils.py:210
↓ 3 callersFunctionde_parallel
(model)
gnt/model.py:7
↓ 3 callersMethodget_all
(self)
gnt/sample_ray.py:96
↓ 3 callersFunctionposes_avg
(poses)
gnt/data_loaders/llff_data_utils.py:186
↓ 3 callersFunctionrender_single_image
:param ray_sampler: RaySamplingSingleImage for this view :param model: {'net_coarse': , 'net_fine': , ...} :param chunk_size: number of
gnt/render_image.py:6
↓ 3 callersMethodswitch_to_eval
(self)
gnt/model.py:94
↓ 2 callersFunctionReadScene
Reads a scene from the directory base_dir.
gnt/data_loaders/spaces_dataset.py:184
↓ 2 callersFunction_ssim
(img1, img2, window, window_size, channel, size_average=True)
utils.py:217
↓ 2 callersFunctioncolor_jitter
Args: img: np.float32 [h, w, 3] transform: Returns: transformed np.float32
gnt/data_loaders/data_utils.py:142
↓ 2 callersMethodcompute
:param xyz: [n_rays, n_samples, 3] :param query_camera: [1, 34], 34 = img_size(2) + intrinsics(16) + extrinsics(16) :param tr
gnt/projection.py:84
↓ 2 callersFunctioncreate_training_dataset
(args)
gnt/data_loaders/create_training_dataset.py:85
↓ 2 callersFunctionimg2psnr
(x, y, mask=None)
utils.py:71
↓ 2 callersFunctionlog_view
( global_step, args, model, ray_sampler, projector, gt_img, render_stride=1, p
train.py:223
↓ 2 callersFunctionread_cameras
(pose_file)
gnt/data_loaders/nerf_synthetic.py:13
↓ 2 callersFunctionread_cameras
(pose_file)
gnt/data_loaders/nerf_synthetic_render.py:13
↓ 2 callersFunctionread_images_binary
see: src/base/reconstruction.cc void Reconstruction::ReadImagesBinary(const std::string& path) void Reconstruction::WriteImagesBi
gnt/data_loaders/colmap_read_model.py:141
↓ 2 callersFunctionrender_rays
:param ray_batch: {'ray_o': [N_rays, 3] , 'ray_d': [N_rays, 3], 'view_dir': [N_rays, 2]} :param model: {'net_coarse': , 'net_fine': } :
gnt/render_ray.py:194
↓ 2 callersFunctionsample_pdf
:param bins: tensor of shape [N_rays, M+1], M is the number of bins :param weights: tensor of shape [N_rays, M] :param N_samples: number
gnt/render_ray.py:9
↓ 2 callersMethodskipconnect
(self, x1, x2)
gnt/feature_network.py:286
↓ 2 callersFunctionunnormalize_intrinsics
(intrinsics, h, w)
gnt/data_loaders/realestate.py:21
↓ 2 callersFunctionviewmatrix
(z, up, pos)
gnt/data_loaders/llff_data_utils.py:172
↓ 2 callersFunctionviewmatrix
(z, up, pos)
gnt/data_loaders/shiny_data_utils.py:144
↓ 1 callersFunctionReadView
(base_dir, view_json)
gnt/data_loaders/spaces_dataset.py:176
↓ 1 callersFunction_IntrinsicsFromViewDict
Fills the intrinsics matrix from view_params. Args: view_params: Dict view parameters. Returns: A 3x3 matrix representing the
gnt/data_loaders/spaces_dataset.py:161
↓ 1 callersFunction_WorldFromCameraFromViewDict
Fills the world from camera transform from the view_json. Args: view_json: A dictionary of view parameters. Returns: A 4x4 tra
gnt/data_loaders/spaces_dataset.py:137
↓ 1 callersMethod__init__
Initialisation for DatasetFromSampler.
gnt/data_loaders/create_training_dataset.py:16
↓ 1 callersFunction_load_data
(basedir, factor=None, width=None, height=None, load_imgs=True)
gnt/data_loaders/llff_data_utils.py:92
↓ 1 callersFunction_load_data
(basedir, factor=None, width=None, height=None, load_imgs=True)
gnt/data_loaders/shiny_data_utils.py:93
↓ 1 callersFunctionangular_dist_between_2_vectors
(vec1, vec2)
gnt/data_loaders/data_utils.py:201
↓ 1 callersFunctionbatched_angular_dist_rot_matrix
calculate the angular distance between two rotation matrices (batched) :param R1: the first rotation matrix [N, 3, 3] :param R2: the seco
gnt/data_loaders/data_utils.py:208
↓ 1 callersFunctioncolorize_np
turn a grayscale image into a color image :param x: input grayscale, [H, W] :param cmap_name: the colorization method :param mask: th
utils.py:132
↓ 1 callersMethodcompute_angle
:param xyz: [..., 3] :param query_camera: [34, ] :param train_cameras: [n_views, 34] :return: [n_views, ..., 4]; The
gnt/projection.py:57
↓ 1 callersMethodcompute_projections
project 3D points into cameras :param xyz: [..., 3] :param train_cameras: [n_views, 34], 34 = img_size(2) + intrinsics(16) +
gnt/projection.py:31
↓ 1 callersFunctionconfig_parser
()
config.py:4
↓ 1 callersMethodcreate_embedding_fn
(self)
gnt/transformer_network.py:12
↓ 1 callersFunctioncycle
(iterable)
utils.py:75
↓ 1 callersFunctiondeepvoxels_parse_intrinsics
(filepath, trgt_sidelength, invert_y=False)
gnt/data_loaders/data_utils.py:166
↓ 1 callersFunctiondrawpointslines
draw corresponding epilines on img1 for the points in img2
gnt/data_loaders/data_verifier.py:35
↓ 1 callersFunctionepipolar
(coord1, F, img1, img2)
gnt/data_loaders/data_verifier.py:52
↓ 1 callersFunctioneval
(args)
eval.py:41
↓ 1 callersFunctiongaussian
(window_size, sigma)
utils.py:203
↓ 1 callersFunctionget_all_views_in_scene_cam_path
(all_views)
gnt/data_loaders/spaces_dataset.py:75
↓ 1 callersFunctionget_color_jitter_params
(brightness=0.2, contrast=0.2, saturation=0.2, hue=0.2)
gnt/data_loaders/data_utils.py:132
↓ 1 callersFunctionget_intrinsics_from_hwf
(h, w, focal)
gnt/data_loaders/nerf_synthetic.py:40
↓ 1 callersFunctionget_intrinsics_from_hwf
(h, w, focal)
gnt/data_loaders/nerf_synthetic_render.py:40
↓ 1 callersMethodget_rays_single_image
:param H: image height :param W: image width :param intrinsics: 4 by 4 intrinsic matrix :param c2w: 4 by 4 camera to
gnt/sample_ray.py:70
↓ 1 callersFunctionget_vertical_colorbar
:param w: pixels :param h: pixels :param vmin: min value :param vmax: max value :param cmap_name: :param label :return:
utils.py:81
↓ 1 callersFunctionimread
(f)
gnt/data_loaders/llff_data_utils.py:152
↓ 1 callersMethodinbound
check if the pixel locations are in valid range :param pixel_locations: [..., 2] :param h: height :param w: weight
gnt/projection.py:9
↓ 1 callersMethodload_from_ckpt
load model from existing checkpoints and return the current step :param out_folder: the directory that stores ckpts :return:
gnt/model.py:137
↓ 1 callersFunctionload_llff_data
( basedir, factor=8, recenter=True, bd_factor=0.75, spherify=False, path_zflat=False,
gnt/data_loaders/shiny_data_utils.py:274
↓ 1 callersMethodload_model
(self, filename, load_opt=True, load_scheduler=True)
gnt/model.py:119
↓ 1 callersFunctionlog_view
( global_step, args, model, ray_sampler, projector, gt_img, render_stride=1, p
eval.py:125
↓ 1 callersFunctionlog_view
( global_step, args, model, ray_sampler, projector, render_stride=1, prefix="",
render.py:100
↓ 1 callersFunctionlpips
(img1, img2, net="vgg", format="NCHW")
utils.py:290
↓ 1 callersFunctionmain
()
gnt/data_loaders/colmap_read_model.py:303
↓ 1 callersFunctionmin_line_dist
(rays_o, rays_d)
gnt/data_loaders/llff_data_utils.py:237
↓ 1 callersFunctionmin_line_dist
(rays_o, rays_d)
gnt/data_loaders/shiny_data_utils.py:213
↓ 1 callersMethodnormalize
(self, pixel_locations, h, w)
gnt/projection.py:24
↓ 1 callersFunctionparse_camera
(params)
gnt/sample_ray.py:13
↓ 1 callersFunctionparse_llff_pose
convert llff format pose to 4x4 matrix of intrinsics and extrinsics (opencv convention) Args: pose: matrix [3, 4] Returns: intrin
gnt/data_loaders/llff_data_utils.py:11
↓ 1 callersFunctionparse_llff_pose
convert llff format pose to 4x4 matrix of intrinsics and extrinsics (opencv convention) Args: pose: matrix [3, 4] Returns: intrin
gnt/data_loaders/shiny_data_utils.py:12
↓ 1 callersFunctionparse_pose
(path, num_views)
gnt/data_loaders/nmr_dataset.py:15
↓ 1 callersFunctionparse_pose_file
(file)
gnt/data_loaders/realestate.py:27
↓ 1 callersFunctionquaternion_about_axis
Return quaternion for rotation about axis.
gnt/data_loaders/data_utils.py:30
↓ 1 callersFunctionquaternion_matrix
Return homogeneous rotation matrix from quaternion.
gnt/data_loaders/data_utils.py:41
↓ 1 callersFunctionqvec2rotmat
(qvec)
gnt/data_loaders/colmap_read_model.py:261
↓ 1 callersMethodrandom_sample
:param N_rand: number of rays to be casted :return:
gnt/sample_ray.py:131
↓ 1 callersFunctionread_cameras_binary
see: src/base/reconstruction.cc void Reconstruction::WriteCamerasBinary(const std::string& path) void Reconstruction::ReadCameras
gnt/data_loaders/colmap_read_model.py:77
↓ 1 callersFunctionread_cameras_text
see: src/base/reconstruction.cc void Reconstruction::WriteCamerasText(const std::string& path) void Reconstruction::ReadCamerasTe
gnt/data_loaders/colmap_read_model.py:51
↓ 1 callersFunctionread_images_text
see: src/base/reconstruction.cc void Reconstruction::ReadImagesText(const std::string& path) void Reconstruction::WriteImagesText
gnt/data_loaders/colmap_read_model.py:104
↓ 1 callersFunctionread_model
(path, ext)
gnt/data_loaders/colmap_read_model.py:249
↓ 1 callersFunctionread_points3D_text
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DText(const std::string& path) void Reconstruction::WritePoints3D
gnt/data_loaders/colmap_read_model.py:183
↓ 1 callersFunctionread_points3d_binary
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DBinary(const std::string& path) void Reconstruction::WritePoints
gnt/data_loaders/colmap_read_model.py:215
↓ 1 callersFunctionrecenter_poses
(poses)
gnt/data_loaders/llff_data_utils.py:212
↓ 1 callersFunctionrecenter_poses
(poses)
gnt/data_loaders/shiny_data_utils.py:186
↓ 1 callersFunctionrender
(args)
render.py:41
↓ 1 callersFunctionrender_path_spiral
(c2w, up, rads, focal, zdelta, zrate, rots, N)
gnt/data_loaders/llff_data_utils.py:197
↓ 1 callersFunctionrender_path_spiral
(c2w, up, rads, focal, zdelta, zrate, rots, N)
gnt/data_loaders/shiny_data_utils.py:170
↓ 1 callersFunctionsample_along_camera_ray
:param ray_o: origin of the ray in scene coordinate system; tensor of shape [N_rays, 3] :param ray_d: homogeneous ray direction vectors in sc
gnt/render_ray.py:58
next →1–100 of 215, ranked by callers