Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ais-lab/d2s
/ functions
Functions
206 in github.com/ais-lab/d2s
⨍
Functions
206
◇
Types & classes
30
↓ 19 callers
Function
write_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
util/read_write_model.py:85
↓ 14 callers
Method
write
(self, message)
util/logger.py:11
↓ 13 callers
Function
read_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.
util/read_write_model.py:73
↓ 9 callers
Function
read_model
(path, ext="")
util/read_write_model.py:419
↓ 5 callers
Method
run
(self, output, data, target, mode='train')
util/pose_estimator.py:36
↓ 4 callers
Method
compute_dense_descriptor
Compute keypoints, scores, descriptors for image
detectors/point2d/SuperPoint/superpoint.py:167
↓ 4 callers
Method
get_image_path
Return a path to image
datasets/_base.py:246
↓ 4 callers
Function
qvec2rotmat
(qvec)
util/read_write_model.py:453
↓ 4 callers
Function
read_image
(path, grayscale=False)
util/io.py:5
↓ 4 callers
Method
save
(self, output, data)
util/io.py:26
↓ 4 callers
Method
show
(self)
runners/trainer.py:146
↓ 4 callers
Method
update
(self, epoch, batch_idx, loss, lr)
runners/trainer.py:117
↓ 3 callers
Method
adjust_lr
Adjust learning rate based on epoch. Optional: call this function if keep training the model after loading checkpoint
models/util_learner.py:235
↓ 3 callers
Function
getPoint3D_from_modeloutput
get uncertainty and remove point with high uncertainty args: points3D: numpy array (1, 4, N) return: points3D (N, 3)
util/help_evaluation.py:87
↓ 3 callers
Method
load_checkpoint
Load regressor checkpoint from path
models/pipeline.py:42
↓ 3 callers
Function
max_pool
(x)
detectors/point2d/SuperPoint/superpoint.py:55
↓ 3 callers
Function
open3d_get_line_set
convert a list of line3D objects to an Open3D lines set Args: lines (list[:class:`datasets._base.Line3D`] or numpy array of Nx6): The
util/visualize.py:23
↓ 3 callers
Function
open3d_get_point_set
convert a numpy array of points3D to an Open3D lines set Args: points (numpy array of Nx3): The 3D point map color (list[floa
util/visualize.py:240
↓ 3 callers
Function
rotate_points_dsacstar
(points, M)
datasets/augmentation.py:225
↓ 3 callers
Function
sample_descriptors
Interpolate descriptors at keypoint locations
detectors/point2d/SuperPoint/superpoint.py:84
↓ 3 callers
Function
step_fwd
A training/validation step.
runners/trainer.py:62
↓ 2 callers
Function
MLP
(channels:list)
models/d2s.py:35
↓ 2 callers
Method
__init__
(self, feature_dim: int, no_layers: int)
models/d2s.py:83
↓ 2 callers
Function
detect_model_format
(path, ext)
util/read_write_model.py:409
↓ 2 callers
Method
detect_points2D
Read an image, preprocess it, and Return a keypoints from that image using loaded point detector model.
datasets/_base.py:220
↓ 2 callers
Function
eval
(eval_cfg, spath, mode)
util/help_evaluation.py:108
↓ 2 callers
Method
eval
(self)
runners/evaluator.py:60
↓ 2 callers
Function
get_model
(name, _type = "detector")
models/util.py:20
↓ 2 callers
Method
get_pose_vector
Return a pose vector [tvec, qvec]
datasets/_base.py:67
↓ 2 callers
Method
image_loader
(use only for point2d detector model) Read an image, do augmentation if needed, preprocess it, and return a dictionary of im
datasets/_base.py:177
↓ 2 callers
Method
item
(self)
models/util_learner.py:47
↓ 2 callers
Function
map_dict_to_torch
(data)
datasets/dataloader.py:48
↓ 2 callers
Function
open3d_vis_3d_lines
Visualize a 3D line map with `Open3D <http://www.open3d.org/>`_ Args: lines (list[:class:`datasets._base.Line3D` and/or None]): The
util/visualize.py:80
↓ 2 callers
Function
open3d_vis_3d_points
Visualize a 3D point map with `Open3D <http://www.open3d.org/>`_ Args: points3D (list[:class:`datasets._base.Line3D` and/or None]):
util/visualize.py:259
↓ 2 callers
Method
rmfiles
(self, rm_list)
util/pose_estimator.py:31
↓ 2 callers
Function
test_point_inside_ranges
(point, ranges)
util/visualize.py:10
↓ 2 callers
Method
train
(self)
runners/trainer.py:47
↓ 2 callers
Method
update_fig
(self, idx, epoch_count, value)
runners/trainer.py:112
↓ 2 callers
Method
vis
(self)
util/help_evaluation.py:54
↓ 2 callers
Function
visualize_img_withlinesandpoints
(image, points, lines, augmented=False)
util/visualize.py:391
↓ 2 callers
Function
weighted_tanh
(repro_errs, weight)
models/util_learner.py:99
↓ 1 callers
Function
MLP
(channels:list)
models/mlp.py:32
↓ 1 callers
Method
__init__
(self)
models/util_learner.py:45
↓ 1 callers
Method
_forward
To be implemented by child class.
models/base_model.py:48
↓ 1 callers
Method
_forward_default
(self, data)
detectors/point2d/SuperPoint/superpoint.py:283
↓ 1 callers
Method
_init
To be implemented by child class.
models/base_model.py:44
↓ 1 callers
Function
add_camera
(vis, pose, camera, scale=0.1, gt = False, othermethod = False)
util/visualize.py:483
↓ 1 callers
Method
add_fig
(self, name, start_epoch)
runners/trainer.py:108
↓ 1 callers
Function
attention
(query: torch.Tensor, key: torch.Tensor, value: torch.Tensor)
models/d2s.py:46
↓ 1 callers
Function
camera_pose_estimation
(output, camera_params=None, uncertainty_point=0.5)
inference.py:23
↓ 1 callers
Function
camera_pose_estimation
Creating same inputs for limap library and estimate camera pose
util/pose_estimator.py:40
↓ 1 callers
Method
checkexist
Check if the files exist, if yes, remove them
util/pose_estimator.py:18
↓ 1 callers
Method
compute_point
(self, reprojection_error_b1, pred_cam_coords_b31, iteration, valids)
models/util_learner.py:133
↓ 1 callers
Function
count_parameters
(model)
models/pipeline.py:69
↓ 1 callers
Method
create_dict_losses
(self)
runners/trainer.py:168
↓ 1 callers
Function
create_test_list
(reference_sfm, workspace_path, test_rate=0.1)
run_custom_data/pipeline.py:83
↓ 1 callers
Method
do_augmentation
(self, image, image_infor_class, debug = False)
datasets/_base.py:139
↓ 1 callers
Method
download_model
(self, path)
detectors/point2d/SuperPoint/superpoint.py:158
↓ 1 callers
Function
draw_camera
Create axis, plane and pyramed geometries in Open3D format. :param K: calibration matrix (camera intrinsics) :param R: rotation matrix :pa
util/visualize.py:413
↓ 1 callers
Function
evaluate_pose_results
Evaluate the estimated poses with ground truth poses args: spath: path to save the estimated poses and ground truth poses mo
util/help_evaluation.py:119
↓ 1 callers
Method
final_compute
(self, repro_errs_b1N, iteration)
models/util_learner.py:150
↓ 1 callers
Function
getPoint3D_from_modeloutput
get uncertainty and remove point with high uncertainty args: points3D: numpy array (1, 4, N) return: points3D (N, 3)
inference.py:9
↓ 1 callers
Method
get_camera_vector_colmap
Return a camera vector from a colmap camera object return: numpy array of camera vector [modelid, width, height, foca
datasets/_base.py:35
↓ 1 callers
Function
get_class
Get the class object which inherits from BaseClass and is defined in the module named mod_name, child of base_path.
models/util.py:5
↓ 1 callers
Method
get_detector_models
(self)
datasets/_base.py:134
↓ 1 callers
Method
get_dict_camera
Return a dictionary of camera
datasets/_base.py:49
↓ 1 callers
Method
get_point_detector_model
Return a point detector model
datasets/_base.py:125
↓ 1 callers
Function
get_pointdetector
Get a point detector
detectors/point2d/register_pointdetector.py:1
↓ 1 callers
Method
get_pose_Tmatrix
Return a pose matrix [R|t]
datasets/_base.py:72
↓ 1 callers
Function
is_in_borders
Removes keypoints too close to the border
datasets/augmentation.py:246
↓ 1 callers
Function
is_inside_img
(points, img_shape)
datasets/augmentation.py:242
↓ 1 callers
Method
is_save_checkpoint
(self)
runners/trainer.py:131
↓ 1 callers
Method
load_all_2Dpoints_7scenes
(self)
datasets/data_collection.py:40
↓ 1 callers
Method
load_all_2Dpoints_BKC
(self)
datasets/data_collection.py:139
↓ 1 callers
Method
load_all_2Dpoints_Cambridge
(self)
datasets/data_collection.py:81
↓ 1 callers
Method
load_all_2Dpoints_by_dataset
(self, dataset)
datasets/data_collection.py:27
↓ 1 callers
Method
load_all_2Dpoints_custom
(self)
datasets/data_collection.py:184
↓ 1 callers
Method
lr_scheduler
(self, epoch=None)
models/util_learner.py:250
↓ 1 callers
Function
main
()
runners/eval.py:30
↓ 1 callers
Function
main
()
runners/train.py:36
↓ 1 callers
Function
main
()
datasets/test.py:58
↓ 1 callers
Function
main
()
util/read_write_model.py:480
↓ 1 callers
Function
map_dict_to_torch
(data, device)
inference.py:66
↓ 1 callers
Function
map_tensor
(input_, func)
models/util.py:40
↓ 1 callers
Function
open3d_vis_3d_lines_with_hightlightFrame
Save 3D line map with `Open3D <http://www.open3d.org/>`_ Args: lines3D: numpy array of Nx6 hightlight_lines3D: numpy array o
util/visualize.py:51
↓ 1 callers
Function
open3d_vis_3d_points_with_hightlightFrame
Save 3D point map with `Open3D <http://www.open3d.org/>`_ Args: points3D (list[:class:`datasets._base.Line3D` and/or None]): The 3D
util/visualize.py:291
↓ 1 callers
Function
parse_config
()
runners/eval.py:10
↓ 1 callers
Function
parse_config
()
runners/train.py:11
↓ 1 callers
Function
parse_config
()
datasets/test.py:10
↓ 1 callers
Function
pose_evaluator
Evaluate the estimated poses with ground truth poses args: eval_cfg: evaluation config spath: path to save the estimated pose
util/help_evaluation.py:101
↓ 1 callers
Function
project_loss_points
gt_pt2Ds: 2xN pt3Ds: 3xN c_pose: 1x7 camera: 1x5 valids: 1xN
models/util_learner.py:63
↓ 1 callers
Function
qvec2rotmat
(qvec)
models/util_learner.py:51
↓ 1 callers
Function
re_triangulation
(args, colmap_sfm="colmap_sift/sparse/0", test_rate=0.1)
run_custom_data/pipeline.py:19
↓ 1 callers
Function
read_cameras_binary
see: src/base/reconstruction.cc void Reconstruction::WriteCamerasBinary(const std::string& path) void Reconstruction::ReadCameras
util/read_write_model.py:127
↓ 1 callers
Function
read_cameras_text
see: src/base/reconstruction.cc void Reconstruction::WriteCamerasText(const std::string& path) void Reconstruction::ReadCamerasTe
util/read_write_model.py:101
↓ 1 callers
Function
read_image
(path, grayscale=False)
inference.py:42
↓ 1 callers
Function
read_images_binary
see: src/base/reconstruction.cc void Reconstruction::ReadImagesBinary(const std::string& path) void Reconstruction::WriteImagesBi
util/read_write_model.py:224
↓ 1 callers
Function
read_images_text
see: src/base/reconstruction.cc void Reconstruction::ReadImagesText(const std::string& path) void Reconstruction::WriteImagesText
util/read_write_model.py:193
↓ 1 callers
Function
read_points3D_binary
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DBinary(const std::string& path) void Reconstruction::WritePoints
util/read_write_model.py:335
↓ 1 callers
Function
read_points3D_text
see: src/base/reconstruction.cc void Reconstruction::ReadPoints3DText(const std::string& path) void Reconstruction::WritePoints3D
util/read_write_model.py:308
next →
1–100 of 206, ranked by callers