Functions — github.com/amazon-science/long-short-term-transformer

hub / github.com/amazon-science/long-short-term-transformer / functions

Functions108 in github.com/amazon-science/long-short-term-transformer

⨍ Functions 108◇ Types & classes 24 ↳ Endpoints 2

↓ 15 callersMethodload

(self, model, optimizer=None)

src/rekognition_online_action_detection/utils/checkpointer.py:24

↓ 6 callersFunctionto_device

(x, dtype=np.float32)

src/rekognition_online_action_detection/engines/lstr/lstr_inference.py:42

↓ 4 callersMethod__init__

(self, d_model=512, nhead=8, num_encoder_layers=6, num_decoder_layers=6, dim_feedforward=2048

src/rekognition_online_action_detection/models/transformer/transformer.py:15

↓ 3 callersMethod__init__

(self, optimizer, max_iters, last_epoch=-1)

src/rekognition_online_action_detection/optimizers/lr_scheduler.py:97

↓ 3 callersMethod__init__

(self, embed_dim, num_heads, dropout=0.0, bias=True, kdim=None, vdim=None)

src/rekognition_online_action_detection/models/transformer/multihead_attention.py:67

↓ 2 callersMethod__init__

(self, reduction='mean', ignore_index=-100)

src/rekognition_online_action_detection/criterions/criterions.py:18

↓ 2 callersFunction_get_activation_fn

(activation)

src/rekognition_online_action_detection/models/transformer/transformer.py:226

↓ 2 callersFunction_get_clones

(module, N)

src/rekognition_online_action_detection/models/transformer/transformer.py:222

↓ 2 callersFunction_get_warmup_factor_at_iter

(warmup_method, this_iter, warmup_iters,

src/rekognition_online_action_detection/optimizers/lr_scheduler.py:12

↓ 2 callersMethod_init_dataset

(self)

src/rekognition_online_action_detection/datasets/perframe_data_layers.py:37

↓ 2 callersFunction_register_generic

(module_dict, module_name, module)

src/rekognition_online_action_detection/utils/registry.py:4

↓ 2 callersFunctionbuild_dataset

(cfg, phase, tag='')

src/rekognition_online_action_detection/datasets/datasets.py:16

↓ 2 callersFunctionbuild_feature_head

(cfg)

src/rekognition_online_action_detection/models/feature_head.py:67

↓ 2 callersFunctionbuild_model

(cfg, device=None)

src/rekognition_online_action_detection/models/models.py:11

↓ 2 callersFunctionload_cfg

()

src/rekognition_online_action_detection/utils/parser.py:88

↓ 2 callersFunctionpostprocessing

(data_name)

src/rekognition_online_action_detection/evaluation/postprocessing.py:7

↓ 2 callersFunctionsetup_checkpointer

(cfg, phase)

src/rekognition_online_action_detection/utils/checkpointer.py:43

↓ 2 callersFunctionsetup_environment

(cfg)

src/rekognition_online_action_detection/utils/env.py:27

↓ 2 callersFunctionsetup_logger

(cfg, phase, quiet=False)

src/rekognition_online_action_detection/utils/logger.py:12

↓ 1 callersMethod__init__

(self, cfg)

src/rekognition_online_action_detection/models/lstr.py:14

↓ 1 callersMethod__setstate__

(self, state)

src/rekognition_online_action_detection/models/transformer/transformer.py:139

↓ 1 callersMethod_load_checkpoint

(self, checkpoint)

src/rekognition_online_action_detection/utils/checkpointer.py:37

↓ 1 callersFunctionassert_and_infer_cfg

(cfg, args)

src/rekognition_online_action_detection/utils/parser.py:34

↓ 1 callersFunctionbuild_criterion

(cfg, device=None)

src/rekognition_online_action_detection/criterions/criterions.py:73

↓ 1 callersFunctionbuild_data_loader

(cfg, phase)

src/rekognition_online_action_detection/datasets/datasets.py:21

↓ 1 callersFunctionbuild_optimizer

(cfg, model)

src/rekognition_online_action_detection/optimizers/optimizers.py:9

↓ 1 callersFunctionbuild_scheduler

Unlike the PyTorch version, our schedulers adjust the learning rate according to iteration rather than epoch.

src/rekognition_online_action_detection/optimizers/lr_scheduler.py:153

↓ 1 callersFunctiondo_inference

(cfg, model, device, logger)

src/rekognition_online_action_detection/engines/engines.py:31

↓ 1 callersFunctiondo_lstr_stream_inference

(cfg, model, device, logger)

src/rekognition_online_action_detection/engines/lstr/lstr_inference.py:34

↓ 1 callersFunctiondo_perframe_det_batch_inference

(cfg, model, device, logger)

src/rekognition_online_action_detection/engines/base_inferences/perframe_det_batch_inference.py:16

↓ 1 callersFunctiondo_perframe_det_train

(cfg, data_loaders, model, crite

src/rekognition_online_action_detection/engines/base_trainers/perframe_det_trainer.py:13

↓ 1 callersFunctiondo_train

(cfg, data_loaders, model, criterion, optimizer,

src/rekognition_online_action_detection/engines/engines.py:10

↓ 1 callersFunctioneval_perframe

(pred_scores_file)

tools/eval/eval_perframe.py:14

↓ 1 callersFunctioneval_perstage

(pred_scores_file)

tools/eval/eval_perstage.py:14

↓ 1 callersFunctionget_cfg

()

src/rekognition_online_action_detection/config/defaults.py:134

↓ 1 callersMethodget_lr

(self)

src/rekognition_online_action_detection/optimizers/lr_scheduler.py:43

↓ 1 callersMethodget_lr

(self)

src/rekognition_online_action_detection/optimizers/lr_scheduler.py:76

↓ 1 callersMethodget_lr

(self)

src/rekognition_online_action_detection/optimizers/lr_scheduler.py:104

↓ 1 callersMethodget_lr

(self)

src/rekognition_online_action_detection/optimizers/lr_scheduler.py:133

↓ 1 callersFunctionget_stage_pred_scores

(gt_targets, pred_scores, perc_s, perc_e)

src/rekognition_online_action_detection/evaluation/metrics.py:66

↓ 1 callersFunctionmain

(cfg)

tools/test_net.py:12

↓ 1 callersFunctionmain

(cfg)

tools/train_net.py:16

↓ 1 callersFunctionparse_args

()

src/rekognition_online_action_detection/utils/parser.py:11

↓ 1 callersFunctionperframe_average_precision

Compute (frame-level) average precision between ground truth and predictions data frames.

src/rekognition_online_action_detection/evaluation/metrics.py:26

↓ 1 callersFunctionperstage_average_precision

(ground_truth, prediction, class_names,

src/rekognition_online_action_detection/evaluation/metrics.py:92

↓ 1 callersMethodsave

(self, epoch, model, optimizer)

src/rekognition_online_action_detection/utils/checkpointer.py:30

↓ 1 callersMethodsegment_sampler

(self, start, end, num_samples)

src/rekognition_online_action_detection/datasets/perframe_data_layers.py:49

↓ 1 callersFunctionsetup_random_seed

(seed)

src/rekognition_online_action_detection/utils/env.py:16

↓ 1 callersMethodshuffle

(self)

src/rekognition_online_action_detection/datasets/perframe_data_layers.py:34

↓ 1 callersMethodstream_inference

(self, long_visual_inputs, long_motion_inputs,

src/rekognition_online_action_detection/models/lstr.py:155

↓ 1 callersMethodstream_inference

(self, tgt, memory, pos, tgt_mask=None, memory_mask=None, tgt_key_padding_mask=None,

src/rekognition_online_action_detection/models/transformer/transformer.py:84

↓ 1 callersMethoduniform_sampler

(self, start, end, num_samples, sample_rate)

src/rekognition_online_action_detection/datasets/perframe_data_layers.py:53

↓ 1 callersMethoduniform_sampler

(self, start, end, num_samples, sample_rate)

src/rekognition_online_action_detection/datasets/perframe_data_layers.py:156

Method__getitem__

(self, index)

src/rekognition_online_action_detection/datasets/perframe_data_layers.py:60

Method__getitem__

(self, index)

src/rekognition_online_action_detection/datasets/perframe_data_layers.py:163

Method__init__

(self, reduction='mean', ignore_index=-100)

src/rekognition_online_action_detection/criterions/criterions.py:30

Method__init__

(self, reduction='mean', ignore_index=-100)

src/rekognition_online_action_detection/criterions/criterions.py:43

Method__init__

(self, cfg, phase)

src/rekognition_online_action_detection/utils/checkpointer.py:13

Method__init__

(self, *args, **kwargs)

src/rekognition_online_action_detection/utils/registry.py:11

Method__init__

(self, optimizer, milestones, gamma=0.1, l

src/rekognition_online_action_detection/optimizers/lr_scheduler.py:30

Method__init__

(self, optimizer, milestones, gamma=0.1, w

src/rekognition_online_action_detection/optimizers/lr_scheduler.py:57

Method__init__

(self, optimizer, max_iters, gamma=0.1, wa

src/rekognition_online_action_detection/optimizers/lr_scheduler.py:119

Method__init__

(self, cfg, phase='train')

src/rekognition_online_action_detection/datasets/perframe_data_layers.py:18

Method__init__

(self, cfg, phase='test')

src/rekognition_online_action_detection/datasets/perframe_data_layers.py:131

Method__init__

(self, cfg)

src/rekognition_online_action_detection/models/lstr.py:146

Method__init__

(self, cfg)

src/rekognition_online_action_detection/models/feature_head.py:26

Method__init__

(self, encoder_layer, num_layers, norm=None)

src/rekognition_online_action_detection/models/transformer/transformer.py:55

Method__init__

(self, decoder_layer, num_layers, norm=None)

src/rekognition_online_action_detection/models/transformer/transformer.py:77

Method__init__

(self, d_model, nhead, dim_feedforward=2048, dropout=0.1, activation='relu')

src/rekognition_online_action_detection/models/transformer/transformer.py:122

Method__init__

(self, d_model, nhead, dim_feedforward=2048, dropout=0.1, activation='relu')

src/rekognition_online_action_detection/models/transformer/transformer.py:157

Method__init__

(self, d_model, dropout=0.1, max_len=5000)

src/rekognition_online_action_detection/models/transformer/position_encoding.py:12

Method__init__

(self, dropout=0.0)

src/rekognition_online_action_detection/models/transformer/multihead_attention.py:11

Method__init__

(self, dropout=0.0)

src/rekognition_online_action_detection/models/transformer/multihead_attention.py:32

Method__init__

(self, embed_dim, num_heads, dropout=0.0, bias=True, kdim=None, vdim=None)

src/rekognition_online_action_detection/models/transformer/multihead_attention.py:162

Method__len__

(self)

src/rekognition_online_action_detection/datasets/perframe_data_layers.py:123

Method__len__

(self)

src/rekognition_online_action_detection/datasets/perframe_data_layers.py:222

Method__setstate__

(self, state)

src/rekognition_online_action_detection/models/transformer/transformer.py:182

Method_compute_values

(self)

src/rekognition_online_action_detection/optimizers/lr_scheduler.py:50

Method_compute_values

(self)

src/rekognition_online_action_detection/optimizers/lr_scheduler.py:90

Method_compute_values

(self)

src/rekognition_online_action_detection/optimizers/lr_scheduler.py:112

Method_compute_values

(self)

src/rekognition_online_action_detection/optimizers/lr_scheduler.py:148

Functioncalibrated_average_precision_score

Compute calibrated average precision (cAP), which is particularly proposed for the TVSeries dataset.

src/rekognition_online_action_detection/evaluation/metrics.py:11

Functiondo_lstr_batch_inference

(cfg, model, device, logge

src/rekognition_online_action_detection/engines/lstr/lstr_inference.py:18

Functiondo_lstr_train

(cfg, data_loaders, model, criterion,

src/rekognition_online_action_detection/engines/lstr/lstr_trainer.py:9

Functioneval_perframe

(cfg, ground_truth, prediction, **kwargs)

src/rekognition_online_action_detection/evaluation/evalution.py:14

Functioneval_perstage

(cfg, ground_truth, prediction, **kwargs)

src/rekognition_online_action_detection/evaluation/evalution.py:30

Methodforward

(self, input, target)

src/rekognition_online_action_detection/criterions/criterions.py:23

Methodforward

(self, input, target)

src/rekognition_online_action_detection/criterions/criterions.py:36

Methodforward

(self, input, target)

src/rekognition_online_action_detection/criterions/criterions.py:49

Methodforward

(self, visual_inputs, motion_inputs, memory_key_padding_mask=None)

src/rekognition_online_action_detection/models/lstr.py:81

Methodforward

(self, visual_input, motion_input)

src/rekognition_online_action_detection/models/feature_head.py:56

Methodforward

(self, src, tgt, src_mask=None, tgt_mask=None, memory_mask=None, src_key_padding_mask=None,

src/rekognition_online_action_detection/models/transformer/transformer.py:37

Methodforward

(self, src, src_mask=None, src_key_padding_mask=None)

src/rekognition_online_action_detection/models/transformer/transformer.py:62

Methodforward

(self, tgt, memory, tgt_mask=None, memory_mask=None, tgt_key_padding_mask=None,

src/rekognition_online_action_detection/models/transformer/transformer.py:103

Methodforward

(self, src, src_mask=None, src_key_padding_mask=None)

src/rekognition_online_action_detection/models/transformer/transformer.py:144

Methodforward

(self, tgt, memory, tgt_mask=None, memory_mask=None, tgt_key_padding_mask=None, memory_key_pad

src/rekognition_online_action_detection/models/transformer/transformer.py:206

Methodforward

(self, x, padding=0)

src/rekognition_online_action_detection/models/transformer/position_encoding.py:25

Methodforward

(self, q, k, v, attn_mask=None)

src/rekognition_online_action_detection/models/transformer/multihead_attention.py:16

Methodforward

(self, q, k, v, attn_mask=None, key_padding_mask=None)

src/rekognition_online_action_detection/models/transformer/multihead_attention.py:97

Functiongenerate_square_subsequent_mask

(sz)

src/rekognition_online_action_detection/models/transformer/utils.py:13

next →1–100 of 108, ranked by callers