MCPcopy Create free account

hub / github.com/EGO4D/episodic-memory / functions

Functions1,142 in github.com/EGO4D/episodic-memory

↓ 1 callersMethod_split_inputs
(self, inputs)
EgoTracks/tracking/dataset/transforms.py:68
↓ 1 callersMethod_stats_new_epoch
(self)
EgoTracks/tools/trainers/starkst_trainer.py:233
↓ 1 callersFunction_update_config
(base_cfg, exp_cfg)
EgoTracks/tracking/models/stark_tracker/config/stark_st2/config.py:287
↓ 1 callersFunction_update_dict
(cfg, value)
NLQ/2D-TAN/lib/core/config.py:89
↓ 1 callersMethod_update_stats
(self, new_stats: OrderedDict, batch_size, loader)
EgoTracks/tools/trainers/starkst_trainer.py:196
↓ 1 callersMethod_whctrs
(self, anchor)
MQ/Models/AnchorGenerator.py:55
↓ 1 callersFunctionadd_path
(path)
NLQ/2D-TAN/moment_localization/_init_paths.py:16
↓ 1 callersMethodadd_template
Add a new object template to the list of templates. Args: image: the image to be used as a template bbox: th
EgoTracks/tracking/models/template.py:82
↓ 1 callersMethodadjust
EgoTracks/tracking/models/stark_tracker/stark_s.py:126
↓ 1 callersFunctionangle_error_mat
(R1, R2)
VQ3D/camera_pose_estimation/SuperGlueMatching/models/utils.py:377
↓ 1 callersFunctionangle_error_vec
(v1, v2)
VQ3D/camera_pose_estimation/SuperGlueMatching/models/utils.py:383
↓ 1 callersFunctionapprox_equal_durations
(dur1, dur2, thresh=1.0)
VQ2D/tools/validate_extracted_clips.py:13
↓ 1 callersFunctionattention
(query, key, value)
VQ3D/camera_pose_estimation/SuperGlueMatching/models/superglue.py:85
↓ 1 callersFunctionbatchify_video_uids
(video_uids, batch_size)
VQ2D/convert_videos_to_clips.py:122
↓ 1 callersFunctionbatchify_video_uids
(video_uids, batch_size)
VQ2D/convert_videos_to_images.py:95
↓ 1 callersFunctionbox_cxcywh_to_xyxy
(x)
EgoTracks/tracking/models/stark_tracker/utils/box_ops.py:7
↓ 1 callersFunctionbox_iou
:param boxes1: (N, 4) (x1,y1,x2,y2) :param boxes2: (N, 4) (x1,y1,x2,y2) :return:
EgoTracks/tracking/models/stark_tracker/utils/box_ops.py:35
↓ 1 callersFunctionbox_xywh_to_xyxy
(x)
EgoTracks/tracking/models/stark_tracker/utils/box_ops.py:13
↓ 1 callersFunctionbox_xyxy_to_cxcywh
(x)
EgoTracks/tracking/models/stark_tracker/utils/box_ops.py:25
↓ 1 callersMethodbuild_box
(self)
VQ3D/annotation_API/API/bounding_box.py:80
↓ 1 callersFunctionbuild_calculate_metrics_function
(dataset_name)
EgoTracks/tools/eval_datasets/build.py:17
↓ 1 callersFunctionbuild_dataloaders
(cfg, local_rank=-1)
EgoTracks/tracking/dataset/build.py:124
↓ 1 callersFunctionbuild_eval_function
(dataset_name)
EgoTracks/tools/eval_datasets/build.py:13
↓ 1 callersFunctionbuild_loss_function
(cfg)
EgoTracks/tools/trainers/base_trainer.py:25
↓ 1 callersFunctionbuild_optimizer_and_scheduler
(model, configs)
NLQ/VSLNet/model/VSLNet.py:19
↓ 1 callersFunctionbuild_optimizer_scheduler
(cfg, model)
EgoTracks/tracking/solver/build.py:7
↓ 1 callersFunctionbuild_position_encoding
(cfg)
EgoTracks/tracking/models/stark_tracker/position_encoding.py:123
↓ 1 callersFunctionbuild_scheduler
(cfg, optimizer)
EgoTracks/tracking/solver/build.py:69
↓ 1 callersFunctionbuild_siam_model
(cfg)
VQ2D/train_siam_rcnn.py:186
↓ 1 callersMethodcache_text_feature
(self, sentence, rnd=False)
NLQ/2D-TAN/lib/datasets/ego4d_clip.py:218
↓ 1 callersMethodcal_loss
(self, start_offsets, end_offsets, anchors, gt_bbox, num_gt)
MQ/Models/BoundaryAdjust.py:66
↓ 1 callersFunctioncheck_clip
(clip_uid, clip_annotations, clip_metadata, args)
VQ2D/tools/validate_extracted_clips.py:17
↓ 1 callersFunctioncheck_inf
(tensor)
EgoTracks/tracking/models/stark_tracker/transformer.py:19
↓ 1 callersFunctioncheck_nan
(tensor)
EgoTracks/tracking/models/stark_tracker/transformer.py:23
↓ 1 callersMethodclear
(self)
EgoTracks/tools/trainers/base_trainer.py:288
↓ 1 callersMethodclear
(self)
EgoTracks/tools/trainers/base_trainer.py:310
↓ 1 callersMethodcls_loss_func
(self, cls_pred, cls_labels)
MQ/Models/Loss.py:79
↓ 1 callersFunctioncombine_clip_info
(params)
EgoTracks/tools/preprocess/extract_ego4d_clip_frames.py:94
↓ 1 callersMethodcombine_last_two_dim
(x)
NLQ/VSLNet/model/layers.py:277
↓ 1 callersFunctioncompute_IoU
Compute the IoU given predicted and ground truth windows.
NLQ/VSLNet/utils/evaluate_ego4d_nlq.py:35
↓ 1 callersFunctioncompute_average_flops
(args)
VQ2D/vq2d/tools/get_average_detector_flops.py:102
↓ 1 callersFunctioncompute_average_precision_detection
Compute average precision (detection task) between ground truth and predictions data frames. If multiple predictions occurs for the same predi
VQ2D/vq2d/metrics/spatio_temporal_metrics.py:93
↓ 1 callersFunctioncompute_average_precision_detection
Compute average precision (detection task) between ground truth and predictions data frames. If multiple predictions occurs for the same predi
VQ2D/vq2d/metrics/temporal_metrics.py:99
↓ 1 callersMethodcompute_cross_entropy_loss
(start_logits, end_logits, start_labels, end_labels)
NLQ/VSLNet/model/layers.py:567
↓ 1 callersFunctioncompute_epipolar_error
(kpts0, kpts1, T_0to1, K0, K1)
VQ3D/camera_pose_estimation/SuperGlueMatching/models/utils.py:355
↓ 1 callersMethodcompute_highlight_loss
(self, scores, labels, mask)
NLQ/VSLNet/model/VSLNet.py:140
↓ 1 callersMethodcompute_losses
(self, pred_dict, gt_bbox, return_status=True)
EgoTracks/tools/trainers/starkst_trainer.py:128
↓ 1 callersMethodcompute_losses
(self, pred_dict, labels, return_status=True)
EgoTracks/tools/trainers/starkst_trainer.py:263
↓ 1 callersFunctioncompute_overlap
(pred, gt)
NLQ/VSLNet/utils/data_util.py:84
↓ 1 callersFunctioncompute_overlaps
(y_pred, y_gt)
EgoTracks/tracking/metrics/miou.py:5
↓ 1 callersFunctioncompute_pose_error
(T_0to1, R, t)
VQ3D/camera_pose_estimation/SuperGlueMatching/models/utils.py:388
↓ 1 callersFunctioncompute_success
Compute success %, i.e., the % of cases where there is sufficient overlap between ground truth and predictions data frames. If multiple prediction
VQ2D/vq2d/metrics/success_metrics.py:99
↓ 1 callersFunctioncompute_tracking_metrics
Compute % frames from the ground truth that were accurately recovered by the predictions. If multiple predictions occurs for the same predicte
VQ2D/vq2d/metrics/tracking_metrics.py:103
↓ 1 callersMethodconvert
(self, data_dir, predictor=None)
NLQ/VSLNet/utils/data_gen.py:81
↓ 1 callersFunctionconvert_2d_to_3d
(u, v, z, K)
VQ3D/camera_pose_estimation/sfm_api_wsuperglue.py:16
↓ 1 callersFunctionconvert_2d_to_3d
(u, v, z, K)
VQ3D/camera_pose_estimation/SuperGlueMatching/extract_visual_database.py:25
↓ 1 callersFunctionconvert_annotations_to_clipwise_list
(annotations)
VQ2D/extract_vq_detection_scores.py:241
↓ 1 callersFunctionconvert_annotations_to_clipwise_list
(annotations)
VQ2D/perform_vq_inference.py:329
↓ 1 callersFunctionconvert_ego4d_dataset
Convert the Ego4D dataset for VSLNet.
NLQ/VSLNet/utils/prepare_ego4d_dataset.py:87
↓ 1 callersFunctionconvert_tall_c3d_features
(sampling_rate)
NLQ/2D-TAN/data/TACoS/merge_npys_to_hdf5.py:8
↓ 1 callersFunctioncreate_executor
(configs)
NLQ/VSLNet/main.py:258
↓ 1 callersFunctioncreate_logger
(cfg, cfg_name, tag='train')
NLQ/2D-TAN/lib/core/utils.py:27
↓ 1 callersFunctioncreate_similarity_network
(pretrained: bool = True)
VQ2D/vq2d/baselines/utils.py:49
↓ 1 callersFunctioncreate_traj_azure
(output_traj, K, Ci_T_G=None)
VQ3D/camera_pose_estimation/Visualization/visualize_render_images.py:19
↓ 1 callersFunctioncustom_draw_geometry_with_camera_trajectory
(pcd, output_path='', output_path_depth='',
VQ3D/camera_pose_estimation/Visualization/visualize_render_images.py:46
↓ 1 callersMethodcycle_dataset
Do a cycle of training or validation.
EgoTracks/tools/trainers/starkst_trainer.py:40
↓ 1 callersMethoddecompose
(self)
EgoTracks/tracking/models/stark_tracker/utils/misc.py:55
↓ 1 callersFunctiondeep_metric
x - (N, F) numpy array y - (1, F) numpy array
VQ2D/vq2d/tracking/particle_filter.py:68
↓ 1 callersMethoddisplay
(self, batch)
EgoTracks/tracking/utils/meters.py:35
↓ 1 callersFunctiondisplay_results
(results, mIoU, thresholds, topK, title=None)
NLQ/VSLNet/utils/evaluate_ego4d_nlq.py:17
↓ 1 callersFunctiondo_train
(cfg, model, resume=False)
VQ2D/train_siam_rcnn.py:85
↓ 1 callersFunctiondraw_border
(image, color=(255, 0, 0), thickness=5)
VQ2D/visualizations/visualize_annotations.py:43
↓ 1 callersMethodencode
(self, gt_boxes, anchors)
MQ/Models/BoxCoder.py:10
↓ 1 callersFunctionerror_colormap
(x)
VQ3D/camera_pose_estimation/SuperGlueMatching/models/utils.py:553
↓ 1 callersFunctionestimate_pose
(kpts0, kpts1, K0, K1, thresh, conf=0.99999)
VQ3D/camera_pose_estimation/SuperGlueMatching/models/utils.py:288
↓ 1 callersFunctioneval_predictions
(segments, data, verbose=True, merge_window=False)
NLQ/2D-TAN/lib/core/eval.py:75
↓ 1 callersFunctionevaluate
(gt_file, pred_file)
VQ2D/evaluate_vq.py:72
↓ 1 callersMethodevaluate
(self)
MQ/Evaluation/ego4d/get_retrieval_performance.py:86
↓ 1 callersMethodevaluate
Evaluates a prediction file. For the detection task we measure the interpolated mean average precision to measure the performance of a
MQ/Evaluation/ego4d/eval_detection.py:200
↓ 1 callersMethodevaluate
Evaluates a prediction file. For the detection task we measure the interpolated average precision to measure the performance of a meth
VQ2D/vq2d/metrics/success_metrics.py:73
↓ 1 callersMethodevaluate
Evaluates a prediction file. For the detection task we measure the interpolated average precision to measure the performance of a meth
VQ2D/vq2d/metrics/spatio_temporal_metrics.py:67
↓ 1 callersMethodevaluate
Evaluates a prediction file. For the detection task we measure the interpolated average precision to measure the performance of a meth
VQ2D/vq2d/metrics/tracking_metrics.py:73
↓ 1 callersMethodevaluate
Evaluates a prediction file. For the detection task we measure the interpolated mean average precision to measure the performance of a
VQ2D/vq2d/metrics/temporal_metrics.py:71
↓ 1 callersFunctionevaluate_nlq_performance
Evalutes the performances.
NLQ/VSLNet/utils/evaluate_ego4d_nlq.py:59
↓ 1 callersMethodextract
(self, filename: str, output_dir: str)
VQ3D/camera_pose_estimation/Camera_Intrinsics_API/extract_frames.py:5
↓ 1 callersFunctionextract_clip
Extracts clips from a video Save path format: {save_root}/{clip_uid}.mp4 Args: video_path - path to video clip_data - a
VQ2D/convert_videos_to_clips.py:59
↓ 1 callersFunctionextract_clip_bboxes_and_scores
Given a visual crop and frames from a clip, retrieve the bounding box proposal from each frame that is most similar to the visual crop.
VQ2D/extract_vq_detection_scores.py:29
↓ 1 callersFunctionextract_clip_frame_nos
Extracts frame numbers corresponding to the VQ annotation for a given clip Args: video_md - a dictionary of video metadata c
VQ2D/convert_videos_to_images.py:60
↓ 1 callersFunctionextract_clip_ids
(file_path: str)
EgoTracks/tools/preprocess/extract_ego4d_clip_frames.py:62
↓ 1 callersFunctionextract_crop_from_image
(image, box)
VQ2D/visualizations/visualize_annotations.py:16
↓ 1 callersFunctionextract_detection_scores
(annotations, cfg)
VQ2D/extract_vq_detection_scores.py:200
↓ 1 callersMethodextract_index
(self, start_logits, end_logits)
NLQ/VSLNet/model/VSLNet.py:135
↓ 1 callersFunctionfetch_graph
(g, i, f_idx, start_i)
VQ3D/camera_pose_estimation/superglue_tracker.py:268
↓ 1 callersFunctionfilter_checkpoints
(model_dir, suffix="t7", max_to_keep=5)
NLQ/VSLNet/utils/runner_utils.py:27
↓ 1 callersFunctionfilter_glove_embedding
(word_dict, glove_path)
NLQ/VSLNet/utils/data_gen.py:110
↓ 1 callersFunctionformat_predictions
(annotations, predicted_rts)
VQ2D/perform_vq_inference.py:361
↓ 1 callersMethodforward_and_compute_loss
(self, data)
EgoTracks/tools/trainers/starkst_trainer.py:76
↓ 1 callersMethodforward_head
hs: output embeddings (1, B, N, C) memory: encoder embeddings (HW1+HW2, B, C)
EgoTracks/tracking/models/stark_tracker/stark_st.py:74
↓ 1 callersMethodforward_post
( self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: Optional
EgoTracks/tracking/models/stark_tracker/transformer.py:310
↓ 1 callersMethodforward_post
seq_dict: sequence dict of both the search region and the template (concatenated)
EgoTracks/tracking/models/stark_tracker/transformer.py:396
← previousnext →301–400 of 1,142, ranked by callers