Functions — github.com/CompVis/flow-poke-transformer

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

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

⨍ Functions 559◇ Types & classes 54 ↳ Endpoints 4

Functionload_idx_owm

(data_root: Path, idx: int, labels: pd.DataFrame, context_frames: int = 8, video_length: int = 80)

scripts/myriad_eval/openset_prediction.py:229

Functionload_idx_physics_iq

(data_root: Path, idx: int, labels: pd.DataFrame, context_frames: int = 8, video_length: int = 80)

scripts/myriad_eval/openset_prediction.py:198

Functionload_idx_physion

(data_root: Path, idx: int, labels: pd.DataFrame, context_frames: int = 16, video_length: int = 150)

scripts/myriad_eval/openset_prediction.py:138

Methodloss

(self, x: Float[torch.Tensor, "... d_out"])

myriad/model.py:1099

Methodloss

(self, x: torch.Tensor)

myriad/model.py:1331

Methodmake_loader

(self)

flow_poke/data.py:190

Functionmask_mod

(batch, head, q_idx, kv_idx)

myriad/model.py:76

Functionmask_mod

(batch, head, q_idx, kv_idx)

flow_poke/model.py:168

Functionmousedown

(e)

docs/static/js/jquery.event.move.js:259

Functionmouseend

(e, data)

docs/static/js/jquery.event.move.js:274

Functionmousemove

(e, data)

docs/static/js/jquery.event.move.js:270

Functionmovestart

(e)

docs/static/js/jquery.event.move.js:469

Functionmyriad_billiard

Billiard MYRIAD model

hubconf.py:44

Functionmyriad_make_train_fns

Step protocol for the Myriad MyriadStepByStep models (myriad.model).

train.py:130

Functionmyriad_openset

Open-Set MYRIAD model

hubconf.py:27

Functionnc

(c)

docs/static/js/fontawesome.all.min.js:5

Functionnoop

()

docs/static/js/jquery.event.move.js:206

Functionoffset

(element)

docs/static/js/bulma-carousel.js:123

FunctiononMove

(e)

docs/static/js/jquery.twentytwenty.js:106

FunctiononMoveEnd

()

docs/static/js/jquery.twentytwenty.js:112

FunctiononMoveStart

(e)

docs/static/js/jquery.twentytwenty.js:93

FunctionouterHeight

(element)

docs/static/js/bulma-carousel.js:140

FunctionouterWidth

(element)

docs/static/js/bulma-carousel.js:151

Functionpin_xy_to_norm

(points, *, W, H)

scripts/myriad_eval/openset_prediction.py:291

Functionposition

(element)

docs/static/js/bulma-carousel.js:162

Functionpredict

( inputs: SampleState, model: ModelState, prediction_mode: Literal["parallel", "autoregressive"] =

scripts/demo/app.py:341

Methodpredict_parallel

Predict motion for one or multiple query points in parallel. All queries see all pokes, but are mutually independent.

flow_poke/model.py:667

Functionpreprocess_image

( image_input: UInt8[np.ndarray, "h w c"] | None, inputs: SampleState, model: ModelState

scripts/demo/app.py:462

FunctionpreventDefault

(e)

docs/static/js/jquery.event.move.js:208

Functionquery_causal_mask_mod

Seq: [img_tokens, pokes, queries], len(pokes) = l_seq, len(queries) = n_query * l_seq n_query queries attend to the same set of pokes each, n

myriad/model.py:67

Functionquery_causal_mask_mod

Seq: [pokes, queries], len(pokes) = sequence_length, len(queries) = n_query * sequence_length n_query queries attend to the same set of pokes

flow_poke/model.py:160

Functionra

()

docs/static/js/jquery-3.2.1.min.js:2

Functionrc

(c)

docs/static/js/fontawesome.all.min.js:5

FunctionremoveClasses

(element, classes)

docs/static/js/bulma-carousel.js:102

Functionresample_generated_to_conditioning

(conditioning: np.ndarray, generated: np.ndarray)

scripts/myriad_eval/openset_prediction.py:301

Methodreset_kv_cache

(self, b: int, l: int)

myriad/model.py:814

Methodrsample

(self, sample_shape=torch.Size())

myriad/model.py:1103

Methodrsample

(self, sample_shape=torch.Size())

myriad/model.py:1334

Functionshow

(elements)

docs/static/js/bulma-carousel.js:109

Functiontouchstart

(e)

docs/static/js/jquery.event.move.js:283

Functiontrain_2d

Train on 2-D tracker shards (myriad.data_2d.TrackerShardsDataModule).

train.py:507

Functiontrain_billiards

Train on procedurally generated billiards data (myriad.data_billiards.BilliardSimDataModule).

train.py:545

Methodtrain_dataloader

(self)

myriad/data_billiards.py:387

Functiontrain_fpt

Train the original FlowPokeTransformer (flow_poke.model).

train.py:481

Functionupdate

(time)

docs/static/js/jquery.event.move.js:497

Functionuuid

()

docs/static/js/bulma-carousel.js:985

Functionv

(a,b)

docs/static/js/jquery-3.2.1.min.js:2

Methodval_dataloader

(self)

myriad/data_2d.py:505

Functionvc

()

docs/static/js/fontawesome.all.min.js:5

Functionvit_base

(patch_size=14, num_register_tokens=0, **kwargs)

flow_poke/dinov2.py:289

Functionvit_giant2

(patch_size=14, num_register_tokens=0, **kwargs)

flow_poke/dinov2.py:313

Functionvit_large

(patch_size=14, num_register_tokens=0, **kwargs)

flow_poke/dinov2.py:301

Functionvit_small

(patch_size=14, num_register_tokens=0, **kwargs)

flow_poke/dinov2.py:277

Functionwarmup

(*args, **kwargs)

scripts/demo/app.py:413

Functionwds_filter

(sample: dict | None)

scripts/data/test_shard.py:92

Functionwds_filter

(sample: dict | None)

scripts/data/track_shard.py:256

Functionwidth

(element)

docs/static/js/bulma-carousel.js:132

Functionworker_init_fn

Function to seed all underlying libraries in the workers

myriad/data_billiards.py:40

Functionwrapped_fn

(*args, **kwargs)

scripts/myriad_eval/qual.py:244

← previous501–559 of 559, ranked by callers