Functions — github.com/NVlabs/RADIO

hub / github.com/NVlabs/RADIO / functions

Functions572 in github.com/NVlabs/RADIO

⨍ Functions 572◇ Types & classes 107

Method__init__

(self, a, b, kernel_size=1, stride=1, padding=None, g=1, dilation=1, bn_weight_init=1, bias=False, act=True)

radio/conv.py:28

Method__init__

( self, input_size: int, hidden_size: int, output_size: int, num_inner

radio/adaptor_attn.py:23

Method__init__

(self, main_config: Namespace, adaptor_config: dict_t, state: state_t)

radio/open_clip_adaptor.py:20

Method__init__

(self, input_scale: float, norm_mean: norm_t, norm_std: nor

radio/input_conditioner.py:18

Method__init__

( self, requires_summary_and_spatial: bool, handles_summary_and_spatial: bool = False

radio/adaptor_base.py:36

Method__init__

( self, radio_model: torch.nn.Module, **kwargs, )

examples/position_predictor.py:307

Method__init__

(self, model)

examples/count_params.py:45

Method__init__

(self, dino_model: nn.Module, patch_attn: bool = True)

examples/common/model_loader.py:37

Method__init__

(self, clip_model: nn.Module, tokenizer, adaptor_name: str, clip_mode: bool = False)

examples/common/model_loader.py:69

Method__init__

(self, clip_model, tokenizer, adaptor_name, clip_mode = False)

examples/common/model_loader.py:130

Method__init__

(self, clip_model, tokenizer, proc, adaptor_name, clip_mode = False, patch_size: int = 16, is_dynamic: bool =

examples/common/model_loader.py:144

Method__init__

(self, model: nn.Module, tokenizer)

examples/common/model_loader.py:231

Method__init__

(self, model)

examples/common/model_loader.py:286

Method__init__

(self, size: Iterable[int], resize_multiple: int = 1, max_dim: bool = False)

examples/common/resize_transform.py:25

Method__init__

(self, pad_mean = None)

examples/common/resize_transform.py:84

Method__init__

(self, num_ops: int = 3, magnitude: int = 9, p: float = 1.0, mode="all", cut_out = False, **kwargs)

examples/common/rand_augment.py:74

Method_check_input

(self, inpt: Any)

examples/common/resize_transform.py:20

Function_collate

(samples: List[torch.Tensor])

examples/position_predictor.py:289

Method_enter_blocks

(self, _, input: List[torch.Tensor])

radio/vitdet.py:95

Method_enter_model

(self, _, input: List[torch.Tensor])

radio/vitdet.py:92

Method_exit_model

(self, _, inputs: List[torch.Tensor], patches: torch.Tensor)

radio/vitdet.py:134

Function_forward_cpe_dinov2

(self: DinoWrapper, x: torch.Tensor)

radio/enable_cpe_support.py:80

Function_forward_intermediates_cpe_dinov2

(self: DinoWrapper, *args, **kwargs)

radio/enable_cpe_support.py:86

Method_get_params

(self, flat_inputs: List[Any])

examples/position_predictor.py:240

Method_get_params

(self, flat_inputs: List[Any])

examples/common/resize_transform.py:16

Function_hook

(module: Attention, input: Tuple[torch.Tensor], output: torch.Tensor)

examples/model_stats.py:124

Method_init_weights

(self, m)

radio/eradio_model.py:1151

Function_load_dino_v2

(dino_v2_model, cache_dir: Optional[str] = None, pretrained=True, **kwargs)

radio/extra_models.py:102

Method_load_embed

(self, src_embed: torch.Tensor, targ_embed: nn.Parameter)

radio/vit_patch_generator.py:128

Method_load_from_state_dict

(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs)

radio/dinov2_arch.py:297

Method_load_from_state_dict

(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs)

radio/enable_spectral_reparam.py:95

Method_load_from_state_dict

(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs)

radio/radio1d.py:1154

Method_load_projection

(self, src_proj_weight: torch.Tensor, targ_proj_weight: torch.Tensor)

radio/vit_patch_generator.py:139

Method_num_cls_tokens

(self)

radio/radio_model.py:84

Function_replace_ln_with_dyt

(ln: nn.LayerNorm, depth: int)

radio/extra_timm_models.py:225

Method_sample

(self, batch_size: int, max_tokens: int)

radio/radio1d.py:715

Method_sample

(self, batch_size: int, max_tokens: int)

radio/radio1d.py:722

Method_sample

(self, batch_size: int, max_tokens: int)

radio/radio1d.py:748

Method_sample

(self, batch_size: int, max_tokens: int)

radio/radio1d.py:768

Method_sample

(self, batch_size: int, max_tokens: int)

radio/radio1d.py:804

Function_stage

(img, c, r, c_pix, r_pix)

tools/im_join.py:114

Function_take_indices

( num_blocks: int, n: Optional[Union[int, List[int], Tuple[int]]], )

radio/enable_cpe_support.py:51

Method_to_global

(self, _, input: List[torch.Tensor])

radio/vitdet.py:117

Method_to_windows

(self, _, input: List[torch.Tensor])

radio/vitdet.py:103

Method_transform

(self, inpt: Any, params: Dict[str, Any])

examples/position_predictor.py:274

Method_transform

(self, inpt: Any, params: Dict[str, Any])

examples/common/resize_transform.py:18

Function_video_mode

Context manager to temporarily set the model in video mode. This is used to handle models that support both image and video inputs.

radio/enable_cpe_support.py:38

Methodadaptors

(self)

radio/hf_model.py:163

Methodapply_cls_token

(self)

radio/vit_patch_generator.py:104

Methodattn_residual_func

(x: torch.Tensor)

radio/dinov2_arch.py:369

Methodattn_residual_func

(x: torch.Tensor, attn_bias=None)

radio/dinov2_arch.py:405

Functionaugment

(ex: Dict[str, torch.Tensor])

examples/ssl_metrics.py:267

Methodbackward

(self, x: torch.Tensor)

radio/input_conditioner.py:37

Methodblocks

(self)

radio/extra_models.py:34

Functioncalc_lidar_eq_pt1

(all_features: torch.Tensor, all_transforms: torch.Tensor, image_sizes: torch.Tensor, input_size: Tuple[int, i

examples/ssl_metrics.py:409

Functioncalc_lidar_eq_pt2

(cov_b: torch.Tensor, cov_w: torch.Tensor, mu_x: torch.Tensor, gamma: float = 0.75, delta: float = DELTA)

examples/ssl_metrics.py:499

Functioncollate

(samples: List[Dict[str, torch.Tensor]], group: bool = True)

examples/common/utils.py:17

Functioncreate_open_clip_adaptor

(main_config: Namespace, adaptor_config: dict_t, state: state_t)

radio/open_clip_adaptor.py:40

Functioncreate_siglip2_adaptor

(main_config: Namespace, adaptor_config: dict_t, state: state_t)

radio/siglip2_adaptor.py:95

Methoddecorator

(factory_function)

radio/adaptor_registry.py:24

Functiondino_v2_g_student

(**kwargs)

radio/extra_models.py:208

Functiondino_v2_l_student

(**kwargs)

radio/extra_models.py:204

Functiondinov2_vitb14

DINOv2 ViT-B/14 model (optionally) pretrained on the LVD-142M dataset.

radio/dinov2_arch.py:941

Functiondinov2_vitb14_reg

DINOv2 ViT-B/14 model with registers (optionally) pretrained on the LVD-142M dataset.

radio/dinov2_arch.py:979

Functiondinov2_vitg14

DINOv2 ViT-g/14 model (optionally) pretrained on the LVD-142M dataset.

radio/dinov2_arch.py:955

Functiondinov2_vitg14_reg

DINOv2 ViT-g/14 model with registers (optionally) pretrained on the LVD-142M dataset.

radio/dinov2_arch.py:1005

Functiondinov2_vitl14

DINOv2 ViT-L/14 model (optionally) pretrained on the LVD-142M dataset.

radio/dinov2_arch.py:948

Functiondinov2_vitl14_reg

DINOv2 ViT-L/14 model with registers (optionally) pretrained on the LVD-142M dataset.

radio/dinov2_arch.py:992

Functiondinov2_vits14

DINOv2 ViT-S/14 model (optionally) pretrained on the LVD-142M dataset.

radio/dinov2_arch.py:934

Functiondinov2_vits14_reg

DINOv2 ViT-S/14 model with registers (optionally) pretrained on the LVD-142M dataset.

radio/dinov2_arch.py:966

Methoddisable

(self)

radio/cls_token.py:40

MethoddivisorGenerator

(n)

radio/eradio_model.py:1271

Functionduplicate_interleave

(m)

radio/vision_transformer_xpos.py:32

Functiondv2_sdpa

(self, x: torch.Tensor)

radio/extra_models.py:86

Functiondv2_sdpa

(self, x: torch.Tensor)

examples/common/model_loader.py:19

Methodembed_dim

(self)

radio/radio_model.py:137

Methodembed_dim

(self)

radio/extra_models.py:38

Methodembed_dim

(self)

radio/extra_models.py:127

Methodembed_dim

(self)

examples/common/model_loader.py:204

Methodembed_dim

(self)

examples/common/model_loader.py:242

Methodencode_image

(self, image, normalize: bool = False)

examples/common/model_loader.py:267

Methodencode_text

(self, text, normalize: bool = False)

radio/open_clip_adaptor.py:35

Methodencode_text

(self, text, normalize: bool = False)

examples/common/model_loader.py:183

Methodencode_text

(self, text, normalize: bool = False)

examples/common/model_loader.py:276

Functioneradio

(pretrained=False, **kwargs)

radio/eradio_model.py:1397

Functioneradio_xxxtiny

(pretrained=False, **kwargs)

radio/eradio_model.py:1324

Functioneradio_xxxtiny_8x_ws12

(pretrained=False, **kwargs)

radio/eradio_model.py:1348

Functioneradio_xxxtiny_8x_ws16

(pretrained=False, **kwargs)

radio/eradio_model.py:1373

Methodf

(*args, **kwargs)

radio/dinov2_arch.py:642

Methodffn_residual_func

(x: torch.Tensor)

radio/dinov2_arch.py:372

Methodffn_residual_func

(x: torch.Tensor, attn_bias=None)

radio/dinov2_arch.py:408

Methodflops

(self)

radio/dinov2_arch.py:111

Methodforward

Forward function.

mmseg/linear_head.py:114

(self, x: torch.Tensor)

radio/feature_normalizer.py:34

Methodforward

(self, x: torch.Tensor, index: int, rot_index: int = None, skip: Optional[int] = None)

radio/feature_normalizer.py:45

Methodforward

(self, x: torch.Tensor, index: int, rot_index: int = None, skip: Optional[int] = None)

radio/feature_normalizer.py:61

Methodforward

(self, x: torch.Tensor, index: int, rot_index: int = None, skip: Optional[int] = None)

radio/feature_normalizer.py:110

Methodforward

(self, x)

radio/extra_timm_models.py:200

Methodforward

(self, x: torch.Tensor)

radio/dinov2_arch.py:96

← previous next →301–400 of 572, ranked by callers