MCPcopy Create free account

hub / github.com/Livioni/OmniVGGT-official / functions

Functions660 in github.com/Livioni/OmniVGGT-official

⨍ Functions 660◇ Types & classes 66
Method__len__
(self)
omnivggt/datasets/waymo.py:175
Method__len__
(self)
omnivggt/datasets/bedlam.py:153
Method__len__
(self)
omnivggt/datasets/co3d.py:183
Method__len__
(self)
omnivggt/datasets/mapfree.py:166
Method__len__
(self)
omnivggt/datasets/spring.py:175
Method__len__
(self)
omnivggt/datasets/unreal4k.py:181
Method__len__
(self)
omnivggt/datasets/utils/cropping.py:35
Method__len__
(self)
omnivggt/datasets/base/easy_dataset.py:53
Method__len__
(self)
omnivggt/datasets/base/easy_dataset.py:81
Method__len__
(self)
omnivggt/datasets/base/easy_dataset.py:133
Method__len__
(self)
omnivggt/datasets/base/base_stereo_view_dataset.py:50
Method__len__
(self)
omnivggt/datasets/base/batched_sampler.py:35
Method__len__
(self)
omnivggt/datasets/base/batched_sampler.py:87
Method__len__
(self)
omnivggt/heads/datasets/dl3dv.py:175
Method__repr__
(self)
omnivggt/datasets/base/easy_dataset.py:56
Method__repr__
(self)
omnivggt/datasets/base/easy_dataset.py:84
Method__repr__
(self)
omnivggt/datasets/base/easy_dataset.py:136
Method__repr__
(self)
omnivggt/datasets/base/base_stereo_view_dataset.py:56
Method__rmatmul__
(self, factor)
omnivggt/datasets/base/easy_dataset.py:27
Method__rmul__
(self, factor)
omnivggt/datasets/base/easy_dataset.py:24
Method__str__
(self)
omnivggt/utils/general.py:87
Method__str__
String representation showing current and average values.
omnivggt/utils/general.py:285
Method__str__
(self)
omnivggt/datasets/utils/misc.py:83
Method__str__
(self)
omnivggt/datasets/utils/misc.py:114
Method_crop_resize_if_necessary
This function: - first downsizes the image with LANCZOS inteprolation, which is better than bilinear interpolation in
omnivggt/datasets/base/base_stereo_view_dataset.py:142
Method_mark_recent
(local_pos: int)
omnivggt/datasets/base/batched_sampler.py:109
Method_read_depthmap
(self, depthpath, max_depth=None)
omnivggt/datasets/scannet.py:166
Method_read_depthmap
(self, depthpath, max_depth=None)
omnivggt/datasets/mvs_synth.py:164
Method_read_depthmap
(self, depthpath, max_depth=None)
omnivggt/datasets/vkitti.py:170
Method_read_depthmap
(self, depthpath, max_depth=None)
omnivggt/datasets/uasol.py:164
Method_read_depthmap
(self, depthpath, max_depth=None)
omnivggt/datasets/mp3d.py:163
Method_read_depthmap
(self, depthpath, max_depth=None)
omnivggt/datasets/scannetppv2.py:173
Method_read_depthmap
(self, depthpath, max_depth=None)
omnivggt/datasets/waymo.py:164
Method_read_depthmap
(self, depthpath, max_depth=None)
omnivggt/datasets/spring.py:164
Method_read_depthmap
(self, depthpath, max_depth=None)
omnivggt/datasets/unreal4k.py:170
Method_resolutions
(self)
omnivggt/datasets/base/easy_dataset.py:67
Method_resolutions
(self)
omnivggt/datasets/base/easy_dataset.py:119
Method_resolutions
(self)
omnivggt/datasets/base/easy_dataset.py:161
Method_save_paths_to_json
(self, paths, filename)
omnivggt/datasets/wildrgb.py:177
Method_save_paths_to_json
(self, paths, filename)
omnivggt/datasets/waymo.py:170
Methodadd_cameras
(self, poses, focals=None, images=None, imsizes=None, colors=None, **kw)
omnivggt/viz.py:204
Methodadd_meter
(self, name, meter)
omnivggt/datasets/utils/misc.py:126
Methodadd_rgbd
(self, image, depth, intrinsics=None, cam2world=None, zfar=np.inf, mask=None)
omnivggt/viz.py:125
Functionadjust_learning_rate
Decay the learning rate with half-cycle cosine after warmup
omnivggt/datasets/utils/misc.py:451
Functionall_reduce_mean
(x)
omnivggt/datasets/utils/misc.py:334
Functionapply_3x3
(RT, xy)
omnivggt/utils/po_utils/geom.py:133
Functionapply_4x4
(RT, xyz)
omnivggt/utils/po_utils/geom.py:106
Functionapply_4x4_py
(RT, xyz)
omnivggt/utils/po_utils/geom.py:117
Functionapply_pix_T_cam
(pix_T_cam, xyz)
omnivggt/utils/po_utils/geom.py:49
Functionapply_pix_T_cam_py
(pix_T_cam, xyz)
omnivggt/utils/po_utils/geom.py:73
Functionapply_transform
(input_)
omnivggt/utils/po_utils/improc.py:34
Functionargmax2d
(heat, hard=True)
omnivggt/utils/po_utils/basic.py:85
Functionassert_same_shape
(t1, t2)
omnivggt/utils/po_utils/basic.py:30
Methodattn_residual_func
(x: Tensor, pos=None)
omnivggt/layers/block.py:82
Methodattn_residual_func
(x: Tensor, attn_bias=None)
omnivggt/layers/block.py:268
Methodaverage
Get the running average.
omnivggt/utils/general.py:296
Methodavg
(self)
omnivggt/datasets/utils/misc.py:67
Functionbbox_from_intrinsics_in_out
(input_camera_matrix, output_camera_matrix, output_resolution)
omnivggt/datasets/utils/cropping.py:180
Functioncalculate_averages
(results)
omnivggt/utils/vo_eval.py:326
Functioncam_read
Read camera data, return (M,N) tuple. M is the intrinsic matrix, N is the extrinsic matrix, so that x = M*N*X, with x being a point
omnivggt/utils/image.py:42
Functioncamera2pixels
(xyz, pix_T_cam)
omnivggt/utils/po_utils/geom.py:517
Functioncat_meshes
(meshes)
omnivggt/viz.py:80
Functioncenter_crop_image_depthmap
Jointly center-crop an image and its depthmap, and adjust the camera intrinsics accordingly. Parameters: - image: PIL.Image or similar,
omnivggt/datasets/utils/cropping.py:87
Functioncollate_with_cat
(whatever, lists=False)
omnivggt/utils/device.py:47
Functionconvert_box2d_to_intrinsics
(box2d, pix_T_cam, H, W, use_image_aspect_ratio=True, mult_padding=1.0)
omnivggt/utils/po_utils/geom.py:425
Functioncopy_data_to_device
Function that recursively copies data to a torch.device. Args: data: The data to copy to device device: The device to which the d
omnivggt/utils/general.py:139
Functioncount_parameters
(model)
omnivggt/utils/po_utils/misc.py:6
Functioncrop_and_resize
(im, boxlist, PH, PW, boxlist_is_normalized=False)
omnivggt/utils/po_utils/geom.py:322
Functioncrop_image_depth_and_intrinsic_by_pp
TODO: some names of width and height seem not consistent. Need to check. Crops the given image and depth map around the camera's pr
omnivggt/datasets/utils/cropping.py:287
Functioncrop_image_depthmap
Return a crop of the input view.
omnivggt/datasets/utils/cropping.py:163
Functiondepth2pointcloud
(z, pix_T_cam)
omnivggt/utils/po_utils/geom.py:540
Functiondepth_read
Read depth data from file, return as numpy array.
omnivggt/utils/image.py:30
Functiondepth_read
(filename)
omnivggt/datasets/tartanair.py:30
Functiondepth_to_world_coords_points
Converts a depth map to world coordinates (HxWx3) given the camera extrinsic and intrinsic. Returns both the world coordinates and the interm
omnivggt/datasets/utils/misc.py:543
Functiondilate2d
(im, times=1, device='cuda', mode='square')
omnivggt/utils/po_utils/improc.py:1517
Methoddisplay
(self, batch)
omnivggt/utils/general.py:107
Functiondraw_circles_at_xy
(xy, Y, X, sigma=12.5, round=False)
omnivggt/utils/po_utils/improc.py:245
Functionenlarge_seg_masks
(folder, kernel_size=5, prefix="dynamic_mask")
omnivggt/utils/image.py:276
Functionerode2d
(im, times=1, device='cuda')
omnivggt/utils/po_utils/improc.py:1511
Functioneval_metrics
(pred_traj, gt_traj=None, seq="", filename="", sample_stride=1)
omnivggt/utils/vo_eval.py:163
Functionexpand_extrinsic
Expand extrinsic matrix from (B, N, 3, 4) to (B, N, 4, 4). Args: extrinsic_3x4 (numpy.ndarray or torch.Tensor): Extrinsic matrix with sh
omnivggt/utils/geometry.py:533
Functioneye_3x3
(B, device='cuda')
omnivggt/utils/po_utils/geom.py:13
Functioneye_4x4
(B, device='cuda')
omnivggt/utils/po_utils/geom.py:17
Methodf
(*args, **kwargs)
omnivggt/layers/vision_transformer.py:130
Functionfarthest_point_sample
Input: xyz: pointcloud data, [B, N, C], where C is probably 3 npoint: number of samples Return: inds: sampled pointcl
omnivggt/utils/po_utils/misc.py:114
Functionfarthest_point_sample_py
(xyz, npoint)
omnivggt/utils/po_utils/misc.py:150
Methodfetch
(self, num=None)
omnivggt/utils/po_utils/misc.py:76
Methodffn_residual_func
(x: Tensor)
omnivggt/layers/block.py:85
Methodffn_residual_func
(x: Tensor, attn_bias=None)
omnivggt/layers/block.py:271
Functionfilename
transform a python obj or cmd into a proper filename. - \1 gets replaced by slash '/' - \2 gets replaced by comma ','
omnivggt/datasets/utils/misc.py:359
Functionfill_default_args
(kwargs, func)
omnivggt/utils/misc.py:32
Functionfind_reciprocal_matches
returns 3 values: 1 - reciprocal_in_P2: a boolean array of size P2.shape[0], a "True" value indicates a match 2 - nn2_in_P1: a int array
omnivggt/utils/geometry.py:435
Functionflip
flip so that tensor[0::2] <=> tensor[1::2]
omnivggt/utils/misc.py:65
Methodflops
(self)
omnivggt/layers/patch_embed.py:83
Functionflow_read
Read optical flow from file, return (U,V) tuple. Original code by Deqing Sun, adapted from Daniel Scharstein.
omnivggt/utils/image.py:58
Methodforward
Compute the total multi-task loss. Args: predictions: Dict containing model predictions for different tasks
omnivggt/loss.py:35
Methodforward
Applies 2D rotary position embeddings to input tokens. Args: tokens: Input tensor of shape (batch_size, n_heads, n_tokens, dim).
omnivggt/layers/rope.py:154
Methodforward
(self, x)
omnivggt/layers/drop_path.py:33
Methodforward
Args: x: Tensor, shape = (B, N, 9) Returns: Tensor, shape = (B, N, 1024)
omnivggt/layers/block.py:122
Methodforward
Args: x: Tensor, shape = (B, N, 9) Returns: Tensor, shape = (B, N, 1024)
omnivggt/layers/block.py:146
← previousnext →401–500 of 660, ranked by callers