MCPcopy Create free account

hub / github.com/MAC-VO/MAC-VO / functions

Functions937 in github.com/MAC-VO/MAC-VO

⨍ Functions 937◇ Types & classes 200
↓ 1 callersMethodCPUTimingContext
(cls, name: str)
Utility/Timer.py:66
↓ 1 callersFunctionCalculateOnePatch
depth_mean: (N,) torch.float depth_var: (N,) torch.float selector_dist: (N,) sum to one probabilty density tensor @returns
Scripts/Statistic/Montecarlo_DepthVariance.py:22
↓ 1 callersMethodGPUTimingContext
(cls, name: str, stream: torch.cuda.Stream)
Utility/Timer.py:78
↓ 1 callersFunctionGetFlops
(model: IFrontend[ModelContext] | IMatcher[ModelContext] | IStereoDepth[ModelContext])
Scripts/AdHoc/FLOPS_analyzer.py:11
↓ 1 callersFunctionMontecarlo_sample_depth
(disparity: float, gamma: float, bl: float, fx: float, num_sample: int)
Scripts/Statistic/Montecarlo_DisparityVariance.py:7
↓ 1 callersFunctionPlotTrajectory
(gt_traj: PlotableTrajectory, trajs: list[PlotableTrajectory], file_name: Path)
Scripts/AdHoc/PlotBeautifulCompare.py:94
↓ 1 callersFunctionSimulateOnePatch
depth_mean: (N,) torch.float depth_var: (N,) torch.float selector_dist: (N,) sum to one probabilty density tensor @returns
Scripts/Statistic/Montecarlo_DepthVariance.py:4
↓ 1 callersFunctionVisualizeRerunCallback
(frame: StereoFrame, system: MACVO, pb: ColoredTqdm)
MACVO.py:19
↓ 1 callersFunctionVisualizeVRAMUsage
(frame: StereoFrame, system: MACVO, pb: ColoredTqdm)
MACVO.py:38
↓ 1 callersFunction__build_dynamic_config
(spec: DynamicConfigSpec)
Utility/Config.py:53
↓ 1 callersMethod__get_curr_time
()
Utility/Sandbox.py:152
↓ 1 callersMethod__get_git_version
()
Utility/Sandbox.py:157
↓ 1 callersMethod__get_sys_command
()
Utility/Sandbox.py:148
↓ 1 callersMethod__init__
(self, config)
DataLoader/Transform.py:155
↓ 1 callersMethod__init__
(self, config: SimpleNamespace | dict[str, Any])
DataLoader/Dataset/TartanAir2.py:37
↓ 1 callersMethod__init__
(self, config: SimpleNamespace | dict[str, Any], **_)
DataLoader/Dataset/VBR.py:71
↓ 1 callersMethod__init__
(self, config: SimpleNamespace | dict[str, Any], **_)
DataLoader/Dataset/KITTI.py:26
↓ 1 callersMethod__init__
(self, directory: Path, format: Literal["png", "jpg"])
DataLoader/Dataset/GeneralStereo.py:83
↓ 1 callersMethod__init__
(self, config: SimpleNamespace)
Module/KeypointSelector.py:56
↓ 1 callersMethod__init__
(self, k_list: List[int] = [7, 7, 7], c_list: List[int] = [128, 64, 32, 2], s_list: L
Module/Network/CovHead.py:10
↓ 1 callersMethod__init__
(self, intrinsic=False, down_scale=False, config=0, stereo=False, autoDistTarget=0., uncertainty=0, out_featur
Module/Network/TartanVOStereo/FlowPoseNet.py:46
↓ 1 callersMethod__init__
(self, in_planes)
Module/Network/TartanVOStereo/StereoNet.py:22
↓ 1 callersMethod__init__
(self, maxdisp=192, group_norm=False)
Module/Network/TartanVOStereo/PSM/stackhourglass.py:51
↓ 1 callersMethod__init__
(self, cfg)
Module/Network/StereoCov/network.py:52
↓ 1 callersMethod__init__
(self, in_planes)
Module/Network/StereoCov/StereoNet.py:23
↓ 1 callersMethod__init__
(self, md=4, flow_norm=20.0)
Module/Network/PWCNet/RAFTCov.py:12
↓ 1 callersMethod__init__
(self, in_planes, uncertainty=False)
Module/Network/PWCNet/pwc/pwc_model_tartanvo.py:45
↓ 1 callersMethod__init__
(self)
Utility/PrettyPrint.py:105
↓ 1 callersMethod__launch_optim_parallel
(self, graph_data: T_GraphInput)
Module/Optimization/Interface.py:139
↓ 1 callersMethod__launch_optim_sequential
(self, graph_data: T_GraphInput)
Module/Optimization/Interface.py:130
↓ 1 callersMethod__len__
(self)
Utility/Extensions/TensorExtension.py:194
↓ 1 callersMethod_collate
(batch: T.Sequence[Tp | None], collate_fn: CollateFn)
DataLoader/Interface.py:51
↓ 1 callersMethod_interpolate_rotation
(self, time, data)
DataLoader/Dataset/TartanAir.py:515
↓ 1 callersMethod_interpolate_trajectory
(self, time, data)
DataLoader/Dataset/TartanAir.py:526
↓ 1 callersMethod_interpolate_translation
(self, time, data)
DataLoader/Dataset/TartanAir.py:497
↓ 1 callersMethod_scale_up_to
(self, size: int)
Utility/Extensions/TensorExtension.py:86
↓ 1 callersMethod_stableNoiseModel
(self)
Module/MotionModel.py:56
↓ 1 callersMethodalignWithCameraTime
Align IMU Sequence with camera time The alignment relationship is stored in self.cam2imuIdx array as camidx -> imuidx
DataLoader/Dataset/TartanAir.py:353
↓ 1 callersMethodalignWithCameraTime
Align IMU Sequence with camera time The alignment relationship is stored in self.cam2imuIdx array as camidx -> imuidx
DataLoader/Dataset/TartanAir.py:538
↓ 1 callersMethodalignWithCameraTime
Align IMU Sequence with camera time The alignment relationship is stored in self.cam2imuIdx array as camidx -> imuidx
DataLoader/Dataset/EuRoC.py:394
↓ 1 callersMethodattend_with_rpe
(self, Q, K)
Module/Network/PWCNet/pwc_cov/attention.py:67
↓ 1 callersMethodbackward
(self, gradOutput)
Module/Network/PWCNet/pwc/correlation.py:328
↓ 1 callersMethodbuild_jacobian
This function should be implemented by the user. It should return the jacobian of the model's previous forward call with respect to m
Module/Optimization/PyposeOptimizers.py:39
↓ 1 callersMethodcast_dtype
(dtype: T_SUPPORT)
DataLoader/Transform.py:166
↓ 1 callersFunctioncenterCropTo
(x: torch.Tensor, shape: int | list[int], dim: int | list[int])
Utility/Utils.py:65
↓ 1 callersMethodcollate
A default collate function that will handle torch.Tensor, pp.LieTensor and np.array automatically. You can perform more customized co
DataLoader/Interface.py:19
↓ 1 callersMethodcorrect_distortion
(self, image: np.ndarray)
DataLoader/Dataset/EuRoC.py:231
↓ 1 callersMethodcorrect_distortion
(self, image: np.ndarray)
DataLoader/Dataset/VBR.py:60
↓ 1 callersMethodcovariance_array
(self)
Module/Optimization/TwoFramePGO/Graphs.py:64
↓ 1 callersMethodcuda_event_elapsed
(from_event: T_CUDAEvent, to_event: T_CUDAEvent)
Utility/Timer.py:158
↓ 1 callersMethodcuda_graph_estimate
If does not exist a cuda graph build one and run inference through it. Store the resulted graph in frontend for futu
Module/Frontend/Frontend.py:301
↓ 1 callersFunctioncvt2str
(i)
Utility/PrettyPrint.py:36
↓ 1 callersMethoddepth_rgba_float32
Referenced from TartanVO codebase
DataLoader/Dataset/TartanAir.py:232
↓ 1 callersMethodestimate
(self, frame: StereoData)
Module/Frontend/StereoDepth.py:89
↓ 1 callersFunctionevaluate
(model: nn.DataParallel[FlowFormerCov], loader: DataLoader, length: int, usewandb)
Train/MatchingNet/evaluate.py:27
↓ 1 callersFunctionevaluateATE
(gt_traj: PosePath3D, est_traj: PosePath3D, correct_scale=False)
Evaluation/MetricsSeq.py:18
↓ 1 callersFunctionevaluateROE
Evaluates error of rotation
Evaluation/MetricsSeq.py:26
↓ 1 callersFunctionevaluateRPE
Evaluates error of se(3) pose
Evaluation/MetricsSeq.py:43
↓ 1 callersFunctionevaluateRTE
(gt_traj: PosePath3D, est_traj: PosePath3D, correct_scale=False)
Evaluation/MetricsSeq.py:9
↓ 1 callersFunctionevaluate_depth
(depth: IStereoDepth, seq: SequenceBase[StereoFrame], max_depth: float = 80.)
Evaluation/EvalDepth.py:13
↓ 1 callersFunctionevaluate_depthcov
(depth: IStereoDepth, seq: SequenceBase[StereoFrame], max_depth: float = 80.)
Evaluation/EvalDepth.py:43
↓ 1 callersFunctionevaluate_flow
(matcher: IMatcher, seq: SequenceBase[StereoFrame], max_flow: float, huge_epe_warn: float | None = None, use_g
Evaluation/EvalFlow.py:14
↓ 1 callersFunctionevaluate_flowcov
(matcher: IMatcher, seq: SequenceBase[StereoFrame], max_flow: float, use_gt_mask: bool=False)
Evaluation/EvalFlow.py:60
↓ 1 callersFunctionexecute_experiment
(name, cfg, cfg_dict, root_box: Sandbox)
Scripts/Experiment/Experiment_TartanVO.py:12
↓ 1 callersFunctionexecute_experiment
(name, cfg, cfg_dict, root_box: Sandbox)
Scripts/Experiment/Experiment_MACVO_resolution.py:12
↓ 1 callersFunctionexecute_experiment
(name, cfg, cfg_dict, root_box: Sandbox)
Scripts/Experiment/Experiment_DPVO.py:12
↓ 1 callersFunctionexecute_experiment
(name, cfg, cfg_dict, root_box: Sandbox)
Scripts/Experiment/Experiment_MACVO.py:12
↓ 1 callersMethodfilter
(self, values: TensorBundle, device: torch.device)
Module/OutlierFilter.py:63
↓ 1 callersFunctionfilterPointsInRange
(pts1:torch.Tensor, u_range: tuple[int, int], v_range: tuple[int, int])
Utility/Point.py:5
↓ 1 callersMethodflow16to32
Referenced from TartanVO project. flow_32b (float32) [-512.0, 511.984375] flow_16b (uint16) [0 - 65535] flow_32b = (f
DataLoader/Dataset/TartanAir.py:281
↓ 1 callersMethodforward
(self, frame: StereoFrame)
DataLoader/Transform.py:90
↓ 1 callersMethodforward
(self, image1, image2)
Module/Network/FlowFormerCov/flownet.py:18
↓ 1 callersMethodforward
(self, frame_t1: StereoData, frame_t2: StereoData)
Module/Frontend/Matching.py:66
↓ 1 callersMethodframeRangeQuery
Retrieve IMU data in range of [start_frame, end_frame) Args: start_frame (int): start image frame to get IMU sequence
DataLoader/Dataset/TartanAir.py:383
↓ 1 callersMethodfrom_sandbox_mayberef
(cls, box: Sandbox)
Utility/Trajectory.py:122
↓ 1 callersFunctiongaussian_mixture_mean_var
Calculate the mean of B gaussian mixtures, each with N subpolulation. batch_means: B x N torch.float, mean of each subpopulation batch
Utility/Math.py:66
↓ 1 callersFunctiongetConsecutiveRange
(values: Sequence[I], pred: Callable[[I], bool])
Utility/Utils.py:136
↓ 1 callersMethodgetCropMargin
(shape: torch.Size)
Module/Network/TartanVOStereo/StereoVO_Interface.py:108
↓ 1 callersMethodgetCropMargin
(shape: torch.Size)
Module/Network/StereoCov/network.py:24
↓ 1 callersMethodget_children
(self)
Utility/Sandbox.py:111
↓ 1 callersMethodget_fn_mode
(func: T.Callable[T.Concatenate[str, T_Input], None])
Utility/Visualize/Rerun_Visualize.py:48
↓ 1 callersMethodget_frame2map
(self, frame: FrameNode)
Module/Map/VisualMap.py:101
↓ 1 callersMethodget_graph_data
(self, global_map: VisualMap, frame_idx: torch.Tensor, observations: torch.Tensor | N
Module/Optimization/Interface.py:118
↓ 1 callersMethodget_graph_data
(self, global_map: VisualMap, frame_idx: torch.Tensor, observations: torch.Tensor | Non
Module/Optimization/TwoFramePGO/Optimizer.py:25
↓ 1 callersMethodget_map
Provides the VisualMap built across multiple calls of .run(...).
Odometry/Interface.py:79
↓ 1 callersFunctionget_optimizer
(optim_string: T_Optimizer)
Train/MatchingNet/utils.py:43
↓ 1 callersFunctionget_scheduler
(scheduler_string: T_Scheduler)
Train/MatchingNet/utils.py:34
↓ 1 callersMethodinit_connect
(application_id: str)
Utility/Visualize/Rerun_Visualize.py:30
↓ 1 callersMethodinit_context
Given config, initialize a *mutable* context object that is preserved between optimizations. Can also be used to avoid re
Module/Optimization/Interface.py:101
↓ 1 callersMethodinitialize
(self, frame0: T_SensorFrame)
Odometry/MACVO.py:158
↓ 1 callersMethodinterpolate_rotate
(ev_time: torch.Tensor, time: torch.Tensor, quat: torch.Tensor)
DataLoader/Dataset/EuRoC.py:375
↓ 1 callersMethodis_valid_config
(cls, config: SimpleNamespace | None)
DataLoader/Dataset/TartanAir.py:701
↓ 1 callersMethodis_valid_config
(cls, config: SimpleNamespace | None)
Module/KeypointSelector.py:243
↓ 1 callersMethodis_valid_config
(cls, config: SimpleNamespace | None)
Module/Frontend/StereoDepth.py:96
↓ 1 callersMethodjacobian
Returns the jacobian of the model's previous forward call with respect to model's parameters. Should be called only after forward() h
Module/Optimization/PyposeOptimizers.py:49
↓ 1 callersFunctionloadKITTIGTPoses
(pose_dir: Path)
DataLoader/Dataset/KITTI.py:110
↓ 1 callersFunctionloadVBRGTPoses
(pose_dir: Path, cam_time: np.ndarray)
DataLoader/Dataset/VBR.py:195
↓ 1 callersMethodloadWeight
(self, weight: Path | str)
Module/Network/TartanVOStereo/StereoVO_Interface.py:56
↓ 1 callersFunctionload_EurocGTPose
(csv_file_path: Path, cam_time: np.ndarray)
DataLoader/Dataset/EuRoC.py:239
↓ 1 callersMethodload_ddp_state_dict
(self, ckpt)
Module/Network/StereoCov/network.py:75
← previousnext →201–300 of 937, ranked by callers