Functions — github.com/whai362/PVT

hub / github.com/whai362/PVT / functions

Functions229 in github.com/whai362/PVT

⨍ Functions 229◇ Types & classes 58

↓ 36 callersFunctionprint

(*args, **kwargs)

classification/utils.py:179

↓ 12 callersMethod__init__

(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0., linear=False)

detection/pvt_v2.py:16

↓ 12 callersMethodupdate

(self, **kwargs)

classification/utils.py:85

↓ 11 callersFunctionbuild_dataset

(is_train, args)

classification/datasets.py:57

↓ 9 callersMethod__init__

(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)

detection/pvt.py:15

↓ 8 callersMethod__init__

(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)

segmentation/pvt.py:15

↓ 5 callersMethod__init__

(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0., linear=False)

classification/pvt_v2.py:13

↓ 4 callersMethod__init__

(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)

classification/pvt.py:16

↓ 4 callersFunctionli_sra_flops

(h, w, dim)

detection/get_flops.py:44

↓ 4 callersFunctionli_sra_flops

(h, w, dim)

classification/get_flops.py:31

↓ 4 callersFunctionsra_flops

(h, w, r, dim)

detection/get_flops.py:40

↓ 4 callersFunctionsra_flops

(h, w, r, dim)

classification/get_flops.py:27

↓ 3 callersFunctionget_shared_folder

()

classification/run_with_submitit.py:31

↓ 3 callersFunctionis_dist_avail_and_initialized

()

classification/utils.py:187

↓ 2 callersMethod_align

(self, img, size_divisor, interpolation=None)

segmentation/align_resize.py:152

↓ 2 callersMethod_get_pos_embed

(self, pos_embed, patch_embed, H, W)

detection/pvt.py:183

↓ 2 callersMethod_get_pos_embed

(self, pos_embed, patch_embed, H, W)

classification/pvt.py:201

↓ 2 callersMethod_get_pos_embed

(self, pos_embed, patch_embed, H, W)

segmentation/pvt.py:183

↓ 2 callersMethod_save_image_gts_results

(self, dataset, results, mAPs, out_dir=None)

detection/analyze_results.py:63

↓ 2 callersFunctionevaluate

(data_loader, model, device)

classification/engine.py:71

↓ 2 callersFunctionget_init_file

()

classification/run_with_submitit.py:40

↓ 2 callersMethodinit_weights

(self, pretrained=None)

detection/pvt.py:169

↓ 2 callersMethodlog_every

(self, iterable, print_freq, header=None)

classification/utils.py:115

↓ 2 callersMethodpreload

(self)

classification/mcloader/data_prefetcher.py:10

↓ 2 callersMethodrandom_sample

Randomly sample an img_scale when ``multiscale_mode=='range'``. Args: img_scales (list[tuple]): Images scale range for sampling.

segmentation/align_resize.py:61

↓ 2 callersMethodrandom_sample_ratio

Randomly sample an img_scale when ``ratio_range`` is specified. A ratio will be randomly sampled from the range specified by ``ratio_

segmentation/align_resize.py:88

↓ 2 callersFunctionset_random_seed

Set random seed. Args: seed (int): Seed to be used. deterministic (bool): Whether to set the deterministic option for

detection/mmdet_custom/apis/train.py:22

↓ 2 callersMethodsynchronize_between_processes

(self)

classification/utils.py:108

↓ 2 callersMethodupdate

(self, value, n=1)

classification/utils.py:31

↓ 1 callersMethod_init_memcached

(self)

classification/mcloader/image_list.py:23

↓ 1 callersMethod_random_scale

Randomly sample an img_scale according to ``ratio_range`` and ``multiscale_mode``. If ``ratio_range`` is specified, a ratio will be s

segmentation/align_resize.py:114

↓ 1 callersMethod_resize_img

Resize images with ``results['scale']``.

segmentation/align_resize.py:161

↓ 1 callersMethod_resize_seg

Resize semantic segmentation map with ``results['scale']``.

segmentation/align_resize.py:190

↓ 1 callersMethod_setup_gpu_args

(self)

classification/run_with_submitit.py:71

↓ 1 callersMethodadd_meter

(self, name, meter)

classification/utils.py:112

↓ 1 callersFunctionbuild_transform

(is_train, args)

classification/datasets.py:85

↓ 1 callersMethodevaluate_and_show

Evaluate and show results. Args: dataset (Dataset): A PyTorch dataset. results (list): Det results from test results

detection/analyze_results.py:89

↓ 1 callersMethodforward_features

(self, x)

detection/pvt.py:191

↓ 1 callersMethodforward_features

(self, x)

detection/pvt_v2.py:291

↓ 1 callersMethodforward_features

(self, x)

classification/pvt.py:209

↓ 1 callersMethodforward_features

(self, x)

classification/pvt_v2.py:281

↓ 1 callersMethodforward_features

(self, x)

segmentation/pvt.py:191

↓ 1 callersFunctionget_args_parser

()

classification/main.py:31

