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
Method__init__
( self, # Transformer parameters width: int, depth: int, layer_params:
flow_poke/model.py:567
Method__iter__
(self)
myriad/data_billiards.py:103
Method__iter__
(self)
myriad/data_billiards.py:275
Function_a
(a,b,c,d,e)
docs/static/js/jquery-3.2.1.min.js:3
Method_decode
(self, sample: dict[str, bytes])
myriad/data_2d.py:313
Method_filter_valid
(self, sample: dict[str, torch.Tensor])
myriad/data_2d.py:431
Function_map_many
Version of wds.map() that .
myriad/data_2d.py:45
Function_ours_pred_function
( model: MyriadStepByStep, x: Float[torch.Tensor, "b h w c"], pos_orig: Float[torch.Tensor, "b n 2
scripts/myriad_eval/billiard_planning.py:137
Functionaa
(a,b,c)
docs/static/js/jquery-3.2.1.min.js:2
FunctionactiveMouseend
(e, data)
docs/static/js/jquery.event.move.js:384
FunctionactiveMousemove
(e, data)
docs/static/js/jquery.event.move.js:376
Functionadd
(handleObj)
docs/static/js/jquery.event.move.js:538
FunctionaddClasses
(element, classes)
docs/static/js/bulma-carousel.js:95
Methodapply_emb
(self, x, theta)
myriad/model.py:253
Functionbl
(c)
docs/static/js/fontawesome.all.min.js:5
FunctionbulmaCarousel
(selector)
docs/static/js/bulma-carousel.js:504
FunctionbulmaSlider
(selector)
docs/static/js/bulma-slider.js:105
Functionca
(a,b)
docs/static/js/jquery-3.2.1.min.js:2
Functioncamelize
(str)
docs/static/js/bulma-carousel.js:1008
Functioncommon_options
(func)
train.py:443
Methodconstruct_sample
(sample: dict[str, bytes])
flow_poke/data.py:191
Functioncss
(element, obj)
docs/static/js/bulma-carousel.js:169
Functiondecode_mp4
( sample:dict[str, any], **kwargs, )
scripts/data/track_shard.py:117
Functiondecode_mpeg
( sample:dict[str, any], **kwargs, )
scripts/data/test_shard.py:45
Functiondefer
()
docs/static/js/bulma-carousel.js:995
Functiondemo
( checkpoint: str | None = None, # Will be automatically downloaded from torch.hub() if not specified
scripts/demo/app.py:364
FunctiondetectSupportsPassive
()
docs/static/js/bulma-carousel.js:189
Functiondict_collation_fn
Take a list of samples (as dictionary) and create a batch, preserving the keys. If `tensors` is True, `ndarray` objects are combined into ten
myriad/data_billiards.py:48
Functiondict_collation_fn
(samples, combine_tensors=True, combine_scalars=True, **kwargs)
scripts/data/test_shard.py:69
Functiondict_collation_fn
(samples, combine_tensors=True, combine_scalars=True, **kwargs)
scripts/data/track_shard.py:142
Functiondict_collation_fn
(samples, combine_tensors=True, combine_scalars=True, **kwargs)
flow_poke/data.py:18
Functiondict_join_collation_fn
(samples: list[dict[str, list | torch.Tensor]])
myriad/data_billiards.py:80
Functiondinov3_base
()
myriad/dinov3.py:112
Functiondinov3_large
()
myriad/dinov3.py:102
Functiondinov3_small
()
myriad/dinov3.py:122
Functionec
(c)
docs/static/js/fontawesome.all.min.js:5
Methodembed_image
Get the image features, pass to forward() etc. Image is expected to be square and in [-1, 1]. If the resolution is not the expected o
flow_poke/model.py:604
Functionembedder_mlp
(in_features: int, out_features: int, depth: int)
myriad/model.py:339
FunctionenableMove1
(e)
docs/static/js/jquery.event.move.js:534
FunctionenableMove2
(e)
docs/static/js/jquery.event.move.js:535
FunctionenableMove3
(e)
docs/static/js/jquery.event.move.js:536
Methodextra_repr
(self)
myriad/model.py:286
Methodextra_repr
(self)
myriad/model.py:303
Functionfl
(c)
docs/static/js/fontawesome.all.min.js:5
Methodforward
(self, imgs: Float[torch.Tensor, "B C H W"])
myriad/dinov3.py:79
Methodforward
(self, imgs: Float[torch.Tensor, "b c h w"])
myriad/dinov3.py:191
Methodforward
(self, x: torch.Tensor)
myriad/model.py:54
Methodforward
(self, x, *args, **kwargs)
myriad/model.py:58
Methodforward
(self, input)
myriad/model.py:131
Methodforward
( self, pos: Float[torch.Tensor, "... 2"], pos_orig: Float[torch.Tensor, "
myriad/model.py:257
Methodforward
(self, x)
myriad/model.py:289
Methodforward
(self, x: Float[torch.Tensor, "... l d"], scale: Float[torch.Tensor, "... d"])
myriad/model.py:306
Methodforward
(self, x, cond)
myriad/model.py:323
Methodforward
( self, x: Float[torch.Tensor, "b l 2"], pos: Float[torch.Tensor, "b l 2"], po
myriad/model.py:451
Methodforward
(self, x: Float[torch.Tensor, "... width"])
myriad/model.py:513
Methodforward
( self, x: Float[torch.Tensor, "b l d_model"], # this is fused x + x_cross theta: Flo
myriad/model.py:592
Methodforward
( self, x: Float[torch.Tensor, "b *DIMS C"], x_cross: Float[torch.Tensor, "b l_c d_cro
myriad/model.py:767
Methodforward
(self, x, scale, shift)
myriad/model.py:850
Methodforward
( self, x: Float[torch.Tensor, "... d_out"], cond: Float[torch.Tensor, "... d_cond"],
myriad/model.py:935
Methodforward
( self, cond: Float[torch.Tensor, "... d_cond"], steps: int | None = None, cfg
myriad/model.py:1154
Methodforward
(self, cond: Float[torch.Tensor, "... d_model"])
myriad/model.py:1181
Methodforward
(self, cond: Float[torch.Tensor, "... d_model"])
myriad/model.py:1291
Methodforward
(self, cond: Float[torch.Tensor, "... d_model"])
myriad/model.py:1373
Methodforward
(self, x)
flow_poke/dinov2.py:33
Methodforward
(self, x)
flow_poke/dinov2.py:48
Methodforward
(self, x)
flow_poke/dinov2.py:64
Methodforward
(self, x)
flow_poke/dinov2.py:80
Methodforward
(self, x)
flow_poke/dinov2.py:102
Methodforward
(self, x)
flow_poke/dinov2.py:140
Methodforward
(self, x, is_training=False, masks=None)
flow_poke/dinov2.py:272
Methodforward
(self, imgs: Float[torch.Tensor, "b c h w"])
flow_poke/dinov2.py:424
Methodforward
(self, x: torch.Tensor)
flow_poke/model.py:43
Methodforward
(self, pos: Float[torch.Tensor, "b l 2"])
flow_poke/model.py:101
Methodforward
( self, x: Float[torch.Tensor, "b l d_model"], x_cross: Float[torch.Tensor, "b l' d_cr
flow_poke/model.py:313
Methodforward
( self, x: Float[torch.Tensor, "b l d_in"], pos: Float[torch.Tensor, "b l 2"],
flow_poke/model.py:339
Methodforward
( self, pos: Float[torch.Tensor, "b l 2"], flow: Float[torch.Tensor, "b l d_flow"],
flow_poke/model.py:512
Functionfpt_base
Standard Flow Poke Transformer.
hubconf.py:10
Functionfpt_make_train_fns
Step protocol for the original FlowPokeTransformer (flow_poke.model).
train.py:69
FunctiongetNodeIndex
(node)
docs/static/js/bulma-carousel.js:1005
Functionget_color
( hue: float, # [0,1] t_rel: int, # [0,N] N is `norm_scale` norm_scale: int, min_intensity:
scripts/data/test_shard.py:97
Functionget_extra_query_pts_default
( row, **kwargs, )
scripts/myriad_eval/openset_prediction.py:254
Methodget_pokes
(self, mode='fully_poked')
scripts/myriad_eval/qual.py:52
Functiongl
(c)
docs/static/js/fontawesome.all.min.js:5
Functionheight
(element)
docs/static/js/bulma-carousel.js:136
Functionhide
(elements)
docs/static/js/bulma-carousel.js:116
Functionhl
(c)
docs/static/js/fontawesome.all.min.js:5
Functionib
(a,b,c)
docs/static/js/jquery-3.2.1.min.js:3
Functionic
(c)
docs/static/js/fontawesome.all.min.js:5
Functioninput_click
(image_input, inputs: SampleState, evt: gr.SelectData)
scripts/demo/app.py:499
FunctionisDate
(unknown)
docs/static/js/bulma-carousel.js:234
FunctionisEmptyObject
(unknown)
docs/static/js/bulma-carousel.js:240
FunctionisFunction
(unknown)
docs/static/js/bulma-carousel.js:225
FunctionisInViewport
(element, html)
docs/static/js/bulma-carousel.js:1481
FunctionisNumber
(unknown)
docs/static/js/bulma-carousel.js:228
FunctionisObject
(unknown)
docs/static/js/bulma-carousel.js:237
FunctionisRtl
()
docs/static/js/bulma-carousel.js:991
FunctionisString
(unknown)
docs/static/js/bulma-carousel.js:231
FunctionisVideo
(unknown)
docs/static/js/bulma-carousel.js:252
Functionjc
()
docs/static/js/fontawesome.all.min.js:5
Functionl
(t,e)
docs/static/js/bulma-carousel.min.js:1
← previousnext →401–500 of 559, ranked by callers