MCPcopy Create free account

hub / github.com/dcharatan/flowmap / functions

Functions274 in github.com/dcharatan/flowmap

⨍ Functions 274◇ Types & classes 102↳ Endpoints 4
↓ 1 callersMethodcompute_unweighted_loss
( self, batch: Batch, flows: Flows, tracks: list[Tracks] | None, model
flowmap/loss/loss.py:50
↓ 1 callersMethodconvert_images
( self, images: list[UInt8[Tensor, "..."]], )
flowmap/dataset/dataset_re10k.py:141
↓ 1 callersMethodconvert_poses
( self, poses: Float[Tensor, "batch 18"], )
flowmap/dataset/dataset_re10k.py:118
↓ 1 callersFunctioncrop
Crop images so they have the same aspect ratio.
paper/generate_splat_comparison.py:100
↓ 1 callersFunctiondetect_msaa_pixels
( image: Float[Tensor, "batch 4 height width"], )
flowmap/visualization/drawing/rendering.py:29
↓ 1 callersFunctiondownload_checkpoint
( run_id: str, download_dir: Path, version: str | None, )
flowmap/misc/wandb_tools.py:13
↓ 1 callersFunctiondraw_label
Draw a black label on a white background with no border.
flowmap/visualization/layout.py:240
↓ 1 callersMethodexport
(self, device: torch.device)
flowmap/model/model_wrapper_overfit.py:107
↓ 1 callersFunctionexport_to_colmap
( exports: ModelExports, frame_paths: list[Path], uncropped_exports_shape: tuple[int, int], un
flowmap/export/colmap.py:56
↓ 1 callersFunctionfig_to_image
( fig: Figure, dpi: int = 100, device: torch.device = torch.device("cpu"), )
flowmap/misc/image_io.py:21
↓ 1 callersMethodforward
( self, videos: Float[Tensor, "batch frame 3 height width"], query_frame: int, )
flowmap/tracking/track_predictor.py:29
↓ 1 callersMethodforward_undistorted
( self, delta: Float[Tensor, "*batch 2"], )
flowmap/loss/mapping/mapping.py:46
↓ 1 callersFunctiongenerate_plot
( trajectories: list[Float[np.ndarray, "frame 3"]], labels: list[str], margin: float = 0.2, )
flowmap/visualization/visualizer_trajectory.py:23
↓ 1 callersFunctiongenerate_video_tracks
( tracker: TrackPredictor, videos: Float[Tensor, "batch frame 3 height width"], interval: int,
flowmap/tracking/__init__.py:49
↓ 1 callersFunctionget_backbone
( cfg: BackboneCfg, num_frames: int | None, image_shape: tuple[int, int] | None, )
flowmap/model/backbone/__init__.py:13
↓ 1 callersFunctionget_cache_key
( dataset: str, scene: str, indices: Int64[Tensor, " frame"], min_num_tracks: int, max_tra
flowmap/tracking/__init__.py:31
↓ 1 callersFunctionget_distinct_color
(index: int)
paper/colors.py:33
↓ 1 callersFunctionget_extrinsics
( cfg: ExtrinsicsCfg, num_frames: int | None, )
flowmap/model/extrinsics/__init__.py:13
↓ 1 callersFunctionget_frame_sampler
(cfg: FrameSamplerCfg)
flowmap/frame_sampler/__init__.py:15
↓ 1 callersFunctionget_frustums
( exports: ModelExports, z_value: float, )
paper/render_point_cloud.py:41
↓ 1 callersFunctionget_highlight_mask
( image_shape: tuple[int, int], device: torch.device, )
paper/point_clouds/birds.py:25
↓ 1 callersFunctionget_intrinsics
(cfg: IntrinsicsCfg)
flowmap/model/intrinsics/__init__.py:15
↓ 1 callersFunctionget_rotation
( points: Float[np.ndarray, "point 3"], flip: bool, # For when PCA doesn't give the desired orientati
paper/generate_trajectory_plots.py:34
↓ 1 callersFunctionget_track_predictor
(cfg: TrackPredictorCfg)
flowmap/tracking/__init__.py:25
↓ 1 callersFunctionget_typed_config
( data_class: Type[T], cfg: DictConfig, extra_type_hooks: dict = {}, )
flowmap/misc/config_tools.py:16
↓ 1 callersMethodgrid_sample_features
( self, features: Float[Tensor, "batch frame channel height width"], grid: Float[Tenso
flowmap/model/backbone/backbone_midas.py:114
↓ 1 callersFunctionload_birds
(device: torch.device)
paper/point_clouds/birds.py:49
↓ 1 callersMethodload_category_sequences
(self, category_path: Path)
flowmap/dataset/dataset_co3d.py:64
↓ 1 callersFunctionload_image
( path: Path, shape: tuple[int, int] | None, )
flowmap/dataset/dataset_llff.py:27
↓ 1 callersMethodload_metadata
(path: Path)
flowmap/dataset/dataset_llff.py:105
↓ 1 callersMethodload_sequences
(self)
flowmap/dataset/dataset_co3d.py:109
↓ 1 callersFunctionmain
(input_path: Path, output_path: Path)
scripts/colmap/run_dense.py:63
↓ 1 callersFunctionmain
( input_path: Path, output_path: Path, seed: int | None, )
scripts/colmap/run_sparse.py:54
↓ 1 callersFunctionmain
( input_path: Path, output_sparse_path: Path, output_dense_path: Path, workspace_path: Path, )
scripts/colmap/run_both_at_flowmap_resolution.py:23
↓ 1 callersFunctionmain
(input_path: Path)
scripts/colmap/check_models.py:8
↓ 1 callersFunctionmake_cache
(location: Path | None)
flowmap/misc/disk_cache.py:11
↓ 1 callersFunctionmake_net
(dims)
flowmap/model/backbone/backbone_midas.py:16
↓ 1 callersFunctionoverfit
(cfg_dict: DictConfig)
flowmap/overfit.py:44
↓ 1 callersFunctionoverlay
( base: Float[Tensor, "channel base_height base_width"], overlay: Float[Tensor, "channel overlay_heigh
flowmap/visualization/layout.py:85
↓ 1 callersFunctionpretrain
(cfg_dict: DictConfig)
flowmap/pretrain.py:28
↓ 1 callersFunctionquaternion_to_matrix
( quaternions: Float[Tensor, "*batch 4"], eps: float = 1e-8, )
flowmap/model/extrinsics/extrinsics_regressed.py:17
↓ 1 callersMethodread_camera_parameters
( self, sequence: Sequence, frame_index_in_sequence: int, )
flowmap/dataset/dataset_co3d.py:170
↓ 1 callersMethodread_image
( self, sequence: Sequence, frame_index_in_sequence: int, )
flowmap/dataset/dataset_co3d.py:206
↓ 1 callersMethodrecursive_slice
(value: Any, slices: tuple[SliceLike, ...])
flowmap/misc/manipulable.py:62
↓ 1 callersFunctionreduce_straight_alpha
( rgba: Float[Tensor, "batch 4 height width"], )
flowmap/visualization/drawing/rendering.py:59
↓ 1 callersFunctionrender
( shape: tuple[int, int], color_function: ColorFunction, device: torch.device, subdivision: in
flowmap/visualization/drawing/rendering.py:118
↓ 1 callersFunctionrender_scene_flow_projection
( device: torch.device, birds: LoadedScene, background_mask: Bool[Tensor, "height width"], can
paper/render_point_cloud.py:364
↓ 1 callersFunctionround_values
Round values to the specified precision.
paper/table.py:10
↓ 1 callersFunctionrun_msaa_pass
( xy: Float[Tensor, "batch height width 2"], color_function: ColorFunction, scale: float, subd
flowmap/visualization/drawing/rendering.py:76
↓ 1 callersFunctionsubsample
(cfg_dict: DictConfig)
flowmap/subsample.py:42
↓ 1 callersFunctionsubsample_frames
( flow_predictor: FlowPredictor, full_video_path: Path, subsampled_path: Path, target_num_fram
flowmap/subsample.py:98
↓ 1 callersFunctionto_tuple
(lst)
flowmap/misc/manipulable.py:15
↓ 1 callersFunctiontransform_world2cam
Transform points from 3D world coordinates to 3D camera coordinates.
flowmap/model/projection.py:41
↓ 1 callersFunctionupdate_checkpoint_path
(path: str | None, cfg: WandbCfg)
flowmap/misc/wandb_tools.py:44
↓ 1 callersFunctionvideo_to_frames
Convert a full video to frames using ffmpeg.
flowmap/subsample.py:70
↓ 1 callersFunctionwrite_ply
( path: Path, xyz: Float[np.ndarray, "point 3"], rgb: Float[np.ndarray, "point 3"], )
flowmap/export/colmap.py:30
Method__call__
( self, xy: Float[Tensor, "*batch 2"], )
flowmap/visualization/drawing/coordinate_conversion.py:12
Method__call__
( self, xy: Float[Tensor, "point 2"], )
flowmap/visualization/drawing/rendering.py:11
Method__getitem__
(self, index: int)
flowmap/dataset/dataset_images.py:43
Method__getitem__
(self, index: int)
flowmap/dataset/dataset_co3d.py:240
Method__getitem__
(self, index: int)
flowmap/dataset/validation_wrapper.py:25
Method__getitem__
(self, index: int)
flowmap/dataset/dataset_llff.py:67
Method__getitem__
(self, index: int)
flowmap/dataset/dataset_colmap.py:64
Method__getitem__
Return a shallow copy of this instance in which all tensors, arrays, and lists have been sliced according to the specified slices. If the numb
flowmap/misc/manipulable.py:40
Method__init__
(self, cfg: T)
flowmap/flow/flow_predictor.py:28
Method__init__
(self, cfg: FlowPredictorRaftCfg)
flowmap/flow/flow_predictor_raft.py:25
Method__init__
(self, cfg: FlowPredictorGMFlowCfg)
flowmap/flow/flow_predictor_gmflow.py:28
Method__init__
(self, cfg: VisualizerTrajectoryCfg)
flowmap/visualization/visualizer_trajectory.py:60
Method__init__
(self, cfg: T)
flowmap/visualization/visualizer.py:18
Method__init__
( self, cfg: DatasetImagesCfg, stage: Stage, frame_sampler: FrameSampler,
flowmap/dataset/dataset_images.py:25
Method__init__
(self, limit: int | None = None)
flowmap/dataset/data_module_overfit.py:8
Method__init__
( self, cfg: DatasetCO3DCfg, stage: Stage, frame_sampler: FrameSampler, )
flowmap/dataset/dataset_co3d.py:50
Method__init__
( self, cfg: DatasetRE10kCfg, stage: Stage, frame_sampler: FrameSampler, )
flowmap/dataset/dataset_re10k.py:31
Method__init__
( self, dataset_cfgs: list[DatasetCfg], data_module_cfg: DataModulePretrainCfg,
flowmap/dataset/data_module_pretrain.py:40
Method__init__
(self, dataset: Dataset, length: int)
flowmap/dataset/validation_wrapper.py:16
Method__init__
( self, cfg: DatasetLLFFCfg, stage: Stage, frame_sampler: FrameSampler, )
flowmap/dataset/dataset_llff.py:52
Method__init__
( self, cfg: DatasetCOLMAPCfg, stage: Stage, frame_sampler: FrameSampler,
flowmap/dataset/dataset_colmap.py:29
Method__init__
(self, datasets: list[IterableDataset | Dataset])
flowmap/dataset/dataset_merged.py:6
Method__init__
( self, cfg: ModelWrapperOverfitCfg, model: Model, batch: Batch, flows
flowmap/model/model_wrapper_overfit.py:25
Method__init__
( self, cfg: ModelWrapperPretrainCfg, cfg_cropping: CroppingCfg, cfg_flow: Flo
flowmap/model/model_wrapper_pretrain.py:29
Method__init__
( self, cfg: ModelCfg, num_frames: int | None = None, image_shape: tuple[int,
flowmap/model/model.py:42
Method__init__
( self, cfg: BackboneMidasCfg, num_frames: int | None, image_shape: tuple[int,
flowmap/model/backbone/backbone_midas.py:43
Method__init__
( self, cfg: T, num_frames: int | None, image_shape: tuple[int, int] | None,
flowmap/model/backbone/backbone.py:23
Method__init__
( self, cfg: BackboneExplicitDepthCfg, num_frames: int | None, image_shape: tu
flowmap/model/backbone/backbone_explicit_depth.py:20
Method__init__
(self, cfg: T, num_frames: int | None)
flowmap/model/extrinsics/extrinsics.py:18
Method__init__
( self, cfg: ExtrinsicsRegressedCfg, num_frames: int, )
flowmap/model/extrinsics/extrinsics_regressed.py:48
Method__init__
(self, cfg: T)
flowmap/model/intrinsics/intrinsics.py:17
Method__init__
(self, cfg: IntrinsicsSoftminCfg)
flowmap/model/intrinsics/intrinsics_softmin.py:43
Method__init__
(self, cfg: IntrinsicsRegressedCfg)
flowmap/model/intrinsics/intrinsics_regressed.py:23
Method__init__
(self)
flowmap/misc/local_logger.py:13
Method__init__
(self, cfg: T)
flowmap/frame_sampler/frame_sampler.py:20
Method__init__
(self, cfg: T)
flowmap/tracking/track_predictor.py:24
Method__init__
(self, cfg: TrackPredictorCoTrackerCfg)
flowmap/tracking/track_predictor_cotracker.py:21
Method__init__
(self, cfg: T)
flowmap/loss/loss.py:27
Method__init__
(self, cfg: LossFlowCfg)
flowmap/loss/loss_flow.py:27
Method__init__
(self, cfg: LossTrackingCfg)
flowmap/loss/loss_tracking.py:24
Method__init__
(self, cfg: T)
flowmap/loss/mapping/mapping.py:31
Method__iter__
(self)
flowmap/dataset/data_module_overfit.py:11
Method__iter__
(self)
flowmap/dataset/dataset_re10k.py:61
Method__iter__
(self)
flowmap/dataset/dataset_merged.py:9
← previousnext →101–200 of 274, ranked by callers