MCPcopy Create free account

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

Functions209 in github.com/YueLiao/gen-vlkt

⨍ Functions 209◇ Types & classes 39
↓ 80 callersFunctionsave_HICO
(HICO, HICO_dir, classid, begin, finish)
tools/covert_annot_for_official_eval.py:28
↓ 55 callersFunctionprint
(*args, **kwargs)
util/misc.py:339
↓ 36 callersMethodto
(self, device)
util/misc.py:314
↓ 19 callersMethodmax
(self)
util/misc.py:78
↓ 12 callersFunctionbox_cxcywh_to_xyxy
(x)
util/box_ops.py:15
↓ 11 callersMethodupdate
(self, **kwargs)
util/misc.py:169
↓ 6 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
models/gen.py:234
↓ 4 callersMethod__init__
(self, d_model=512, nhead=8, num_encoder_layers=6, num_dec_layers=3, dim_feedforward=2048, dr
models/gen.py:11
↓ 4 callersMethod_get_src_permutation_idx
(self, indices)
models/gen_vlkt.py:360
↓ 4 callersMethodadd_meter
(self, name, meter)
util/misc.py:196
↓ 4 callersFunctiongeneralized_box_iou
Generalized IoU from https://giou.stanford.edu/ The boxes should be in [x0, y0, x1, y1] format Returns a [N, M] pairwise matrix, where
util/box_ops.py:46
↓ 4 callersFunctionis_dist_avail_and_initialized
()
util/misc.py:347
↓ 3 callersMethod__init__
(self, input_dim, hidden_dim, output_dim, num_layers)
models/gen_vlkt.py:195
↓ 3 callersMethod__init__
(self, backbone, position_embedding)
models/backbone.py:100
↓ 3 callersFunction_run
(command)
util/misc.py:257
↓ 3 callersFunctionbuild_dataset
(image_set, args)
datasets/__init__.py:7
↓ 3 callersFunctioncrop
(image, target, region)
datasets/transforms.py:19
↓ 3 callersFunctionget_world_size
()
util/misc.py:355
↓ 3 callersMethodlog_every
(self, iterable, print_freq, header=None)
util/misc.py:199
↓ 3 callersFunctionnested_tensor_from_tensor_list
(tensor_list: List[Tensor])
util/misc.py:289
↓ 3 callersFunctionsift
(li, low, higt)
util/topk.py:1
↓ 2 callersMethod__init__
(self, input_dim, hidden_dim, output_dim, num_layers)
generate_vcoco_official.py:117
↓ 2 callersFunction_get_activation_fn
Return an activation function given a string
models/gen.py:314
↓ 2 callersFunction_get_clones
(module, N)
models/gen.py:297
↓ 2 callersMethod_neg_loss
Modified focal loss. Exactly the same as CornerNet. Runs faster and costs a little bit more memory
models/gen_vlkt.py:335
↓ 2 callersFunctionbuild_backbone
(args)
models/backbone.py:115
↓ 2 callersFunctionbuild_gen
(args)
models/gen.py:301
↓ 2 callersMethoddecompose
(self)
util/misc.py:325
↓ 2 callersFunctionevaluate_hoi
(dataset_file, model, postprocessors, data_loader, subject_category_id, device, args)
engine.py:80
↓ 2 callersMethodget_loss
(self, loss, outputs, targets, indices, num, **kwargs)
models/gen_vlkt.py:366
↓ 2 callersFunctionset_hoi
(box_annotations, hoi_annotations, verb_classes)
tools/convert_vcoco_annotations.py:29
↓ 2 callersMethodsynchronize_between_processes
(self)
util/misc.py:192
↓ 2 callersMethodupdate
(self, value, n=1)
util/misc.py:45
↓ 2 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
models/gen.py:171
↓ 1 callersFunctionFormat_Pred
(pred_file)
tools/covert_annot_for_official_eval.py:7
↓ 1 callersFunctionGenerate_HICO_detection
(output_file, HICO_dir)
tools/covert_annot_for_official_eval.py:52
↓ 1 callersMethod__init__
(self, num_pos_feats=64, temperature=10000, normalize=False, scale=None)
models/position_encoding.py:20
↓ 1 callersFunction_max_by_axis
(the_list)
util/misc.py:280
↓ 1 callersMethod_reset_parameters
(self)
models/gen.py:43
↓ 1 callersMethod_set_aux_loss_triplet
(self, outputs_hoi_class, outputs_obj_class, outputs_sub_coord, outputs_obj_coor
models/gen_vlkt.py:174
↓ 1 callersFunction_sigmoid
(x)
models/gen_vlkt.py:19
↓ 1 callersFunctionaccuracy
Computes the precision@k for the specified values of k
util/misc.py:403
↓ 1 callersMethodbbox_clip
(self, box, size)
datasets/hico_eval_triplet.py:462
↓ 1 callersMethodbbox_clip
(self, box, size)
datasets/hico_eval.py:516
↓ 1 callersFunctionbox_iou
(boxes1, boxes2)
util/box_ops.py:30
↓ 1 callersFunctionbox_xyxy_to_cxcywh
(x)
util/box_ops.py:22
↓ 1 callersFunctionbuild
(args)
models/gen_vlkt.py:462
↓ 1 callersFunctionbuild_matcher
(args)
models/matcher.py:76
↓ 1 callersFunctionbuild_model
(args)
models/__init__.py:4
↓ 1 callersFunctionbuild_position_encoding
(args)
models/position_encoding.py:82
↓ 1 callersMethodcompute_IOU
(self, bbox1, bbox2)
datasets/hico_eval_triplet.py:281
↓ 1 callersMethodcompute_IOU
(self, bbox1, bbox2)
datasets/hico_eval.py:321
↓ 1 callersMethodcompute_IOU
(self, bbox1, bbox2)
datasets/vcoco_eval.py:224
↓ 1 callersMethodcompute_fptp
(self, pred_hois, gt_hois, match_pairs, bbox_overlaps)
datasets/hico_eval_triplet.py:219
↓ 1 callersMethodcompute_fptp
(self, pred_hois, gt_hois, match_pairs, pred_bboxes, bbox_overlaps)
datasets/hico_eval.py:259
↓ 1 callersMethodcompute_fptp
(self, pred_hois, gt_hois, match_pairs, pred_bboxes, bbox_overlaps)
datasets/vcoco_eval.py:154
↓ 1 callersMethodcompute_iou_mat
(self, bbox_list1, bbox_list2)
datasets/hico_eval_triplet.py:256
↓ 1 callersMethodcompute_iou_mat
(self, bbox_list1, bbox_list2)
datasets/hico_eval.py:296
↓ 1 callersMethodcompute_iou_mat
(self, bbox_list1, bbox_list2)
datasets/vcoco_eval.py:199
↓ 1 callersMethodcompute_map
(self)
datasets/hico_eval_triplet.py:135
↓ 1 callersMethodcompute_map
(self)
datasets/hico_eval.py:191
↓ 1 callersMethodcompute_map
(self)
datasets/vcoco_eval.py:107
↓ 1 callersMethodevaluate
(self)
datasets/vcoco_eval.py:90
↓ 1 callersMethodforward_post
(self, src, src_mask: Optional[Tensor] = None,
models/gen.py:174
↓ 1 callersMethodforward_post
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask:
models/gen.py:237
↓ 1 callersMethodforward_pre
(self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: O
models/gen.py:189
↓ 1 callersMethodforward_pre
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: Op
models/gen.py:260
↓ 1 callersFunctiongenerate
(model, post_processor, data_loader, device, verb_classes, missing_category_id)
generate_vcoco_official.py:392
↓ 1 callersMethodgenerate_vcoco_text_label
(self)
generate_vcoco_official.py:66
↓ 1 callersFunctionget_args
()
tools/convert_parameters.py:7
↓ 1 callersFunctionget_args
()
tools/convert_vcoco_annotations.py:11
↓ 1 callersFunctionget_args_parser
()
generate_vcoco_official.py:284
↓ 1 callersFunctionget_args_parser
()
main.py:20
↓ 1 callersFunctionget_rank
()
util/misc.py:361
↓ 1 callersFunctionget_size
(image_size, size, max_size=None)
datasets/transforms.py:102
↓ 1 callersFunctionget_size_with_aspect_ratio
(image_size, size, max_size=None)
datasets/transforms.py:82
↓ 1 callersFunctionhflip
(image, target)
datasets/transforms.py:62
↓ 1 callersMethodinit_classifier_with_CLIP
(self, hoi_text_label, obj_text_label, unseen_index)
models/gen_vlkt.py:88
↓ 1 callersFunctioninterpolate
Equivalent to nn.functional.interpolate, but with support for empty batch sizes. This will eventually be supported natively by PyTorch, and t
util/misc.py:421
↓ 1 callersFunctionis_main_process
()
util/misc.py:367
↓ 1 callersMethodload_correct_mat
(self, path)
datasets/vcoco.py:164
↓ 1 callersMethodload_correct_mat
(self, path)
datasets/hico.py:227
↓ 1 callersFunctionmain
(args)
generate_vcoco_official.py:343
↓ 1 callersFunctionmain
(args)
main.py:158
↓ 1 callersFunctionmain
(args)
tools/convert_parameters.py:28
↓ 1 callersFunctionmain
(args)
tools/convert_vcoco_annotations.py:70
↓ 1 callersFunctionmake_hico_transforms
(image_set)
datasets/hico.py:232
↓ 1 callersFunctionmake_vcoco_transforms
(image_set)
datasets/vcoco.py:169
↓ 1 callersFunctionpad
(image, target, padding)
datasets/transforms.py:138
↓ 1 callersMethodpairwise_nms
(self, subs, objs, scores)
generate_vcoco_official.py:242
↓ 1 callersMethodpairwise_nms
(self, subs, objs, scores)
datasets/hico_eval_triplet.py:342
↓ 1 callersMethodpairwise_nms
(self, subs, objs, scores)
datasets/hico_eval.py:390
↓ 1 callersMethodpairwise_soft_nms
(self, subs, objs, scores)
datasets/hico_eval_triplet.py:382
↓ 1 callersMethodpairwise_soft_nms
(self, subs, objs, scores)
datasets/hico_eval.py:435
↓ 1 callersMethodreset_parameters
(self)
generate_vcoco_official.py:63
↓ 1 callersMethodreset_parameters
(self)
models/gen_vlkt.py:85
↓ 1 callersMethodreset_parameters
(self)
models/position_encoding.py:64
↓ 1 callersFunctionresize
(image, target, size, max_size=None)
datasets/transforms.py:79
↓ 1 callersMethodset_rare_hois
(self, anno_file)
datasets/hico.py:188
↓ 1 callersFunctionsetup_for_distributed
This function disables printing when not in master process
util/misc.py:332
next →1–100 of 209, ranked by callers