MCPcopy Create free account

hub / github.com/PINTO0309/gazelle-dinov3 / functions

Functions216 in github.com/PINTO0309/gazelle-dinov3

⨍ Functions 216◇ Types & classes 30
↓ 1 callersFunctionis_parsable_to_int
(s)
demo_deimv2_onnx_wholebody34_with_edges.py:986
↓ 1 callersFunctionlist_image_files
(dir_path: str)
demo_deimv2_onnx_wholebody34_with_edges.py:979
↓ 1 callersFunctionload_data_gazefollow
(file)
gazelle/dataloader.py:19
↓ 1 callersFunctionload_data_vat
(file, sample_rate)
gazelle/dataloader.py:10
↓ 1 callersMethodload_partial_state_dict
(model: nn.Module, state_dict: Dict)
gazelle/backbone.py:1236
↓ 1 callersFunctionmain
()
demo_deimv2_onnx_wholebody34_with_edges.py:1143
↓ 1 callersFunctionmain
(PATH)
data_prep/preprocess_vat.py:44
↓ 1 callersFunctionmain
(data_path: str)
data_prep/preprocess_gazefollow.py:14
↓ 1 callersFunctionmain
()
scripts/eval_vat.py:53
↓ 1 callersFunctionmain
()
scripts/eval_gazefollow.py:47
↓ 1 callersFunctionmain
()
scripts/train_gazefollow.py:236
↓ 1 callersFunctionmain
()
scripts/train_vat.py:337
↓ 1 callersFunctionmerge_dfs
(ls)
data_prep/preprocess_vat.py:16
↓ 1 callersFunctionoverlay_gaze_heatmaps
Overlay gaze heatmaps on the provided image.
demo_deimv2_onnx_wholebody34_with_edges.py:954
↓ 1 callersFunctionprint_param_summary
(rows)
scripts/train_gazefollow.py:202
↓ 1 callersFunctionprint_param_summary
(rows)
scripts/train_vat.py:303
↓ 1 callersFunctionprune_best_checkpoints
(exp_dir, keep=10)
scripts/train_gazefollow.py:188
↓ 1 callersFunctionprune_best_checkpoints
(exp_dir, keep=10)
scripts/train_vat.py:289
↓ 1 callersFunctionprune_epoch_checkpoints
(exp_dir, keep=1)
scripts/train_gazefollow.py:174
↓ 1 callersFunctionprune_epoch_checkpoints
(exp_dir, keep=1)
scripts/train_vat.py:275
↓ 1 callersFunctionrestore_arg
(name)
scripts/train_gazefollow.py:247
↓ 1 callersFunctionrestore_arg
(name)
scripts/train_vat.py:346
↓ 1 callersFunctionrotate_half
Rotates half the hidden dims of the input.
gazelle/backbone.py:254
↓ 1 callersMethodrun
Compatibility wrapper.
demo_deimv2_onnx_wholebody34_with_edges.py:922
↓ 1 callersFunctionsmooth_df
(window_size, df)
data_prep/preprocess_vat.py:36
Method__call__
( self, *, input_datas: List[np.ndarray], )
demo_deimv2_onnx_wholebody34_with_edges.py:337
Method__call__
Parameters ---------- image: np.ndarray Entire image disable_generation_identification_mode: bool
demo_deimv2_onnx_wholebody34_with_edges.py:426
Method__call__
Run heatmap inference for the provided heads.
demo_deimv2_onnx_wholebody34_with_edges.py:876
Method__getitem__
(self, idx)
gazelle/dataloader.py:51
Method__getitem__
(self, idx)
scripts/eval_vat.py:32
Method__getitem__
(self, idx)
scripts/eval_gazefollow.py:27
Method__init__
(self, iou_threshold: float = 0.3, max_age: int = 30)
demo_deimv2_onnx_wholebody34_with_edges.py:102
Method__init__
( self, *, runtime: Optional[str] = 'onnx', model_path: Optional[str] = '',
demo_deimv2_onnx_wholebody34_with_edges.py:245
Method__init__
( self, runtime: Optional[str] = 'onnx', model_path: Optional[str] = '', provi
demo_deimv2_onnx_wholebody34_with_edges.py:816
Method__init__
(self, dataset_name, path, split, transform, heatmap_size=(64, 64), in_frame_only=True, sample_rate=1)
gazelle/dataloader.py:25
Method__init__
(self)
gazelle/backbone.py:16
Method__init__
(self, model_name)
gazelle/backbone.py:129
Method__init__
( self, embed_dim: int, *, num_heads: int, base: float | None = 100.0,
gazelle/backbone.py:167
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768)
gazelle/backbone.py:281
Method__init__
(self, drop_prob=None)
gazelle/backbone.py:302
Method__init__
(self, dim, num_heads=8, qkv_bias=False, attn_drop=0., proj_drop=0.)
gazelle/backbone.py:324
Method__init__
(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, drop=0., attn_drop=0., drop_path=0., act_layer=nn.GELU, n
gazelle/backbone.py:364
Method__init__
( self, img_size=224, patch_size=16, in_chans=3, embed_dim=192, depth=12, num_heads=3, mlp_rat
gazelle/backbone.py:379
Method__init__
( self, model_name=None, weights_path=None, interaction_indexes=[], fi
gazelle/backbone.py:489
Method__init__
( self, scale_value=1.0, bias_value=0.0 )
gazelle/backbone.py:612
Method__init__
( self, in_chs, out_chs, kernel_size, stride=1,
gazelle/backbone.py:625
Method__init__
( self, in_chs, out_chs, kernel_size, groups=1,
gazelle/backbone.py:720
Method__init__
(self, in_chs, mid_chs, out_chs, use_lab=False, act='relu')
gazelle/backbone.py:755
Method__init__
(self, chs)
gazelle/backbone.py:812
Method__init__
( self, in_chs, mid_chs, out_chs, layer_num,
gazelle/backbone.py:831
Method__init__
( self, in_chs, mid_chs, out_chs, block_num,
gazelle/backbone.py:925
Method__init__
(self, num_features, eps=1e-5)
gazelle/backbone.py:988
Method__init__
( self, model_name=None, weights_path=None, use_lab=True, return_idx=[
gazelle/backbone.py:1042
Method__init__
( self, backbone: DinoV2Backbone | DinoV3Backbone | HGNetv2Backbone, inout=False,
gazelle/model.py:157
Method__init__
(self, path, img_transform)
scripts/eval_vat.py:23
Method__init__
(self, path, img_transform)
scripts/eval_gazefollow.py:22
Method__len__
(self)
gazelle/dataloader.py:94
Method__len__
(self)
scripts/eval_vat.py:44
Method__len__
(self)
scripts/eval_gazefollow.py:38
Method__str__
(self)
demo_deimv2_onnx_wholebody34_with_edges.py:75
Method_format_repo
(r, ref)
gazelle/backbone.py:510
Method_init_vit_weights
(self, m)
gazelle/backbone.py:418
Method_load
(repo_or_dir)
gazelle/backbone.py:516
Method_load_from_state_dict
(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys
gazelle/backbone.py:1002
Method_postprocess
( self, *, image: np.ndarray, boxes: np.ndarray, )
demo_deimv2_onnx_wholebody34_with_edges.py:374
Method_postprocess
( self, *, image: np.ndarray, boxes: np.ndarray, )
demo_deimv2_onnx_wholebody34_with_edges.py:945
Method_preprocess
( self, *, image: np.ndarray, swap: Optional[Tuple[int,int,int]] = (2, 0, 1),
demo_deimv2_onnx_wholebody34_with_edges.py:365
Functioncheck_alpha
(value)
demo_deimv2_onnx_wholebody34_with_edges.py:1152
Functioncheck_positive
(value)
demo_deimv2_onnx_wholebody34_with_edges.py:1146
Functioncollate
(batch)
scripts/eval_vat.py:47
Functioncollate
(batch)
scripts/eval_gazefollow.py:41
Functioncollate_fn
(batch)
gazelle/dataloader.py:98
Methodextra_repr
(self)
gazelle/backbone.py:1023
Methodfeature_dim
(self)
gazelle/backbone.py:433
Methodforward
(self, x: torch.Tensor)
gazelle/backbone.py:20
Methodforward
(self, x: torch.Tensor)
gazelle/backbone.py:133
Methodforward
(self, *, H: int, W: int)
gazelle/backbone.py:205
Methodforward
(self, x: torch.Tensor)
gazelle/backbone.py:288
Methodforward
(self, x)
gazelle/backbone.py:305
Methodforward
(self, x: torch.Tensor, rope_sincos=None, register_hook=False)
gazelle/backbone.py:334
Methodforward
(self, x, rope_sincos=None, register_hook=False)
gazelle/backbone.py:372
Methodforward
(self, x: torch.Tensor, register_hook=False)
gazelle/backbone.py:457
Methodforward
(self, x: torch.Tensor)
gazelle/backbone.py:570
Methodforward
(self, x)
gazelle/backbone.py:621
Methodforward
(self, x)
gazelle/backbone.py:673
Methodforward
(self, x)
gazelle/backbone.py:748
Methodforward
(self, x)
gazelle/backbone.py:799
Methodforward
(self, x)
gazelle/backbone.py:823
Methodforward
(self, x)
gazelle/backbone.py:912
Methodforward
(self, x)
gazelle/backbone.py:975
Methodforward
(self, x)
gazelle/backbone.py:1012
Methodforward
(self, x: torch.Tensor)
gazelle/backbone.py:1274
Methodforward
(self, image_bgr: torch.Tensor, bboxes: torch.Tensor)
gazelle/model.py:200
Methodforward_pool
(self, x)
gazelle/backbone.py:452
Functiongazelle_dinov2_vitb14
(pretrained=True)
hubconf.py:6
Functiongazelle_dinov2_vitb14_inout
(pretrained=True)
hubconf.py:20
Functiongazelle_dinov2_vitl14
(pretrained=True)
hubconf.py:13
Functiongazelle_dinov2_vitl14_inout
(pretrained=True)
hubconf.py:27
Methodget_dimension
(self)
gazelle/backbone.py:140
Methodget_dimension
(self)
gazelle/backbone.py:593
← previousnext →101–200 of 216, ranked by callers