MCPcopy Create free account

hub / github.com/CompVis/flow-poke-transformer / functions

Functions559 in github.com/CompVis/flow-poke-transformer

⨍ Functions 559◇ Types & classes 54↳ Endpoints 4
↓ 1 callersMethod__init__
( self, model_size: int=256, model_version: str="dinov3_vitl16", gradient_last
myriad/dinov3.py:19
↓ 1 callersMethod_build_targets
(self, sample: dict[str, torch.Tensor])
myriad/data_2d.py:285
↓ 1 callersMethod_bvn_cdf
(a, b, rho, n_nodes: int)
myriad/model.py:1220
↓ 1 callersMethod_bvn_cdf
(a, b, rho, n_nodes: int)
flow_poke/model.py:404
↓ 1 callersMethod_center_crop
( self, x: Float[torch.Tensor, "c h w"], pos: Float[torch.Tensor, "t l 2"], ce
myriad/data_2d.py:159
↓ 1 callersFunction_defineProperty
(obj, key, value)
docs/static/js/bulma-carousel.js:475
↓ 1 callersMethod_fourier_emb_flow
(self, flow: Float[torch.Tensor, "b l 2"])
myriad/model.py:418
↓ 1 callersMethod_fwd_1
( self, x: Float[torch.Tensor, "b l d_model"], # this is fused x + x_cross theta: Flo
myriad/model.py:560
↓ 1 callersMethod_fwd_1
( self, x: Float[torch.Tensor, "b *DIMS C"], x_cross: Float[torch.Tensor, "b l_c d_cro
myriad/model.py:706
↓ 1 callersMethod_gauss_legendre_nodes_weights
(n: int, device, dtype)
myriad/model.py:1206
↓ 1 callersMethod_gauss_legendre_nodes_weights
(n: int, device, dtype)
flow_poke/model.py:390
↓ 1 callersMethod_get_distribution
(self, x_out: Float[torch.Tensor, "b l d_model"])
flow_poke/model.py:645
↓ 1 callersFunction_inherits
(subClass, superClass)
docs/static/js/bulma-slider.js:94
↓ 1 callersMethod_load_frame
(self, video, i_frame: int)
myriad/data_2d.py:134
↓ 1 callersFunction_possibleConstructorReturn
(self, call)
docs/static/js/bulma-slider.js:92
↓ 1 callersFunctionactiveTouchend
(e, data)
docs/static/js/jquery.event.move.js:422
↓ 1 callersFunctionactiveTouchmove
(e, data)
docs/static/js/jquery.event.move.js:405
↓ 1 callersMethodadd_point_ui
(self, point: tuple[float, float])
scripts/demo/app.py:36
↓ 1 callersFunctionadjustContainer
(offset)
docs/static/js/jquery.twentytwenty.js:52
↓ 1 callersFunctionaugment
( clip, interpolation=TVT.InterpolationMode.BICUBIC, size=512, )
myriad/data_2d.py:30
↓ 1 callersFunctionaugment
( clip, interpolation=TVT.InterpolationMode.BICUBIC, size=512, center_crop=True, )
scripts/data/track_shard.py:51
↓ 1 callersFunctionaugment
(clip, size=512)
flow_poke/data.py:48
↓ 1 callersFunctionbenchmark
( *, data_root: str, dataset_name: str, model, ensemble_size: int = 5, get_annotation_
scripts/myriad_eval/openset_prediction.py:398
↓ 1 callersFunctionbetter_resize
(imgs: torch.Tensor, image_size: int)
flow_poke/dinov2.py:363
↓ 1 callersFunctionbounding_box
( h: int, w: int, pixel_aspect_ratio: float = 1.0, )
myriad/model.py:141
↓ 1 callersFunctionbounding_box
(h, w, pixel_aspect_ratio=1.0)
flow_poke/model.py:119
↓ 1 callersMethodbuild_targets
(self, sample: dict[str, torch.Tensor])
flow_poke/data.py:158
↓ 1 callersMethodbuild_timestep_cache
(self, steps: int, dims: tuple[int, ...], device: torch.device, dtype: torch.dtype)
myriad/model.py:1069
↓ 1 callersFunctioncalcOffset
(dimensionPct)
docs/static/js/jquery.twentytwenty.js:40
↓ 1 callersFunctioncli
Flow Poke Transformer / Myriad training.
train.py:472
↓ 1 callersFunctioncompute_step
(batch, block_mask, is_query, L_poke, compute_metrics=False, flow_mask=None)
train.py:93
↓ 1 callersFunctioncreateEvent
(type)
docs/static/js/jquery.event.move.js:100
↓ 1 callersFunctioncreate_sample
( loader: torch.utils.data.DataLoader, dataset: BilliardSimDataset, use_random_goal: bool, act
scripts/myriad_eval/billiard_planning.py:72
↓ 1 callersFunctioneb
()
docs/static/js/jquery-3.2.1.min.js:3
↓ 1 callersMethodembed_camera_static
(self, camera_static: Bool[torch.Tensor, "b"])
myriad/model.py:438
↓ 1 callersFunctionendless_iter
(iterable)
train.py:25
↓ 1 callersFunctionevaluate_for_idx_generic
(row, annotations, pred_sim, query_splits, cut_video_to_len)
scripts/myriad_eval/openset_prediction.py:376
↓ 1 callersMethodextract_training_sample
( self, sample: dict[str, torch.Tensor] )
myriad/data_2d.py:207
↓ 1 callersMethodextract_training_sample
(self, sample: dict[str, torch.Tensor])
flow_poke/data.py:95
↓ 1 callersMethodforward
( self, x: Float[torch.Tensor, "... l d"], cond: Float[torch.Tensor, "... d_cond"] )
flow_poke/model.py:36
↓ 1 callersMethodforward_features
(self, imgs: Float[torch.Tensor, "B C H W"], masks)
myriad/dinov3.py:70
↓ 1 callersFunctionfree_mem
()
scripts/myriad_eval/openset_prediction.py:36
↓ 1 callersMethodfwd_cross
(self, x, x_cross, theta, theta_cross)
flow_poke/model.py:301
↓ 1 callersMethodfwd_ffn
(self, x: Float[torch.Tensor, "... d_model"])
flow_poke/model.py:309
↓ 1 callersMethodfwd_self
( self, x: Float[torch.Tensor, "b l d"], theta, mask: BlockMask | Literal["cau
flow_poke/model.py:269
↓ 1 callersFunctiongc
(c,l)
docs/static/js/fontawesome.all.min.js:5
↓ 1 callersFunctionget_annotated_video
(*, data_root: Path, row, get_extra_query_pts, override_cache = False, **kwargs)
scripts/myriad_eval/openset_prediction.py:320
↓ 1 callersFunctionget_batch
(img: Image.Image, pts: torch.Tensor, num_traj: int, num_steps: int)
scripts/myriad_eval/qual.py:223
↓ 1 callersMethodget_camera_static
(self, sample: dict[str, torch.Tensor])
flow_poke/data.py:152
↓ 1 callersMethodget_camera_static_tensor
(self, flow: Float[torch.Tensor, "t n_t 2"])
myriad/data_2d.py:128
↓ 1 callersFunctionget_colors
Gets colormap for points.
scripts/data/test_shard.py:109
↓ 1 callersFunctionget_data
(duration: int)
scripts/myriad_eval/billiard_planning.py:44
↓ 1 callersFunctionget_example
(example_name: str, mode: str)
scripts/myriad_eval/qual.py:194
↓ 1 callersMethodget_image
(self)
scripts/myriad_eval/qual.py:46
↓ 1 callersMethodget_pokes_queries
(self, mode='fully_poked')
scripts/myriad_eval/qual.py:66
↓ 1 callersFunctionget_tapnext_cache
(data_root: Path, video_path: Path)
scripts/myriad_eval/openset_prediction.py:305
↓ 1 callersMethodgrow_kv_cache
(self, b: int, l: int)
myriad/model.py:819
↓ 1 callersMethodgrow_kv_cache
(self, b: int, l: int)
flow_poke/model.py:357
↓ 1 callersFunctionhandler
(e)
docs/static/js/jquery.event.move.js:115
↓ 1 callersFunctionhb
(a,b,c)
docs/static/js/jquery-3.2.1.min.js:3
↓ 1 callersFunctioninference_query_causal_mask_mod
Seq: [pokes, queries], len(pokes) = sequence_length, len(queries) = * Pokes attend causally to each other, but not to queries Queries att
flow_poke/model.py:182
↓ 1 callersFunctioninit_caches
(batch)
train.py:73
↓ 1 callersFunctioninit_cotracker
( repo_url: str = "facebookresearch/co-tracker", model_name: str = "cotracker3_offline", device: t
scripts/data/track_shard.py:29
↓ 1 callersMethodinterpolate_pos_encoding
(self, x, w, h)
flow_poke/dinov2.py:211
↓ 1 callersFunctionisHTML5
(unknown)
docs/static/js/bulma-carousel.js:255
↓ 1 callersFunctionisIgnoreTag
(e)
docs/static/js/jquery.event.move.js:212
↓ 1 callersFunctionisPrimaryButton
(e)
docs/static/js/jquery.event.move.js:216
↓ 1 callersFunctionisString
(unknown)
docs/static/js/bulma-slider.js:98
↓ 1 callersFunctionisVimeo
(unknown)
docs/static/js/bulma-carousel.js:264
↓ 1 callersFunctionisYoutube
(unknown)
docs/static/js/bulma-carousel.js:261
↓ 1 callersFunctionj
()
docs/static/js/jquery-3.2.1.min.js:2
↓ 1 callersFunctionjb
(a,b)
docs/static/js/jquery-3.2.1.min.js:3
↓ 1 callersFunctionkb
(a,b,c)
docs/static/js/jquery-3.2.1.min.js:3
↓ 1 callersFunctionload_mp4
( data, min_sequence_length=0, select_args={}, augment_args={}, frame_skip=0, max_sequ
scripts/data/track_shard.py:70
↓ 1 callersFunctionload_mpeg
( data: bytes, key: str, )
scripts/data/test_shard.py:24
↓ 1 callersFunctionma
(a)
docs/static/js/jquery-3.2.1.min.js:2
↓ 1 callersFunctionmain
( checkpoint_path: str, duration: float = 1.0, method: str = "ours", out_dir: str = "eval_resu
scripts/myriad_eval/billiard_planning.py:600
↓ 1 callersFunctionmain
( data_root: Path, ckpt_path: Path, dataset_name: str, out_path: Path, ensemble_size: int,
scripts/myriad_eval/openset_prediction.py:461
↓ 1 callersFunctionmain
( checkpoint_path: str, example_name: str = "falling_ball", mode: str = "first_query", num_ste
scripts/myriad_eval/qual.py:360
↓ 1 callersFunctionmake_axial_pos_2d
( h: int, w: int, pixel_aspect_ratio=1.0, align_corners=False, dtype=None, device=None, relative_pos=True
flow_poke/model.py:137
↓ 1 callersFunctionmake_grid
( h_pos: Float[torch.Tensor, "(h)"], w_pos: Float[torch.Tensor, "(w)"], )
myriad/model.py:168
↓ 1 callersFunctionmake_grid
(h_pos, w_pos)
flow_poke/model.py:113
↓ 1 callersFunctionmake_scheduler
Build a LR scheduler. scheduler_type="linear": linear warmup → linear decay to 0 scheduler_type="cosine": linear warmup → cosine annealing
train.py:30
↓ 1 callersFunctionmask_pad
( mask: Bool[torch.Tensor, "..."], raw: Float[torch.Tensor, "..."], replace: Float[torch.Tensor, "
myriad/model.py:355
↓ 1 callersFunctionminMaxNumber
(num, min, max)
docs/static/js/jquery.twentytwenty.js:71
↓ 1 callersFunctionmyriad_predict_function_with_past_cond
(row, model: MyriadStepByStep, ensemble_size, gt_tracks, traj_kind, dt_scaling_factor, context_length=1, **kwa
scripts/myriad_eval/openset_prediction.py:71
↓ 1 callersFunctiono
(t)
docs/static/js/bulma-slider.min.js:1
↓ 1 callersFunctionpaint_point_track
Converts a sequence of points to color code video. Args: frames: [num_frames, height, width, 3], np.uint8, [0, 255] point_tracks: [num
scripts/data/test_shard.py:125
↓ 1 callersFunctionpl
(c)
docs/static/js/fontawesome.all.min.js:5
↓ 1 callersFunctionplan_action_ensembled
( model: MyriadStepByStep, pred_function: types.FunctionType, batch: dict[str, torch.Tensor],
scripts/myriad_eval/billiard_planning.py:409
↓ 1 callersMethodpredict_autoregressive
Sample motion for multiple (or one) query points in autoregressive manner. All queries see all pokes and the sampled motion for prece
flow_poke/model.py:705
↓ 1 callersFunctionpredict_distribution
(inputs: SampleState, model: ModelState)
scripts/demo/app.py:163
↓ 1 callersFunctionpredict_flow
( inputs: SampleState, model: ModelState, prediction_mode: Literal["parallel", "autoregressive"],
scripts/demo/app.py:82
↓ 1 callersFunctionprepare_points_for_tapnext
(points)
scripts/myriad_eval/openset_prediction.py:287
↓ 1 callersMethodprepare_tokens_with_masks
(self, x, masks=None)
flow_poke/dinov2.py:245
↓ 1 callersFunctionremoveActiveMouse
()
docs/static/js/jquery.event.move.js:400
↓ 1 callersFunctionremoveActiveTouch
(data)
docs/static/js/jquery.event.move.js:435
↓ 1 callersFunctionremoveMouse
()
docs/static/js/jquery.event.move.js:278
↓ 1 callersFunctionremoveTouch
(data)
docs/static/js/jquery.event.move.js:321
↓ 1 callersFunctionrender_input
(inputs: SampleState)
scripts/demo/app.py:44
← previousnext →201–300 of 559, ranked by callers