MCPcopy Create free account

hub / github.com/crodriguezo/TMLGA / functions

Functions78 in github.com/crodriguezo/TMLGA

↓ 7 callersMethodplot
(self, epoch)
utils/visualization.py:120
↓ 4 callersFunctionmkdir
(path)
utils/miscellaneous.py:5
↓ 3 callersMethod__init__
(self)
utils/pooling.py:19
↓ 3 callersMethodmask_softmax
(self, feat, mask)
modeling/localization.py:61
↓ 3 callersMethodrun
(self, indexes, pred_start, pred_end, start, end, lengths, epoch, loss, individual_loss, atten
utils/visualization.py:44
↓ 2 callersMethodadd_tokenized_sentence
(self, tokens)
utils/vocab.py:71
↓ 2 callersFunctionbuild_dataset
(dataset_name, dataset_catalog, cfg)
data/build.py:8
↓ 2 callersFunctionfeed_forward_rnn
Recursive function to encapsulate RNN calls. :param rnn: :param embedded_sequence_batch: :param lengths: :param hidden_tuple:
utils/rnns.py:7
↓ 2 callersMethodfinish
(self)
utils/vocab.py:84
↓ 2 callersMethodget
(name)
config/paths_catalog.py:54
↓ 2 callersFunctionget_embedding_matrix
(embeddings, vocab)
utils/sentence.py:62
↓ 2 callersMethodkl_div
(self, p, gt, length)
modeling/localization.py:64
↓ 2 callersMethodtokens2indices
(self, tokens, add_bos=False, add_eos=False, oovs=None)
utils/vocab.py:183
↓ 1 callersMethod__str__
(self)
utils/vocab.py:22
↓ 1 callersMethodattention
(self, videoFeat, filter, lengths)
modeling/localization.py:34
↓ 1 callersFunctionclean_parsed_sentence
(parsed_sentence)
utils/sentence.py:15
↓ 1 callersFunctionclean_sentence
(sentence)
utils/sentence.py:7
↓ 1 callersMethodcreateIndex
(self)
data/datasets/charades_sta.py:124
↓ 1 callersMethodcreateIndex
(self)
data/datasets/anet_cap.py:138
↓ 1 callersMethodcreate_vocab
(self)
data/datasets/charades_sta.py:58
↓ 1 callersMethodcreate_vocab
(self)
data/datasets/anet_cap.py:72
↓ 1 callersFunctiongather_last
(batch_hidden_states, batch_lengths, bidirectional=True)
utils/rnns.py:100
↓ 1 callersFunctionget_cfg_defaults
Get a yacs CfgNode object with default values for my_project.
config/__init__.py:3
↓ 1 callersMethodget_embedding_matrix
Gets you a torch tensor with the embeddings in the indices given by self.vocab. Unknown (unseen) words are each mapped to a
data/datasets/charades_sta.py:86
↓ 1 callersMethodget_embedding_matrix
Gets you a torch tensor with the embeddings in the indices given by self.vocab. Unknown (unseen) words are each mapped to a
data/datasets/anet_cap.py:100
↓ 1 callersMethodget_mask_from_sequence_lengths
(self, sequence_lengths: torch.Tensor, max_length: int)
modeling/localization.py:38
↓ 1 callersFunctionimport_file
(module_name, file_path, make_importable=False)
utils/imports.py:11
↓ 1 callersFunctionmain
()
main.py:15
↓ 1 callersMethodmasked_softmax
(self, vector: torch.Tensor, mask: torch.Tensor, dim: int = -1, memory_efficient: bool = False, mask_fill_valu
modeling/localization.py:43
↓ 1 callersFunctionmax_pooling
:param batch_hidden_states: torch.Tensor(batch_size, seq_len, hidden_size) :return:
utils/rnns.py:91
↓ 1 callersFunctionmean_pooling
:param batch_hidden_states: torch.Tensor(batch_size, seq_len, hidden_size) :param batch_lengths: list(batch_size) :return:
utils/rnns.py:76
↓ 1 callersFunctionpack_rnn_input
:param embedded_sequence_batch: torch.Tensor(batch_size, seq_len) :param sequence_lengths: list(batch_size) :return:
utils/rnns.py:36
↓ 1 callersFunctionsetup_logger
(name, save_dir, save_file, distributed_rank)
utils/logger.py:6
↓ 1 callersMethodtIoU
(self, start, end, pred_start, pred_end)
utils/visualization.py:31
↓ 1 callersMethodtIoU
(self, start, end, pred_start, pred_end)
data/datasets/anet_cap.py:59
↓ 1 callersFunctiontester
(cfg)
engine/build.py:150
↓ 1 callersFunctiontrainer
(cfg)
engine/build.py:15
↓ 1 callersFunctionunpack_rnn_output
:param packed_rnn_output: torch object :param indices: Variable(LongTensor) of indices to sort output :return:
utils/rnns.py:65
Method__call__
(self, batch)
data/collate_batch.py:7
Method__contains__
(self, key)
utils/vocab.py:179
Method__getitem__
(self, i)
utils/vocab.py:170
Method__getitem__
(self, index)
data/datasets/charades_sta.py:144
Method__getitem__
(self, index)
data/datasets/anet_cap.py:176
Method__init__
(self, cfg)
modeling/localization.py:11
Method__init__
(self, cfg)
modeling/dynamic_filters/build.py:9
Method__init__
(self, cfg)
modeling/dynamic_filters/mlp.py:5
Method__init__
(self, cfg)
modeling/dynamic_filters/lstm.py:7
Method__init__
(self, cfg, dataset_size, is_train=True)
utils/visualization.py:9
Method__init__
(self)
utils/pooling.py:9
Method__init__
(self, bidirectional=False)
utils/pooling.py:28
Method__init__
(self)
utils/pooling.py:39
Method__init__
(self, string, hash=None)
utils/vocab.py:15
Method__init__
:param sentences: :param token_function: :param min_count: :param add_padding: :param add_bos: :param
utils/vocab.py:32
Method__init__
(self, features_path, ann_file_path, embeddings_path,
data/datasets/charades_sta.py:22
Method__init__
(self, features_path, ann_file_path, embeddings_path,
data/datasets/anet_cap.py:22
Method__iter__
(self)
utils/vocab.py:176
Method__len__
(self)
utils/vocab.py:173
Method__len__
(self)
data/datasets/charades_sta.py:190
Method__len__
(self)
data/datasets/anet_cap.py:258
Method__repr__
(self)
utils/vocab.py:25
Functionbuild
(cfg)
modeling/__init__.py:3
Methodforward
(self, videoFeat, videoFeat_lengths, tokens, tokens_lengths, start, end, localiz)
modeling/localization.py:73
Methodforward
(self, sequences, lengths=None)
modeling/dynamic_filters/build.py:22
Methodforward
(self, sentence_embed)
modeling/dynamic_filters/mlp.py:16
Methodforward
(self, sequences, lengths)
modeling/dynamic_filters/lstm.py:25
Methodforward
(self, batch_hidden_states, lengths, **kwargs)
utils/pooling.py:13
Methodforward
(self, batch_hidden_states, lengths, **kwargs)
utils/pooling.py:22
Methodforward
(self, batch_hidden_states, lengths, **kwargs)
utils/pooling.py:32
Methodforward
(self, batch_hidden_states, **kwargs)
utils/pooling.py:42
Methodget
(name)
modeling/dynamic_filters/catalog.py:6
Methodindices2tokens
(self, indices, ignore_ids=(), oovs=(), highlight_oovs=False)
utils/vocab.py:211
Functioniou
(gt, pred, lengths)
utils/loss.py:30
Functionkl_div
This function compute the KL divergency between gt localization and prediction. Parameters ---------- gt: pred_local: le
utils/loss.py:5
Functionmake_dataloader
(cfg, is_train)
data/build.py:22
Functionmake_optimizer
(cfg, model)
solver/build.py:3
Functionpad_sequence
(sequence, batch_first=True)
utils/rnns.py:26
Functionpixel_cross_entropy
(gt, pred, lengths)
utils/loss.py:45
Functionpreprocess_descriptions
(examples)
utils/sentence.py:44