MCPcopy Create free account

hub / github.com/HavenFeng/St4RTrack / functions

Functions1,146 in github.com/HavenFeng/St4RTrack

↓ 361 callersFunctionprint
(*args, **kwargs)
croco/utils/misc.py:188
↓ 96 callersMethodmean
(self, min_size=1)
dust3r/utils/po_utils/misc.py:52
↓ 96 callersMethodsize
(self)
dust3r/datasets/utils/cropping.py:41
↓ 63 callersMethodmax
(self)
croco/utils/misc.py:75
↓ 27 callersMethodadd_pointcloud
(self, pts3d, color=(0,0,0), mask=None, denoise=False)
dust3r/viz.py:139
↓ 26 callersFunctionadd_arg
(name_or_flags, default=None, default_stereo=None, default_flow=None, **kwargs)
croco/stereoflow/train.py:43
↓ 25 callersFunctionrgb
(ftensor, true_shape=None)
dust3r/utils/image.py:91
↓ 24 callersFunctionimread_cv2
Open an image or a depthmap with opencv-python.
dust3r/utils/image.py:78
↓ 24 callersFunctionto_numpy
(x)
dust3r/utils/device.py:42
↓ 23 callersFunctiongeotrf
Apply a geometric transformation to a list of 3-D points. H: 3x3 or 4x4 projection matrix (typically a Homography) p: numpy/torch/tuple of c
dust3r/utils/geometry.py:40
↓ 19 callersFunctionview_name
(view, batch_index=None)
dust3r/datasets/base/base_stereo_view_dataset.py:196
↓ 18 callersMethodresize
(self, *args, **kwargs)
dust3r/datasets/utils/cropping.py:46
↓ 15 callersMethodupdate
(self, **kwargs)
croco/utils/misc.py:100
↓ 14 callersMethodadd_camera
(self, pose_c2w, focal=None, color=(0, 0, 0), image=None, imsize=None, cam_size=0.03)
dust3r/viz.py:192
↓ 14 callersFunctionauto_cam_size
(im_poses)
dust3r/viz.py:117
↓ 14 callersFunctionedge_str
(i, j)
dust3r/cloud_opt/commons.py:12
↓ 14 callersMethodempty
(self)
dust3r/utils/po_utils/misc.py:98
↓ 13 callersMethoddevice
(self)
dust3r/cloud_opt/base_opt.py:174
↓ 13 callersFunctioninv
Invert a torch or numpy matrix
dust3r/utils/geometry.py:104
↓ 12 callersMethod_crop_resize_if_necessary
This function: - first downsizes the image with LANCZOS inteprolation, which is better than bilinear interpolation in
dust3r/datasets/base/base_stereo_view_dataset.py:135
↓ 12 callersMethod_set_root
(self)
croco/stereoflow/datasets_stereo.py:121
↓ 12 callersFunctioninvalid_to_nans
(arr, valid_mask, ndim=999)
dust3r/utils/misc.py:129
↓ 11 callersFunctiondepthmap_to_absolute_camera_coordinates
Args: - depthmap (HxW array): - camera_intrinsics: a 3x3 matrix - camera_pose: a 4x3 or 4x4 cam2world matrix Returns:
dust3r/utils/geometry.py:237
↓ 11 callersMethodsave_glb
(self, path)
dust3r/viz.py:218
↓ 10 callersMethodfrom_pretrained
(cls, pretrained_model_name_or_path, **kw)
dust3r/model.py:93
↓ 10 callersMethodshow
(self, show_pw_cams=False, show_pw_pts3d=False, cam_size=None, **kw)
dust3r/cloud_opt/base_opt.py:427
↓ 9 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, bias=True, drop=0.)
croco/models/blocks.py:61
↓ 9 callersFunctionback2color
(i, blacken_zeros=False)
dust3r/utils/po_utils/improc.py:163
↓ 8 callersFunctioncheck_R_shape
(R)
dust3r/utils/goem_opt.py:419
↓ 8 callersFunctioncompute_average_pts_within_thresh
For the "metric-based" approach, we do either a global or per-trajectory scale, or a global Sim(3) alignment, then measure fraction of points
dust3r/track_eval_util.py:235
↓ 8 callersMethodmedian
(self)
croco/utils/misc.py:59
↓ 8 callersMethodset_epoch
(self, epoch, max_epoch=100, warmup_epoch=-1)
dust3r/datasets/base/easy_dataset.py:179
↓ 8 callersMethodsumm_rgb
(self, name, ims, blacken_zeros=False, frame_id=None, only_return=False, halfres=False)
dust3r/utils/po_utils/improc.py:620
↓ 7 callersMethod__init__
(self, max_gtnorm=None)
croco/stereoflow/criterion.py:22
↓ 7 callersMethod_set_root
(self)
croco/stereoflow/datasets_flow.py:104
↓ 7 callersFunctionpreprocess_color
(x)
dust3r/utils/po_utils/improc.py:47
↓ 7 callersMethodupdate
(self, items)
dust3r/utils/po_utils/misc.py:101
↓ 6 callersMethod__init__
(self, th=3)
dust3r/utils/goem_opt.py:576
↓ 6 callersFunctioncheck_t_shape
(t)
dust3r/utils/goem_opt.py:424
↓ 6 callersMethodcrop
(self, *args, **kwargs)
dust3r/datasets/utils/cropping.py:49
↓ 6 callersMethodgenerate_grid
(self, H, W, device)
dust3r/utils/goem_opt.py:434
↓ 6 callersFunctionget_stride_distribution
(strides, dist_type='uniform')
dust3r/utils/misc.py:14
↓ 6 callersMethodload_state_dict
(self, state_dict)
croco/utils/misc.py:277
↓ 6 callersFunctionsRT_to_4x4
(scale, R, T, device)
dust3r/cloud_opt/init_im_poses.py:271
↓ 6 callersMethodto_pil
(self)
dust3r/datasets/utils/cropping.py:37
↓ 5 callersMethod_encode_image
(self, image, true_shape, mask=None, seq_id=None)
dust3r/model.py:185
↓ 5 callersMethodclose
(self)
croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:218
↓ 5 callersMethodget_im_poses
(self)
dust3r/cloud_opt/pair_viewer.py:114
↓ 5 callersMethodget_pts3d
(self, raw=False, **kwargs)
dust3r/cloud_opt/base_opt.py:260
↓ 5 callersMethodlog_every
(self, iterable, print_freq, header=None, max_iter=None)
croco/utils/misc.py:132
↓ 5 callersFunctionmake_traj
(args)
dust3r/utils/vo_eval.py:147
↓ 5 callersFunctionpnp_denormalize
(offset, pose_norm)
dust3r/cloud_opt/common.py:130
↓ 5 callersMethodrender_viewpoint
(self, viewpoint_position, viewpoint_orientation)
croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:273
↓ 5 callersFunctionsolve_wrapper
(b, A)
dust3r/cloud_opt/levenberg_marquardt.py:15
↓ 5 callersFunctionsplit_intrinsics
(K)
dust3r/utils/po_utils/geom.py:41
↓ 5 callersMethodsumm_rgbs
(self, name, ims, frame_ids=None, blacken_zeros=False, only_return=False)
dust3r/utils/po_utils/improc.py:594
↓ 4 callersFunctionNoGradParamDict
(x)
dust3r/cloud_opt/commons.py:33
↓ 4 callersMethod__init__
(self, criterion, norm_mode='avg_dis', gt_scale=False, traj_loss=False, velo_loss=False)
dust3r/losses.py:367
↓ 4 callersMethod_encode_image
image has B x 3 x img_size x img_size do_mask: whether to perform masking or not return_all_blocks: if True, return the feat
croco/models/croco.py:137
↓ 4 callersFunction_estimate_sim3_closed_form
Closed-form Sim(3) estimation via SVD, aligning A to B. Returns: s, R, t
dust3r/track_eval_util.py:139
↓ 4 callersFunction_get_gtnorm
(gt)
croco/stereoflow/criterion.py:12
↓ 4 callersFunction_read_img
(filename)
croco/stereoflow/datasets_stereo.py:541
↓ 4 callersFunctioncholesky_wrapper
(mat, default_diag=None, force_cpu=True)
dust3r/cloud_opt/epropnp.py:9
↓ 4 callersFunctioncompute_pointcloud
Return a 3D point cloud corresponding to valid pixels of the depth map
croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:39
↓ 4 callersMethodcriterion
Args: gt_poses: Tensor of shape (B, 4, 4) representing the ground truth poses pred_poses: Tensor of shape (B, 4, 4)
dust3r/camera_solver.py:117
↓ 4 callersFunctioncrop_img
(img, size, square_ok=False, nearest=False, crop=True)
dust3r/utils/image.py:163
↓ 4 callersFunctionestimate_focal_knowing_depth
Reprojection method, for when the absolute depth is known: 1) estimate the camera focal using a robust estimator 2) reproject points
dust3r/post_process.py:12
↓ 4 callersFunctionflow_to_image
Convert flow into middlebury color code image :param flow: optical flow map :return: optical flow image in middlebury color
dust3r/utils/flow_vis.py:41
↓ 4 callersMethodget_all_pts3d
(self, gt1, gt2, pred1, pred2, dist_clip=None)
dust3r/losses.py:376
↓ 4 callersMethodget_intrinsics
(self)
dust3r/cloud_opt/pair_viewer.py:104
↓ 4 callersMethodget_masks
(self)
dust3r/cloud_opt/base_opt.py:251
↓ 4 callersFunctionhead_factory
build a prediction head for the decoder
dust3r/heads/__init__.py:11
↓ 4 callersFunctionimg_to_tensor
(img)
croco/stereoflow/datasets_stereo.py:44
↓ 4 callersFunctioninference
(pairs, model, device, batch_size=16, verbose=False, anchor_view=0)
dust3r/inference.py:196
↓ 4 callersFunctionlist_scenes
Generic method iterating through a base_path folder to find scenes.
croco/datasets/habitat_sim/paths.py:72
↓ 4 callersFunctionlog_memory_usage
Log current memory usage.
visualizer_st4rtrack.py:30
↓ 4 callersMethodlog_prob
(self, value)
dust3r/cloud_opt/distributions.py:74
↓ 4 callersFunctionmake_fusion_block
(features, use_bn, width_ratio=1)
croco/models/dpt_block.py:218
↓ 4 callersFunctionmy_formatting
(val)
croco/datasets/habitat_sim/generate_multiview_images.py:123
↓ 4 callersFunctionquaternion_to_rot_mat
Args: quaternions (torch.Tensor): (*, 4) Returns: torch.Tensor: (*, 3, 3)
dust3r/cloud_opt/common.py:22
↓ 4 callersFunctionsave_model
(epoch, fname, best_so_far, best_pose_ate_sofar=None)
dust3r/training.py:365
↓ 4 callersFunctionsel
(x, kept)
dust3r/image_pairs.py:82
↓ 4 callersFunctiontranspose_to_landscape
Predict in the correct aspect-ratio, then transpose the result in landscape and stack everything back together.
dust3r/utils/misc.py:80
↓ 4 callersFunctionyaw_to_rot_mat
Args: yaw (torch.Tensor): (*) Returns: torch.Tensor: (*, 3, 3)
dust3r/cloud_opt/common.py:45
↓ 3 callersMethod__init__
Init. Args: scale_factor (float): scaling mode (str): interpolation mode
croco/models/dpt_block.py:232
↓ 3 callersFunction_crop
(img, sy, sx)
croco/stereoflow/engine.py:273
↓ 3 callersMethod_decoder
return_all_blocks: if True, return the features at the end of every block instead of just the features from the l
croco/models/croco.py:174
↓ 3 callersMethod_encode_image
(self, image, true_shape, mask=None)
dust3r/model_seq.py:140
↓ 3 callersMethod_encode_image_pairs
(self, img1, img2, true_shape1, true_shape2, mask1=None, mask2=None)
dust3r/model_seq.py:163
↓ 3 callersFunction_read_png_disp
(filename, coef=1.0)
croco/stereoflow/datasets_stereo.py:551
↓ 3 callersMethod_set_pose
(self, poses, idx, R, T=None, scale=None, force=False)
dust3r/cloud_opt/base_opt.py:211
↓ 3 callersMethodadd_meter
(self, name, meter)
croco/utils/misc.py:129
↓ 3 callersFunctionadd_scene_cam
(scene, pose_c2w, edge_color, image=None, focal=None, imsize=None, screen_width=0.03, marke
dust3r/viz.py:258
↓ 3 callersMethodapply_rope1d
(self, tokens, pos1d, cos, sin)
dust3r/rope3d.py:82
↓ 3 callersMethodbackward
(ctx, grad_res)
croco/models/curope/curope2d.py:25
↓ 3 callersFunctioncompute_camera_pose_opencv_convention
(camera_position, camera_orientation)
croco/datasets/habitat_sim/multiview_habitat_sim_generator.py:22
↓ 3 callersMethoddraw_circ_on_images_py
(self, rgbs, traj, vis, S=50, linewidth=1, show_dots=False, cmap=None, maxdist=None)
dust3r/utils/po_utils/improc.py:1389
↓ 3 callersMethoddraw_circs_on_image_py
(self, rgb, xy, colors=None, linewidth=10, radius=3, show_dots=False, maxdist=None)
dust3r/utils/po_utils/improc.py:1348
↓ 3 callersFunctiondraw_frame_id_on_vis
(vis, frame_id, scale=0.5, left=5, top=20)
dust3r/utils/po_utils/improc.py:400
↓ 3 callersFunctiondraw_line
(rgb, coord_y, coord_x, color, linewidth)
dust3r/utils/cotracker_vis.py:48
next →1–100 of 1,146, ranked by callers