↓ 1 callersFunctionget_flops

(model, input_shape)

detection/get_flops.py:48

↓ 1 callersFunctionget_flops

(model, input_shape)

classification/get_flops.py:35

↓ 1 callersMethodget_length

(self)

classification/mcloader/image_list.py:29

↓ 1 callersFunctionget_rank

()

classification/utils.py:201

↓ 1 callersMethodget_sample

(self, idx)

classification/mcloader/image_list.py:32

↓ 1 callersMethodinit_weights

(self, pretrained=None)

detection/pvt_v2.py:272

↓ 1 callersFunctionis_main_process

()

classification/utils.py:207

↓ 1 callersFunctionmain

()

detection/train.py:90

↓ 1 callersFunctionmain

()

detection/test.py:107

↓ 1 callersFunctionmain

()

detection/benchmark.py:42

↓ 1 callersFunctionmain

()

detection/analyze_results.py:176

↓ 1 callersFunctionmain

()

detection/get_flops.py:82

↓ 1 callersFunctionmain

(args)

detection/demo.py:27

↓ 1 callersFunctionmain

()

classification/get_flops.py:65

↓ 1 callersFunctionmain

()

classification/run_with_submitit.py:83

↓ 1 callersFunctionmain

(args)

classification/main.py:177

↓ 1 callersFunctionmain

()

segmentation/train.py:65

↓ 1 callersFunctionmain

()

segmentation/test.py:73

↓ 1 callersFunctionparse_args

()

detection/train.py:25

↓ 1 callersFunctionparse_args

()

detection/test.py:22

↓ 1 callersFunctionparse_args

()

detection/benchmark.py:17

↓ 1 callersFunctionparse_args

()

detection/analyze_results.py:137

↓ 1 callersFunctionparse_args

()

detection/get_flops.py:17

↓ 1 callersFunctionparse_args

()

detection/demo.py:10

↓ 1 callersFunctionparse_args

()

classification/get_flops.py:14

↓ 1 callersFunctionparse_args

()

classification/run_with_submitit.py:16

↓ 1 callersFunctionparse_args

()

segmentation/train.py:21

↓ 1 callersFunctionparse_args

()

segmentation/test.py:18

↓ 1 callersFunctionpil_loader

(img_str)

classification/mcloader/mcloader.py:9

↓ 1 callersMethodrandom_select

Randomly select an img_scale from given candidates. Args: img_scales (list[tuple]): Images scales for selection. Returns

segmentation/align_resize.py:43

↓ 1 callersMethodresume

(self, checkpoint, resume_optimizer=True, map_location='default')

detection/mmcv_custom/runner/epoch_based_runner.py:70

↓ 1 callersFunctionsave_checkpoint

Save checkpoint to file. The checkpoint will have 4 fields: ``meta``, ``state_dict`` and ``optimizer``, ``amp``. By default ``meta`` will con

detection/mmcv_custom/runner/checkpoint.py:19

↓ 1 callersMethodset_epoch

(self, epoch)

classification/samplers.py:58

↓ 1 callersFunctionsetup_for_distributed

This function disables printing when not in master process

classification/utils.py:172

↓ 1 callersFunctiontrain_detector

(model, dataset, cfg, distributed=False,

detection/mmdet_custom/apis/train.py:41

↓ 1 callersFunctiontrain_one_epoch

(model: torch.nn.Module, criterion: DistillationLoss, data_loader: Iterable, optimizer: to

classification/engine.py:19

Method__call__

(self)

classification/run_with_submitit.py:53

Method__call__

(self, fn)

classification/mcloader/mcloader.py:26

Method__call__

Call function to resize images, bounding boxes, masks, semantic segmentation map. Args: results (dict): Result dict from

segmentation/align_resize.py:206

Method__getattr__

(self, attr)

classification/utils.py:92

Method__getitem__

(self, idx)

classification/mcloader/classification.py:26

Method__init__

(self, show=False, wait_time=0, score_thr=0)

detection/analyze_results.py:58

Method__init__

(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0., sr_ratio=1)

detection/pvt.py:34

Method__init__

(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., dr

detection/pvt.py:80

Method__init__

(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768)

detection/pvt.py:105

Method__init__

(self, img_size=224, patch_size=16, in_chans=3, num_classes=1000, embed_dims=[64, 128, 256, 512],

(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0., sr_ratio=1, linear=False)

detection/pvt_v2.py:58

Method__init__

(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., dr

detection/pvt_v2.py:134

Method__init__

(self, img_size=224, patch_size=7, stride=4, in_chans=3, embed_dim=768)

detection/pvt_v2.py:176

Method__init__

(self, img_size=224, patch_size=16, in_chans=3, num_classes=1000, embed_dims=[64, 128, 256, 512],

detection/pvt_v2.py:218

Method__init__

(self, dim=768)

detection/pvt_v2.py:317

Method__init__

(self, **kwargs)

detection/pvt_v2.py:343

next →1–100 of 229, ranked by callers