MCPcopy Index your code

hub / github.com/whai362/PVT / functions

Functions229 in github.com/whai362/PVT

Method__init__
(self, **kwargs)
detection/pvt_v2.py:352
Method__init__
(self, **kwargs)
detection/pvt_v2.py:361
Method__init__
(self, **kwargs)
detection/pvt_v2.py:370
Method__init__
(self, **kwargs)
detection/pvt_v2.py:379
Method__init__
(self, **kwargs)
detection/pvt_v2.py:388
Method__init__
(self, **kwargs)
detection/pvt_v2.py:397
Method__init__
(self, update_interval=1, grad_clip=None, coalesce=True, bucket_size_mb=-1, use_fp16=False)
detection/mmcv_custom/runner/optimizer.py:12
Method__init__
(self, dataset, num_replicas=None, rank=None, shuffle=True)
classification/samplers.py:16
Method__init__
(self, window_size=20, fmt=None)
classification/utils.py:23
Method__init__
(self, delimiter="\t")
classification/utils.py:81
Method__init__
(self, base_criterion: torch.nn.Module, teacher_model: torch.nn.Module, distillation_type: st
classification/losses.py:15
Method__init__
(self, root, train=True, year=2018, transform=None, target_transform=None, category='name', l
classification/datasets.py:15
Method__init__
(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0., sr_ratio=1)
classification/pvt.py:35
Method__init__
(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., dr
classification/pvt.py:81
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768)
classification/pvt.py:106
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, num_classes=1000, embed_dims=[64, 128, 256, 512],
classification/pvt.py:131
Method__init__
(self, args)
classification/run_with_submitit.py:50
Method__init__
(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0., sr_ratio=1, linear=False)
classification/pvt_v2.py:55
Method__init__
(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., dr
classification/pvt_v2.py:131
Method__init__
(self, img_size=224, patch_size=7, stride=4, in_chans=3, embed_dim=768)
classification/pvt_v2.py:173
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, num_classes=1000, embed_dims=[64, 128, 256, 512],
classification/pvt_v2.py:216
Method__init__
(self, dim=768)
classification/pvt_v2.py:305
Method__init__
(self, split='train', pipeline=None)
classification/mcloader/classification.py:10
Method__init__
(self, root, list_file, memcached=False, mclient_path=None)
classification/mcloader/image_list.py:9
Method__init__
(self, root, list_file, memcached, mclient_path)
classification/mcloader/imagenet.py:6
Method__init__
(self, loader)
classification/mcloader/data_prefetcher.py:5
Method__init__
(self, mclient_path)
classification/mcloader/mcloader.py:16
Method__init__
(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0., sr_ratio=1)
segmentation/pvt.py:34
Method__init__
(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., dr
segmentation/pvt.py:80
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768)
segmentation/pvt.py:105
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, num_classes=1000, embed_dims=[64, 128, 256, 512],
segmentation/pvt.py:130
Method__init__
(self, **kwargs)
segmentation/pvt.py:237
Method__init__
(self, **kwargs)
segmentation/pvt.py:246
Method__init__
(self, **kwargs)
segmentation/pvt.py:255
Method__init__
(self, **kwargs)
segmentation/pvt.py:264
Method__init__
(self, img_scale=None, multiscale_mode='range', ratio_range
segmentation/align_resize.py:14
Method__iter__
(self)
classification/samplers.py:35
Method__len__
(self)
classification/samplers.py:55
Method__len__
(self)
classification/mcloader/classification.py:23
Method__repr__
(self)
segmentation/align_resize.py:224
Method__str__
(self)
classification/utils.py:71
Method__str__
(self)
classification/utils.py:100
Function_conv_filter
convert patch embedding weight from manual patchify + linear proj to conv
detection/pvt.py:224
Function_conv_filter
convert patch embedding weight from manual patchify + linear proj to conv
detection/pvt_v2.py:330
Function_conv_filter
convert patch embedding weight from manual patchify + linear proj to conv
classification/pvt.py:244
Function_conv_filter
convert patch embedding weight from manual patchify + linear proj to conv
classification/pvt_v2.py:318
Function_conv_filter
convert patch embedding weight from manual patchify + linear proj to conv
segmentation/pvt.py:224
Method_init_weights
(self, m)
detection/pvt.py:174
Method_init_weights
(self, m)
detection/pvt_v2.py:30
Method_init_weights
(self, m)
detection/pvt_v2.py:86
Method_init_weights
(self, m)
detection/pvt_v2.py:150
Method_init_weights
(self, m)
detection/pvt_v2.py:193
Method_init_weights
(self, m)
detection/pvt_v2.py:257
Method_init_weights
(self, m)
classification/pvt.py:180
Method_init_weights
(self, m)
classification/pvt_v2.py:27
Method_init_weights
(self, m)
classification/pvt_v2.py:83
Method_init_weights
(self, m)
classification/pvt_v2.py:147
Method_init_weights
(self, m)
classification/pvt_v2.py:191
Method_init_weights
(self, m)
classification/pvt_v2.py:252
Method_init_weights
(self, m)
segmentation/pvt.py:174
Function_load_checkpoint_for_ema
Workaround for ModelEma._load_checkpoint to accept an already-loaded object
classification/utils.py:162
Methodafter_train_iter
(self, runner)
detection/mmcv_custom/runner/optimizer.py:22
Methodavg
(self)
classification/utils.py:55
Functionbbox_map_eval
Evaluate mAP of single image det result. Args: det_result (list[list]): [[cls1_det, cls2_det, ...], ...]. The outer list indi
detection/analyze_results.py:13
Methodbefore_run
(self, runner)
detection/mmcv_custom/runner/optimizer.py:19
Methodcheckpoint
(self)
classification/run_with_submitit.py:59
Methodforward
(self, x)
detection/pvt.py:24
Methodforward
(self, x, H, W)
detection/pvt.py:54
Methodforward
(self, x, H, W)
detection/pvt.py:94
Methodforward
(self, x)
detection/pvt.py:119
Methodforward
(self, x)
detection/pvt.py:215
Methodforward
(self, x, H, W)
detection/pvt_v2.py:45
Methodforward
(self, x, H, W)
detection/pvt_v2.py:101
Methodforward
(self, x, H, W)
detection/pvt_v2.py:165
Methodforward
(self, x)
detection/pvt_v2.py:208
Methodforward
(self, x)
detection/pvt_v2.py:309
Methodforward
(self, x, H, W)
detection/pvt_v2.py:321
Methodforward
Args: inputs: The original inputs that are feed to the teacher model outputs: the outputs of the model to be trained.
classification/losses.py:25
Methodforward
(self, x)
classification/pvt.py:25
Methodforward
(self, x, H, W)
classification/pvt.py:55
Methodforward
(self, x, H, W)
classification/pvt.py:95
Methodforward
(self, x)
classification/pvt.py:120
Methodforward
(self, x)
classification/pvt.py:237
Methodforward
(self, x, H, W)
classification/pvt_v2.py:42
Methodforward
(self, x, H, W)
classification/pvt_v2.py:98
Methodforward
(self, x, H, W)
classification/pvt_v2.py:162
Methodforward
(self, x)
classification/pvt_v2.py:206
Methodforward
(self, x)
classification/pvt_v2.py:297
Methodforward
(self, x, H, W)
classification/pvt_v2.py:309
Methodforward
(self, x)
segmentation/pvt.py:24
Methodforward
(self, x, H, W)
segmentation/pvt.py:54
Methodforward
(self, x, H, W)
segmentation/pvt.py:94
Methodforward
(self, x)
segmentation/pvt.py:119
Methodforward
(self, x)
segmentation/pvt.py:215
Methodfreeze_patch_emb
(self)
detection/pvt_v2.py:277
Methodfreeze_patch_emb
(self)
classification/pvt_v2.py:267
Methodget_classifier
(self)
detection/pvt_v2.py:284
Methodget_classifier
(self)
classification/pvt.py:194
Methodget_classifier
(self)
classification/pvt_v2.py:274
Functionget_world_size
()
classification/utils.py:195
← previousnext →101–200 of 229, ranked by callers