MCPcopy Create free account

hub / github.com/Yang-Li-2000/Touch-Line-Transformer / functions

Functions656 in github.com/Yang-Li-2000/Touch-Line-Transformer

↓ 277 callersMethodappend
(self, e)
scripts/clevr/clevr_to_coco.py:221
↓ 248 callersFunctionprint
(*args, **kwargs)
util/dist.py:139
↓ 82 callersFunction_do_test
(qid, target_caption, target_objs)
scripts/clevr/clevr_to_coco_test.py:9
↓ 47 callersMethodupdate
(self, **kwargs)
util/metrics.py:80
↓ 41 callersMethodto
(self, *args, **kwargs)
util/misc.py:91
↓ 32 callersMethodfind
(self, x)
scripts/pre-training/flickr_combined.py:199
↓ 28 callersMethodfind
(self, x)
scripts/utils/unionfind.py:17
↓ 24 callersMethodappend
(self, e)
scripts/clevr/refclevr_to_coco.py:249
↓ 20 callersMethod_summarize
(self, summary_type, iou_thr=None, area_rng="all", freq_group_idx=None)
datasets/lvis_eval.py:501
↓ 16 callersFunctionconsolidate_spans
Accepts a list of spans and the the corresponding caption. Returns a cleaned list of spans where: - Overlapping spans are merged -
scripts/utils/spans.py:34
↓ 15 callersFunction_make_detr
( backbone_name: str, num_queries=100, mask=False, qa_dataset=None, predict_final=False,
hubconf.py:32
↓ 15 callersFunctionmake_coco_transforms
(image_set, cautious)
datasets/coco.py:203
↓ 15 callersMethodmax
(self)
util/metrics.py:62
↓ 7 callersMethod__init__
(self, num_channels)
models/backbone.py:132
↓ 7 callersMethodfrom_tensor_list
(cls, tensor_list, do_round=False)
util/misc.py:100
↓ 7 callersFunctionpoint_in_box
(p, b)
datasets/phrasecut_utils/data_transfer.py:83
↓ 7 callersMethodprepare
(self, predictions)
datasets/lvis_eval.py:889
↓ 6 callersMethod__init__
(self, input_dim, hidden_dim, output_dim, num_layers)
models/mdetr.py:1113
↓ 6 callersFunctionbox_area
Computes the area of a set of bounding boxes, which are specified by its (x1, y1, x2, y2) coordinates. Args: boxes (Tensor[N, 4]
datasets/flickr_eval.py:151
↓ 6 callersFunctionbuild_model
(args)
models/__init__.py:6
↓ 6 callersFunctionbuild_postprocessors
(args, dataset_name)
models/postprocessors.py:340
↓ 6 callersFunctionevaluate
( model: torch.nn.Module, criterion: Optional[torch.nn.Module], contrastive_criterion:
engine.py:304
↓ 6 callersFunctionfind_tokens
(refexp, template, node_id, backtrack=True, partial_match=False)
scripts/clevr/refclevr_to_coco.py:368
↓ 6 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:39
↓ 6 callersFunctionpolygons_to_mask
(polygons, w, h)
datasets/phrasecut_utils/data_transfer.py:52
↓ 6 callersMethodsummarize
Compute and display summary metrics for evaluation results.
datasets/lvis_eval.py:526
↓ 6 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
models/transformer.py:410
↓ 5 callersMethod__init__
( self, d_model=512, nhead=8, num_encoder_layers=6,
models/transformer.py:33
↓ 5 callersMethod_get_src_permutation_idx
(self, indices)
models/mdetr.py:829
↓ 5 callersFunctionbuild_dataset
(dataset_file: str, image_set: str, args)
datasets/__init__.py:33
↓ 5 callersFunctionis_same_fun
The function names in programs and templates don't always match 1:1, so we redefine equality
scripts/clevr/clevr_to_coco.py:295
↓ 5 callersFunctionis_same_fun
The function names in programs and templates don't always match 1:1, so we redefine equality
scripts/clevr/refclevr_to_coco.py:323
↓ 5 callersMethodlog_every
(self, iterable, print_freq, header=None, path=None)
util/metrics.py:107
↓ 5 callersMethodupdate
(self, res: List[Dict])
scripts/eval_gqa.py:161
↓ 4 callersMethod__init__
(self, d_model=256, nhead=8, num_encoder_layers=6, num_decoder_layers=6, dim_feedforward=204
models/transformer_ori.py:21
↓ 4 callersFunctionbox_cxcywh_to_xyxy
(x)
util/box_ops.py:10
↓ 4 callersFunctionbox_iou
Return intersection-over-union (Jaccard index) of boxes. Both sets of boxes are expected to be in ``(x1, y1, x2, y2)`` format with ``0 <
scripts/pre-training/flickr_combined.py:165
↓ 4 callersFunctionboxes_to_mask
(boxes, w, h, xywh=True)
datasets/phrasecut_utils/data_transfer.py:68
↓ 4 callersMethoddecompose
(self)
util/misc.py:96
↓ 4 callersMethodfilter_ids
(self, ids)
scripts/clevr/refclevr_to_coco.py:290
↓ 4 callersFunctionfind_node_id_in_template
Given a node idx in the given program, try to find the corresponding node in the template
scripts/clevr/refclevr_to_coco.py:319
↓ 4 callersFunctionfind_tokens
(question, template: Dict, node_id, backtrack=True)
scripts/clevr/clevr_to_coco.py:330
↓ 4 callersMethodget_ann_ids
Get ann ids that satisfy given filter conditions. Args: img_ids (int array): get anns for given imgs cat_ids (int arra
datasets/lvis.py:71
↓ 4 callersFunctionget_pose_loss
(arm, arm_class, target_arm, idx=0)
models/mdetr.py:1043
↓ 4 callersFunctionget_root_and_nouns
Given a sentence, returns a tuple with the following items: -- root text:str : the text associated with the root of the sentence -- negative_
scripts/utils/text.py:28
↓ 4 callersMethodload_anns
Load anns with the specified ids. If ids=None load all anns. Args: ids (int array): integer array of annotation ids Return
datasets/lvis.py:129
↓ 4 callersFunctionpoint_in_seg
(p0, p1, pt)
datasets/phrasecut_utils/data_transfer.py:101
↓ 4 callersFunctionremove_chars
(pos, amount)
scripts/utils/spans.py:94
↓ 4 callersFunctionxywh_to_xyxy
Convert [x y w h] box format to [x1 y1 x2 y2] format.
datasets/phrasecut_utils/data_transfer.py:10
↓ 3 callersFunction_isArrayLike
(obj)
datasets/lvis.py:15
↓ 3 callersMethod_load_helper
(self, _dict, ids)
datasets/lvis.py:121
↓ 3 callersFunction_run
(command)
util/misc.py:20
↓ 3 callersMethodaccumulate
Accumulate per image evaluation results and store the result in self.eval.
datasets/lvis_eval.py:388
↓ 3 callersMethodadd_meter
(self, name, meter)
util/metrics.py:104
↓ 3 callersFunctionboxes_region
:return: [x_min, y_min, x_max, y_max] of all boxes
datasets/phrasecut_utils/data_transfer.py:26
↓ 3 callersFunctioncombine_boxes
Given a set of boxes, returns the average of all clusters of boxes that are highly overlapping. A high overlap is characterized by the iou_thresho
scripts/utils/boxes.py:61
↓ 3 callersFunctionconvert
Do the heavy lifting on the given subset (eg 'train')
scripts/clevr/refclevr_to_coco.py:785
↓ 3 callersFunctionconvert
( subset: str, flickr_path: Path, output_path: Path, merge_ground_truth: bool, next_img_id: int = 0, next_
scripts/pre-training/flickr_combined.py:238
↓ 3 callersFunctionconvert_to_xywh
(boxes)
datasets/lvis_eval.py:956
↓ 3 callersFunctioncreate_positive_map
construct a map such that positive_map[i,j] = True iff box i is associated to token j
datasets/coco.py:80
↓ 3 callersFunctioncrop
(image, target, region)
datasets/transforms.py:19
↓ 3 callersFunctionfind_node_id_in_template
Given a node idx in the given program, try to find the corresponding node in the template
scripts/clevr/clevr_to_coco.py:291
↓ 3 callersFunctionget_init_file
(args)
run_with_submitit_gqa_eval.py:43
↓ 3 callersFunctionget_init_file
(args)
run_with_submitit_lvis_eval.py:43
↓ 3 callersFunctionget_init_file
(args)
run_with_submitit.py:44
↓ 3 callersFunctionget_shared_folder
(args)
run_with_submitit_gqa_eval.py:34
↓ 3 callersFunctionget_shared_folder
(args)
run_with_submitit_lvis_eval.py:34
↓ 3 callersFunctionget_shared_folder
(args)
run_with_submitit.py:35
↓ 3 callersFunctionis_dist_avail_and_initialized
Returns: True if distributed training is enabled
util/dist.py:147
↓ 3 callersMethodload_imgs
Load categories with the specified ids. If ids=None load all images. Args: ids (int array): integer array of image ids Ret
datasets/lvis.py:148
↓ 3 callersFunctionmake_clevr_transforms
(image_set, cautious=False)
datasets/clevr.py:129
↓ 3 callersFunctionpoints_in_line
(p1, p2)
datasets/phrasecut_utils/data_transfer.py:97
↓ 3 callersFunctionpoints_intersect_box
(p0, p1, b)
datasets/phrasecut_utils/data_transfer.py:100
↓ 3 callersFunctionpolygon_to_box
(polygon)
datasets/phrasecut_utils/data_transfer.py:38
↓ 3 callersMethodprint_results
(self)
datasets/lvis_eval.py:568
↓ 3 callersFunctionspan_intersect_span
Returns True if the given spans intersect
scripts/utils/spans.py:13
↓ 3 callersMethodsynchronize_between_processes
(self)
util/metrics.py:100
↓ 3 callersFunctiontargets_to
Moves the target dicts to the given device.
util/misc.py:149
↓ 3 callersMethodupdate
(self, predictions)
datasets/yourefit.py:519
↓ 3 callersMethodwith_pos_embed
(self, tensor, pos_embed: Optional[Tensor])
models/transformer_ori.py:185
↓ 2 callersMethod__init__
Class for reading and visualizing annotations. Args: annotation_path (str): location of annotation file
datasets/lvis.py:20
↓ 2 callersMethod__init__
(self, detr, mask_head="smallconv", freeze_detr=False)
models/segmentation.py:15
↓ 2 callersMethod__len__
(self)
datasets/clevr.py:103
↓ 2 callersMethod_create_index
(self)
datasets/lvis.py:46
↓ 2 callersFunction_encode_answer
(target, answer)
datasets/clevr.py:43
↓ 2 callersFunction_get_activation_fn
Return an activation function given a string
models/transformer_ori.py:233
↓ 2 callersFunction_get_activation_fn
Return an activation function given a string
models/transformer.py:594
↓ 2 callersFunction_get_clones
(module, N)
models/transformer_ori.py:213
↓ 2 callersFunction_get_clones
(module, N)
models/transformer.py:572
↓ 2 callersMethod_get_gt_dt
Create gt, dt which are list of anns/dets. If use_cats is true only anns/dets corresponding to tuple (img_id, cat_id) will be used. El
datasets/lvis_eval.py:255
↓ 2 callersMethod_load_json
(self, path)
datasets/lvis.py:42
↓ 2 callersFunction_merge_lists
(listA, listB, maxN, key)
datasets/lvis_eval.py:725
↓ 2 callersMethod_prepare
Prepare self._gts and self._dts for evaluation based on params.
datasets/lvis_eval.py:176
↓ 2 callersMethod_to_mask
(self, anns, lvis)
datasets/lvis_eval.py:171
↓ 2 callersFunctionadd_alpha_channel
(img,alpha)
vis_attn.py:31
↓ 2 callersMethodadd_negative
Log a negative hit @k for given category
datasets/flickr_eval.py:239
↓ 2 callersMethodadd_positive
Log a positive hit @k for given category
datasets/flickr_eval.py:232
↓ 2 callersFunctionall_gather
Run all_gather on arbitrary picklable data (not necessarily tensors) Args: data: any picklable object Returns: list[data]
util/dist.py:34
↓ 2 callersMethodann_to_rle
Convert annotation which can be polygons, uncompressed RLE to RLE. Args: ann (dict) : annotation object Returns:
datasets/lvis.py:175
↓ 2 callersFunctionbacktrack_previous_nodes
Find previous filtering nodes that might have text associated. The full filter may contain UNION nodes, which means that there might be severa
scripts/clevr/clevr_to_coco.py:331
next →1–100 of 656, ranked by callers