MCPcopy Create free account

hub / github.com/YueLiao/gen-vlkt / functions

Functions209 in github.com/YueLiao/gen-vlkt

⨍ Functions 209◇ Types & classes 39
↓ 1 callersFunctiontop_k
(li, k)
util/topk.py:16
↓ 1 callersFunctiontrain_one_epoch
(model: torch.nn.Module, criterion: torch.nn.Module, data_loader: Iterable, optimizer: tor
engine.py:16
↓ 1 callersMethodtriplet_nms_filter
(self, preds)
generate_vcoco_official.py:212
↓ 1 callersMethodtriplet_nms_filter
(self, preds)
datasets/hico_eval_triplet.py:303
↓ 1 callersMethodtriplet_nms_filter
(self, preds)
datasets/hico_eval.py:350
↓ 1 callersMethodvoc_ap
(self, rec, prec)
datasets/hico_eval_triplet.py:209
↓ 1 callersMethodvoc_ap
(self, rec, prec)
datasets/hico_eval.py:249
↓ 1 callersMethodvoc_ap
(self, rec, prec)
datasets/vcoco_eval.py:144
Method__call__
(self, img, target)
datasets/transforms.py:155
Method__call__
(self, img: PIL.Image.Image, target: dict)
datasets/transforms.py:165
Method__call__
(self, img, target)
datasets/transforms.py:176
Method__call__
(self, img, target)
datasets/transforms.py:188
Method__call__
(self, img, target=None)
datasets/transforms.py:200
Method__call__
(self, img, target)
datasets/transforms.py:209
Method__call__
(self, img, target)
datasets/transforms.py:225
Method__call__
(self, img, target)
datasets/transforms.py:232
Method__call__
(self, img, target)
datasets/transforms.py:241
Method__call__
(self, image, target=None)
datasets/transforms.py:250
Method__call__
(self, image, target)
datasets/transforms.py:268
Method__call__
(self, img, target)
datasets/transforms.py:285
Method__getattr__
(self, attr)
util/misc.py:176
Method__getitem__
(self, idx)
datasets/vcoco.py:44
Method__getitem__
(self, idx)
datasets/hico.py:82
Method__init__
(self, backbone, transformer, num_obj_classes, num_verb_classes, num_queries, args=None)
generate_vcoco_official.py:27
Method__init__
(self, num_queries, subject_category_id, correct_mat, args)
generate_vcoco_official.py:131
Method__init__
(self, size)
datasets/transforms.py:152
Method__init__
(self, min_size: int, max_size: int)
datasets/transforms.py:161
Method__init__
(self, size)
datasets/transforms.py:173
Method__init__
(self, p=0.5)
datasets/transforms.py:185
Method__init__
(self, sizes, max_size=None)
datasets/transforms.py:195
Method__init__
(self, max_pad)
datasets/transforms.py:206
Method__init__
(self, transforms1, transforms2, p=0.5)
datasets/transforms.py:220
Method__init__
(self, *args, **kwargs)
datasets/transforms.py:238
Method__init__
(self, mean, std)
datasets/transforms.py:246
Method__init__
(self, transforms)
datasets/transforms.py:265
Method__init__
(self, brightness=0, contrast=0, saturatio=0, hue=0)
datasets/transforms.py:282
Method__init__
(self, preds, gts, rare_triplets, non_rare_triplets, correct_mat, args)
datasets/hico_eval_triplet.py:12
Method__init__
(self, preds, gts, rare_triplets, non_rare_triplets, correct_mat, args)
datasets/hico_eval.py:7
Method__init__
(self, preds, gts, correct_mat, use_nms_filter=False)
datasets/vcoco_eval.py:9
Method__init__
(self, img_set, img_folder, anno_file, transforms, num_queries, args)
datasets/vcoco.py:17
Method__init__
(self, img_set, img_folder, anno_file, clip_feats_folder, transforms, num_queries, args)
datasets/hico.py:22
Method__init__
(self, encoder_layer, num_layers, norm=None)
models/gen.py:85
Method__init__
(self, decoder_layer, num_layers, norm=None, return_intermediate=False)
models/gen.py:109
Method__init__
(self, d_model, nhead, dim_feedforward=2048, dropout=0.1, activation="relu", normalize_before
models/gen.py:154
Method__init__
(self, d_model, nhead, dim_feedforward=2048, dropout=0.1, activation="relu", normalize_before
models/gen.py:214
Method__init__
(self, cost_obj_class: float = 1, cost_verb_class: float = 1, cost_bbox: float = 1, cost_giou
models/matcher.py:9
Method__init__
(self, backbone, transformer, num_queries, aux_loss=False, args=None)
models/gen_vlkt.py:25
Method__init__
(self, num_obj_classes, num_queries, num_verb_classes, matcher, weight_dict, eos_coef, losses, args)
models/gen_vlkt.py:209
Method__init__
(self, args)
models/gen_vlkt.py:420
Method__init__
(self, n)
models/backbone.py:31
Method__init__
(self, backbone: nn.Module, train_backbone: bool, num_channels: int, return_interm_layers: bool)
models/backbone.py:63
Method__init__
(self, name: str, train_backbone: bool, return_interm_layers: bool,
models/backbone.py:88
Method__init__
(self, num_pos_feats=256)
models/position_encoding.py:58
Method__init__
(self, window_size=20, fmt=None)
util/misc.py:37
Method__init__
(self, delimiter="\t")
util/misc.py:165
Method__init__
(self, tensors, mask: Optional[Tensor])
util/misc.py:310
Method__len__
(self)
datasets/vcoco.py:41
Method__len__
(self)
datasets/hico.py:79
Method__repr__
(self)
datasets/transforms.py:273
Method__repr__
(self)
util/misc.py:328
Method__str__
(self)
util/misc.py:85
Method__str__
(self)
util/misc.py:184
Method_load_from_state_dict
(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys
models/backbone.py:38
Functionall_gather
Run all_gather on arbitrary picklable data (not necessarily tensors) Args: data: any picklable object Returns: list[data]
util/misc.py:94
Methodavg
(self)
util/misc.py:69
Functionbuild
(image_set, args)
datasets/vcoco.py:202
Functionbuild
(image_set, args)
datasets/hico.py:264
Methodcal_prec
(self, rec, prec, t=0.8)
datasets/hico_eval_triplet.py:202
Methodcal_prec
(self, rec, prec, t=0.8)
datasets/hico_eval.py:242
Methodclip_preds_boxes
(self, preds)
datasets/hico_eval_triplet.py:445
Methodclip_preds_boxes
(self, preds)
datasets/hico_eval.py:499
Functioncollate_fn
(batch)
util/misc.py:274
Methodevaluate
(self)
datasets/hico_eval_triplet.py:113
Methodevaluate
(self)
datasets/hico_eval.py:159
Methodforward
(self, samples: NestedTensor)
generate_vcoco_official.py:80
Methodforward
(self, x)
generate_vcoco_official.py:123
Methodforward
(self, outputs, target_sizes)
generate_vcoco_official.py:155
Methodforward
(self, src, mask, query_embed_h, query_embed_o, pos_guided_embed, pos_embed)
models/gen.py:48
Methodforward
(self, src, mask: Optional[Tensor] = None, src_key_padding_mask: Optional[Tens
models/gen.py:91
Methodforward
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: Optional[T
models/gen.py:116
Methodforward
(self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: Optional[
models/gen.py:203
Methodforward
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: Optional[T
models/gen.py:283
Methodforward
(self, outputs, targets)
models/matcher.py:20
Methodforward
(self, samples: NestedTensor, is_training=True)
models/gen_vlkt.py:117
Methodforward
(self, x)
models/gen_vlkt.py:201
Methodforward
(self, outputs, targets)
models/gen_vlkt.py:384
Methodforward
(self, outputs, target_sizes)
models/gen_vlkt.py:425
Methodforward
(self, x)
models/backbone.py:48
Methodforward
(self, tensor_list: NestedTensor)
models/backbone.py:75
Methodforward
(self, tensor_list: NestedTensor)
models/backbone.py:103
Methodforward
(self, tensor_list: NestedTensor)
models/position_encoding.py:31
Methodforward
(self, tensor_list: NestedTensor)
models/position_encoding.py:68
Functionget_sha
()
util/misc.py:254
Methodglobal_avg
(self)
util/misc.py:74
Functioninit_distributed_mode
(args)
util/misc.py:376
Methodloss_hoi_labels
(self, outputs, targets, indices, num_interactions, topk=5)
models/gen_vlkt.py:267
Methodloss_obj_cardinality
(self, outputs, targets, indices, num_interactions)
models/gen_vlkt.py:244
Methodloss_obj_labels
(self, outputs, targets, indices, num_interactions, log=True)
models/gen_vlkt.py:226
Methodloss_sub_obj_boxes
(self, outputs, targets, indices, num_interactions)
models/gen_vlkt.py:294
Methodloss_verb_labels
(self, outputs, targets, indices, num_interactions)
models/gen_vlkt.py:253
← previousnext →101–200 of 209, ranked by callers