MCPcopy Create free account

hub / github.com/Kai-46/nerfplusplus / functions

Functions106 in github.com/Kai-46/nerfplusplus

↓ 19 callersFunctionwrite_next_bytes
pack and write to a binary file. :param fid: :param data: data to send, if multiple elements are sent at the same time, they should be enc
colmap_runner/read_write_model.py:86
↓ 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.
colmap_runner/read_write_model.py:74
↓ 11 callersMethodadd_image
(self, name, camera_id, prior_q=np.zeros(4), prior_t=np.zeros(3), image_id=None)
colmap_runner/database.py:173
↓ 8 callersFunctionarray_to_blob
(array)
colmap_runner/database.py:125
↓ 7 callersFunctionbash_run
(cmd)
colmap_runner/run_colmap_posed.py:10
↓ 7 callersFunctionbash_run
(cmd)
colmap_runner/run_colmap.py:10
↓ 7 callersFunctionfind_files
(dir, exts)
data_loader_split.py:14
↓ 4 callersMethodadd_keypoints
(self, image_id, keypoints)
colmap_runner/database.py:181
↓ 4 callersFunctionblob_to_array
(blob, dtype, shape=(-1,))
colmap_runner/database.py:132
↓ 4 callersFunctionsample_pdf
:param bins: tensor of shape [..., M+1], M is the number of bins :param weights: tensor of shape [..., M] :param N_samples: number of sam
ddp_train_nerf.py:72
↓ 4 callersFunctionsetup_logger
()
ddp_train_nerf.py:22
↓ 3 callersMethodadd_matches
(self, image_id1, image_id2, matches)
colmap_runner/database.py:196
↓ 3 callersFunctioncolorize
(x, cmap_name='jet', append_cbar=False, mask=None)
utils.py:117
↓ 3 callersFunctioncolorize_np
(x, cmap_name='jet', mask=None, append_cbar=False)
utils.py:84
↓ 3 callersMethodget_img
(self)
nerf_sample_ray_split.py:90
↓ 3 callersFunctionimage_ids_to_pair_id
(image_id1, image_id2)
colmap_runner/database.py:113
↓ 3 callersFunctionload_data_split
(basedir, scene, split, skip=1, try_load_min_depth=True, only_img_files=False)
data_loader_split.py:27
↓ 3 callersFunctionrender_single_image
(rank, world_size, models, ray_sampler, chunk_size)
ddp_train_nerf.py:124
↓ 2 callersMethodadd_camera
(self, model, width, height, params, prior_focal_length=False, camera_id=None)
colmap_runner/database.py:164
↓ 2 callersFunctioncleanup
()
ddp_train_nerf.py:282
↓ 2 callersFunctionconfig_parser
()
ddp_train_nerf.py:524
↓ 2 callersMethodconnect
(database_path)
colmap_runner/database.py:142
↓ 2 callersFunctioncreate_nerf
(rank, args)
ddp_train_nerf.py:286
↓ 2 callersFunctionextract_all_to_dir
(sparse_dir, out_dir, ext='.bin')
colmap_runner/extract_sfm.py:86
↓ 2 callersFunctionimg2mse
(x, y, mask=None)
utils.py:12
↓ 2 callersFunctionintersect_sphere
ray_o, ray_d: [..., 3] compute the depth of the intersection point between this ray and unit sphere
ddp_train_nerf.py:42
↓ 2 callersFunctionlog_view_to_tb
(writer, global_step, log_data, gt_img, mask, prefix='')
ddp_train_nerf.py:243
↓ 2 callersFunctionnormalize_cam_dict
(in_cam_dict_file, out_cam_dict_file, target_radius=1., in_geometry_file=None, out_geometry_file=None)
colmap_runner/normalize_cam_dict.py:31
↓ 2 callersFunctionparse_txt
(filename)
data_loader_split.py:29
↓ 2 callersFunctionperturb_samples
(z_vals)
ddp_train_nerf.py:60
↓ 2 callersFunctionread_model
(path, ext)
colmap_runner/read_write_model.py:410
↓ 2 callersFunctionremap_name
(name)
ddp_model.py:146
↓ 2 callersFunctionsetup
(rank, world_size)
ddp_train_nerf.py:273
↓ 1 callersMethod__init__
:param D: network depth :param W: network width :param input_ch: input channels for encodings of (x, y, z) :param inp
nerf_network.py:71
↓ 1 callersMethod__init__
(self, args)
ddp_model.py:49
↓ 1 callersFunctionconvert_cam_dict_to_pinhole_dict
(cam_dict_file, pinhole_dict_file, img_dir)
colmap_runner/run_colmap_posed.py:241
↓ 1 callersFunctioncreate_init_files
(pinhole_dict_file, db_file, out_dir)
colmap_runner/run_colmap_posed.py:53
↓ 1 callersFunctiondepth2pts_outside
ray_o, ray_d: [..., 3] depth: [...]; inverse of distance to sphere origin
ddp_model.py:16
↓ 1 callersFunctiondrawpointslines
(img1, pts1, img2, lines2, colors)
camera_inspector/inspect_epipolar_geometry.py:24
↓ 1 callersFunctionexample_usage
()
colmap_runner/database.py:228
↓ 1 callersFunctionfrustums2lineset
(frustums)
camera_visualizer/visualize_cameras.py:33
↓ 1 callersMethodget_all
(self)
nerf_sample_ray_split.py:96
↓ 1 callersFunctionget_camera_frustum
(img_size, K, W2C, frustum_length=0.5, color=[0., 1., 0.])
camera_visualizer/visualize_cameras.py:6
↓ 1 callersFunctionget_center_and_diag
(cam_centers)
colmap_runner/normalize_cam_dict.py:14
↓ 1 callersFunctionget_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 world extrinsic
nerf_sample_ray_split.py:10
↓ 1 callersFunctionget_tf_cams
(cam_dict, target_radius=1.)
colmap_runner/normalize_cam_dict.py:7
↓ 1 callersFunctionget_vertical_colorbar
(h, vmin, vmax, cmap_name='jet', label=None)
utils.py:44
↓ 1 callersFunctioninspect
(img1, K1, W2C1, img2, K2, W2C2)
camera_inspector/inspect_epipolar_geometry.py:39
↓ 1 callersFunctionmain
()
colmap_runner/read_write_model.py:461
↓ 1 callersFunctionmain
(img_dir, pinhole_dict_file, out_dir)
colmap_runner/run_colmap_posed.py:216
↓ 1 callersFunctionmain
(img_dir, out_dir, run_mvs=False)
colmap_runner/run_colmap.py:115
↓ 1 callersFunctionpair_id_to_image_ids
(pair_id)
colmap_runner/database.py:119
↓ 1 callersFunctionparse_camera_dict
(colmap_cameras, colmap_images)
colmap_runner/extract_sfm.py:48
↓ 1 callersFunctionparse_tracks
(colmap_images, colmap_points3D)
colmap_runner/extract_sfm.py:9
↓ 1 callersFunctionpath2iter
(path)
ddp_train_nerf.py:318
↓ 1 callersFunctionprepare_mvs
(img_dir, sfm_dir, mvs_dir)
colmap_runner/run_colmap_posed.py:133
↓ 1 callersFunctionprepare_mvs
(img_dir, sparse_dir, mvs_dir)
colmap_runner/run_colmap.py:65
↓ 1 callersFunctionqvec2rotmat
(qvec)
colmap_runner/read_write_model.py:434
↓ 1 callersMethodrandom_sample
:param N_rand: number of rays to be casted :return:
nerf_sample_ray_split.py:116
↓ 1 callersFunctionread_cameras_binary
see: src/base/reconstruction.cc void Reconstruction::WriteCamerasBinary(const std::string& path) void Reconstruction::ReadCameras
colmap_runner/read_write_model.py:128
↓ 1 callersFunctionread_cameras_text
see: src/base/reconstruction.cc void Reconstruction::WriteCamerasText(const std::string& path) void Reconstruction::ReadCamerasTe
colmap_runner/read_write_model.py:102
↓ 1 callersFunctionread_images_binary
see: src/base/reconstruction.cc void Reconstruction::ReadImagesBinary(const std::string& path) void Reconstruction::WriteImagesBi
colmap_runner/read_write_model.py:225
↓ 1 callersFunctionread_images_text
see: src/base/reconstruction.cc void Reconstruction::ReadImagesText(const std::string& path) void Reconstruction::WriteImagesText
colmap_runner/read_write_model.py:194
↓ 1 callersFunctionread_points3D_text
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DText(const std::string& path) void Reconstruction::WritePoints3D
colmap_runner/read_write_model.py:309
↓ 1 callersFunctionread_points3d_binary
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DBinary(const std::string& path) void Reconstruction::WritePoints
colmap_runner/read_write_model.py:336
↓ 1 callersFunctionrun_fuse
(mvs_dir, out_ply)
colmap_runner/run_colmap_posed.py:196
↓ 1 callersFunctionrun_fuse
(mvs_dir, out_ply)
colmap_runner/run_colmap.py:94
↓ 1 callersFunctionrun_photometric_mvs
(mvs_dir, window_radius)
colmap_runner/run_colmap_posed.py:180
↓ 1 callersFunctionrun_photometric_mvs
(mvs_dir, window_radius)
colmap_runner/run_colmap.py:78
↓ 1 callersFunctionrun_point_triangulation
(img_dir, db_file, out_dir)
colmap_runner/run_colmap_posed.py:110
↓ 1 callersFunctionrun_possion_mesher
(in_ply, out_ply, trim)
colmap_runner/run_colmap_posed.py:205
↓ 1 callersFunctionrun_possion_mesher
(in_ply, out_ply, trim)
colmap_runner/run_colmap.py:104
↓ 1 callersFunctionrun_sfm
(img_dir, db_file, out_dir)
colmap_runner/run_colmap.py:52
↓ 1 callersFunctionrun_sift_matching
(img_dir, db_file)
colmap_runner/run_colmap_posed.py:25
↓ 1 callersFunctionrun_sift_matching
(img_dir, db_file, remove_exist=False)
colmap_runner/run_colmap.py:21
↓ 1 callersMethodset_resolution_level
(self, resolution_level)
nerf_sample_ray_split.py:58
↓ 1 callersFunctionskew
(x)
camera_inspector/inspect_epipolar_geometry.py:7
↓ 1 callersFunctiontest
()
ddp_test_nerf.py:103
↓ 1 callersFunctiontrain
()
ddp_train_nerf.py:588
↓ 1 callersFunctiontransform_pose
(W2C, translate, scale)
colmap_runner/normalize_cam_dict.py:50
↓ 1 callersFunctiontwo_view_geometry
(intrinsics1, extrinsics1, intrinsics2, extrinsics2)
camera_inspector/inspect_epipolar_geometry.py:13
↓ 1 callersFunctionvisualize_cameras
(colored_camera_dicts, sphere_radius, camera_size=0.1, geometry_file=None, geometry_type='mesh')
camera_visualizer/visualize_cameras.py:51
↓ 1 callersFunctionwrite_cameras_binary
see: src/base/reconstruction.cc void Reconstruction::WriteCamerasBinary(const std::string& path) void Reconstruction::ReadCameras
colmap_runner/read_write_model.py:174
↓ 1 callersFunctionwrite_cameras_text
see: src/base/reconstruction.cc void Reconstruction::WriteCamerasText(const std::string& path) void Reconstruction::ReadCamerasTe
colmap_runner/read_write_model.py:157
↓ 1 callersFunctionwrite_images_binary
see: src/base/reconstruction.cc void Reconstruction::ReadImagesBinary(const std::string& path) void Reconstruction::WriteImagesBi
colmap_runner/read_write_model.py:288
↓ 1 callersFunctionwrite_images_text
see: src/base/reconstruction.cc void Reconstruction::ReadImagesText(const std::string& path) void Reconstruction::WriteImagesText
colmap_runner/read_write_model.py:260
↓ 1 callersFunctionwrite_model
(cameras, images, points3D, path, ext)
colmap_runner/read_write_model.py:422
↓ 1 callersFunctionwrite_points3D_text
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DText(const std::string& path) void Reconstruction::WritePoints3D
colmap_runner/read_write_model.py:366
↓ 1 callersFunctionwrite_points3d_binary
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DBinary(const std::string& path) void Reconstruction::WritePoints
colmap_runner/read_write_model.py:391
Method__init__
:param input_dim: dimension of input to be embedded :param max_freq_log2: log2 of max freq; min freq is 1 by default :param N
nerf_network.py:12
Method__init__
(self, H, W, intrinsics, c2w, img_path=None, resolution_level=1,
nerf_sample_ray_split.py:38
Method__init__
(self, args, optim_autoexpo=False, img_names=None)
ddp_model.py:158
Method__init__
(self, *args, **kwargs)
colmap_runner/database.py:146
Methodadd_descriptors
(self, image_id, descriptors)
colmap_runner/database.py:190
Methodadd_two_view_geometry
(self, image_id1, image_id2, matches, F=np.eye(3), E=np.eye(3), H=np.eye(3), con
colmap_runner/database.py:209
Functionddp_test_nerf
(rank, args)
ddp_test_nerf.py:22
Functionddp_train_nerf
(rank, args)
ddp_train_nerf.py:338
Methodforward
:param input: tensor of shape [..., self.input_dim] :return: tensor of shape [..., self.out_dim]
nerf_network.py:42
Methodforward
:param input: [..., input_ch+input_ch_viewdirs] :return [..., 4]
nerf_network.py:120
Methodforward
:param ray_o, ray_d: [..., 3] :param fg_z_max: [...,] :param fg_z_vals, bg_z_vals: [..., N_samples] :return
ddp_model.py:74
next →1–100 of 106, ranked by callers