MCPcopy Create free account

hub / github.com/dcharatan/flowmap / functions

Functions274 in github.com/dcharatan/flowmap

⨍ Functions 274◇ Types & classes 102↳ Endpoints 4
Method__len__
(self)
flowmap/dataset/dataset_images.py:56
Method__len__
(self)
flowmap/dataset/dataset_co3d.py:237
Method__len__
(self)
flowmap/dataset/dataset_re10k.py:175
Method__len__
(self)
flowmap/dataset/validation_wrapper.py:22
Method__len__
(self)
flowmap/dataset/dataset_llff.py:100
Method__len__
(self)
flowmap/dataset/dataset_colmap.py:82
Method__len__
(self)
flowmap/dataset/dataset_merged.py:30
Function_get_coord_info_new
(self, renderer)
paper/generate_trajectory_plots.py:54
Functioncache
(key: Any, fallback: Callable[[], T])
flowmap/misc/disk_cache.py:12
Functioncolor_function
( xy: Float[Tensor, "point 2"], )
flowmap/visualization/drawing/points.py:40
Methodcompute_unweighted_loss
( self, batch: Batch, flows: Flows, tracks: list[Tracks] | None, model
flowmap/loss/loss_flow.py:31
Methodcompute_unweighted_loss
( self, batch: Batch, flows: Flows, tracks: list[Tracks] | None, model
flowmap/loss/loss_tracking.py:28
Methodconfigure_optimizers
(self)
flowmap/model/model_wrapper_overfit.py:104
Methodconfigure_optimizers
(self)
flowmap/model/model_wrapper_pretrain.py:110
Functionconvert_pixel_to_world
( xy: Float[Tensor, "*batch 2"], )
flowmap/visualization/drawing/coordinate_conversion.py:39
Functionconvert_world_to_pixel
( xy: Float[Tensor, "*batch 2"], )
flowmap/visualization/drawing/coordinate_conversion.py:34
Functioncrop_and_save
( image: Image.Image, path: Path, ratio: float = ASPECT_RATIO, )
paper/generate_splat_comparison.py:115
Methoddata_stage
(self)
flowmap/dataset/dataset_re10k.py:148
Functionencode_image
( image: Float[Tensor, "3 height width"], image_format: Literal["png", "jpeg"] = "png", )
paper/svg_tools.py:15
Methodexport
( self, batch: Batch, flows: Flows, global_step: int, )
flowmap/model/model.py:93
Methodforward
( self, videos: Float[Tensor, "batch frame 3 height width"], )
flowmap/flow/flow_predictor_raft.py:29
Methodforward
( self, videos: Float[Tensor, "batch frame 3 height width"], )
flowmap/flow/flow_predictor_gmflow.py:66
Methodforward
(self, batch: Batch, flows: Flows)
flowmap/model/backbone/backbone_midas.py:72
Methodforward
(self, batch: Batch, flows: Flows)
flowmap/model/backbone/backbone.py:35
Methodforward
(self, batch: Batch, flows: Flows)
flowmap/model/backbone/backbone_explicit_depth.py:34
Methodforward
( self, batch: Batch, flows: Flows, backbone_output: BackboneOutput, s
flowmap/model/extrinsics/extrinsics_procrustes.py:23
Methodforward
( self, batch: Batch, flows: Flows, backbone_output: BackboneOutput, s
flowmap/model/extrinsics/extrinsics.py:24
Methodforward
( self, batch: Batch, flows: Flows, backbone_output: BackboneOutput, s
flowmap/model/extrinsics/extrinsics_regressed.py:65
Methodforward
( self, batch: Batch, flows: Flows, backbone_output: BackboneOutput, g
flowmap/model/intrinsics/intrinsics.py:22
Methodforward
( self, batch: Batch, flows: Flows, backbone_output: BackboneOutput, g
flowmap/model/intrinsics/intrinsics_ground_truth.py:19
Methodforward
( self, batch: Batch, flows: Flows, backbone_output: BackboneOutput, g
flowmap/model/intrinsics/intrinsics_softmin.py:63
Methodforward
( self, batch: Batch, flows: Flows, backbone_output: BackboneOutput, g
flowmap/model/intrinsics/intrinsics_regressed.py:32
Methodforward
( self, videos: Float[Tensor, "batch frame 3 height width"], query_frame: int, )
flowmap/tracking/track_predictor_cotracker.py:27
Methodforward
( self, a: Float[Tensor, "*#batch 2"], b: Float[Tensor, "*#batch 2"], image_sh
flowmap/loss/mapping/mapping.py:35
Methodforward_undistorted
( self, delta: Float[Tensor, "*batch 2"], )
flowmap/loss/mapping/mapping_huber.py:19
Methodforward_undistorted
( self, delta: Float[Tensor, "*batch 2"], )
flowmap/loss/mapping/mapping_l2.py:16
Methodforward_undistorted
( self, delta: Float[Tensor, "*batch 2"], )
flowmap/loss/mapping/mapping_l1.py:16
Functionhomogenize_vectors
Convert batched vectors (xyz) to (xyz0).
flowmap/model/projection.py:18
Methodindex
(self)
flowmap/dataset/dataset_re10k.py:156
Functioninit_weights_normal
(m)
flowmap/model/backbone/backbone_midas.py:17
Methodlog_hyperparams
(self, params)
flowmap/misc/local_logger.py:27
Methodname
(self)
flowmap/misc/local_logger.py:19
Functionproject
( points: Float[Tensor, "*#batch dim"], extrinsics: Float[Tensor, "*#batch dim+1 dim+1"], intrinsi
flowmap/model/projection.py:61
Functionproject
( points: Float[Tensor, "*batch xyz=3"], midpoint: Float[Tensor, "xyz=3"], scale: float, canva
paper/render_point_cloud.py:83
Functionread_ply
(path: Path)
flowmap/export/colmap.py:18
Functionrender_background_scene_flow
( device: torch.device, birds: LoadedScene, proj: Callable, background_mask: Bool[Tensor, "hei
paper/render_point_cloud.py:170
Functionrender_depths
(birds: LoadedScene)
paper/render_point_cloud.py:448
Functionrender_flows
(device: torch.device, birds: LoadedScene)
paper/render_point_cloud.py:454
Functionrender_frustums
( birds: LoadedScene, proj: Callable, )
paper/render_point_cloud.py:110
Functionrender_highlighted_scene_flow
( birds: LoadedScene, proj: Callable, )
paper/render_point_cloud.py:233
Functionrender_joint_point_cloud
( device: torch.device, birds: LoadedScene, proj: Callable, )
paper/render_point_cloud.py:471
Functionrender_point_cloud_silhouettes
( device: torch.device, birds: LoadedScene, proj: Callable, )
paper/render_point_cloud.py:330
Functionrender_point_clouds
( device: torch.device, birds: LoadedScene, proj: Callable, )
paper/render_point_cloud.py:140
Functionresize
( image: Float[Tensor, "channel height width"], shape: Optional[tuple[int, int]] = None, width: Op
flowmap/visualization/layout.py:217
Methodsample
( self, num_frames_in_video: int, device: torch.device, )
flowmap/frame_sampler/frame_sampler_pretrain.py:18
Methodsample
( self, num_frames_in_video: int, device: torch.device, )
flowmap/frame_sampler/frame_sampler_overfit.py:20
Functionseparate_fn
(joined: dict)
flowmap/misc/config_tools.py:45
Functionseparate_fn
(joined: dict)
flowmap/config/tools.py:45
Functionseparate_multiple_defaults
Return a function that will pull individual configurations out of a merged dict. For example, the merged dict might look like this: {
flowmap/misc/config_tools.py:28
Functionto_hex
(color: Float[Tensor, "3"])
paper/colors.py:38
Methodtrain_dataloader
(self)
flowmap/dataset/data_module_overfit.py:25
Methodtrain_dataloader
(self)
flowmap/dataset/data_module_pretrain.py:63
Methodtraining_step
(self, dummy)
flowmap/model/model_wrapper_overfit.py:51
Methodtraining_step
(self, batch)
flowmap/model/model_wrapper_pretrain.py:60
Functiontransform_cam2world
Transform points from 3D camera coordinates to 3D world coordinates.
flowmap/model/projection.py:33
Methodunnormalized_focal_lengths
( self, image_shape: tuple[int, int], )
flowmap/model/intrinsics/intrinsics_softmin.py:143
Methodval_dataloader
(self)
flowmap/dataset/data_module_overfit.py:28
Methodval_dataloader
(self)
flowmap/dataset/data_module_pretrain.py:75
Methodvalidation_step
(self, dummy)
flowmap/model/model_wrapper_overfit.py:75
Methodvalidation_step
(self, batch)
flowmap/model/model_wrapper_pretrain.py:84
Methodversion
(self)
flowmap/misc/local_logger.py:23
Methodvisualize
( self, batch: Batch, flows: Flows, tracks: list[Tracks] | None, model
flowmap/visualization/visualizer_trajectory.py:64
Methodvisualize
( self, batch: Batch, flows: Flows, tracks: list[Tracks] | None, model
flowmap/visualization/visualizer_summary.py:43
Functionworker_init_fn
(worker_id: int)
flowmap/dataset/data_module_pretrain.py:34
← previous201–274 of 274, ranked by callers