MCPcopy Create free account

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

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

Method__radd__
(self, other)
EgoTracks/tracking/utils/tensor.py:72
Method__repr__
(self)
EgoTracks/tracking/dataset/transforms.py:85
Method__repr__
(self)
EgoTracks/tracking/dataset/eval_datasets/base_dataset.py:177
Method__repr__
(self)
EgoTracks/tracking/models/stark_tracker/utils/misc.py:58
Method__repr__
(self)
VQ2D/vq2d/structures.py:82
Method__rmatmul__
(self, other)
EgoTracks/tracking/utils/tensor.py:148
Method__rmod__
(self, other)
EgoTracks/tracking/utils/tensor.py:167
Method__rmul__
(self, other)
EgoTracks/tracking/utils/tensor.py:110
Method__rsub__
(self, other)
EgoTracks/tracking/utils/tensor.py:91
Method__rtruediv__
(self, other)
EgoTracks/tracking/utils/tensor.py:129
Method__str__
(self)
EgoTracks/tracking/utils/meters.py:24
Method__sub__
(self, other)
EgoTracks/tracking/utils/tensor.py:86
Method__truediv__
(self, other)
EgoTracks/tracking/utils/tensor.py:124
Method_forward_test
(self, cls_pred_enc, reg_pred_enc, cls_pred_dec, reg_pred_dec)
MQ/Models/VSGN.py:58
Function_gen_detection_video
(video_name, result, idx_classes, opt, num_prop=200, topk = 2)
MQ/Evaluation/ego4d/generate_detection.py:62
Function_gen_retrieval_video
(video_name, result, classes, idx_classes, opt, test_anno, num_prop=200, topk = 2)
MQ/Evaluation/ego4d/generate_retrieval.py:76
Method_iterable
(a)
EgoTracks/tracking/utils/tensor.py:227
Method_load_from_state_dict
( self, state_dict, prefix, local_metadata, strict, missing_ke
EgoTracks/tracking/models/stark_tracker/backbone.py:31
Methodapply
(self, fn)
EgoTracks/tracking/utils/tensor.py:212
Methodapply_attr
(*args, **kwargs)
EgoTracks/tracking/utils/tensor.py:27
Methodapply_attr
(*args, **kwargs)
EgoTracks/tracking/utils/tensor.py:221
Methodattribute
(self, attr: str, *args)
EgoTracks/tracking/utils/tensor.py:37
Methodattribute
(self, attr: str, *args)
EgoTracks/tracking/utils/tensor.py:209
Functionaug_apply
apply augmentation :param bbox: original bbox in image :param param: augmentation param, shift/scale :param shape: image shape, h, w,
EgoTracks/tracking/utils/bbox_helper.py:99
Functionbarplot
( x, y, figsize=(7, 7), rotation=None, xlabel=None, ylabel=None, title=None, a
VQ2D/vq2d/stats.py:10
Methodbboxes
(self)
VQ2D/vq2d/structures.py:50
Functionbce_rescale_loss
(scores, masks, targets, cfg)
NLQ/2D-TAN/lib/models/loss.py:4
Functionbinary_cross_entropy
Same as `torch.nn.functional.binary_cross_entropy`, but returns 0 (instead of nan) for empty inputs. `flip_class` inverts the target since th
VQ2D/detectron2_extensions/layers/wrappers.py:7
Functionbox_xyxy_to_xywh
(x)
EgoTracks/tracking/models/stark_tracker/utils/box_ops.py:19
Functionbuild_starks
(cfg)
EgoTracks/tracking/models/stark_tracker/stark_s.py:146
Functioncalculate_metrics
(args)
EgoTracks/tools/eval_net.py:69
Methodcat_boxlist
Concatenates a list of BoxList (having the same image size) into a single BoxList Arguments: bboxes (list[BoxLis
MQ/Models/ActionGenerator.py:59
Functioncauchy_noise
Apply diagonal covaraiance Cauchy-distributed noise to the N,D array x. Parameters: ----------- x : array (N,D) array of v
VQ2D/vq2d/tracking/pfilter.py:127
Functioncheck_valid
(tensor, type_name)
EgoTracks/tracking/models/stark_tracker/transformer.py:27
Methodcleanup
(self)
VQ3D/camera_pose_estimation/SuperGlueMatching/models/utils.py:235
Methodclosure
()
NLQ/2D-TAN/lib/core/engine.py:29
Functioncollate_fn
(batch)
NLQ/2D-TAN/lib/datasets/__init__.py:5
Methodcompute
(self, v1:np.ndarray, v2:np.ndarray)
VQ3D/VQ3D/API/metrics.py:14
Methodcompute
(self, t:np.ndarray, box1:BoundingBox, box2:BoundingBox)
VQ3D/VQ3D/API/metrics.py:26
Functioncompute_average_precision_detection
Compute average precision (detection task) between ground truth and predictions data frames. If multiple predictions occurs for the same predi
MQ/Evaluation/ego4d/eval_action_detection.py:90
Functioncompute_average_precision_detection
Compute average precision (detection task) between ground truth and predictions data frames. If multiple predictions occurs for the same predi
MQ/Evaluation/ego4d/eval_detection.py:218
Functioncompute_clip_statistics
Compute the total number of unique clips annotated.
VQ2D/vq2d/stats.py:145
Functioncompute_coverage_statistics
Compute the video hours across all annotations.
VQ2D/vq2d/stats.py:118
Methodcompute_loss
(scores, labels, mask, epsilon=1e-12)
NLQ/VSLNet/model/layers.py:452
Functioncompute_query_statistics
Compute the total number of visual queries annotated.
VQ2D/vq2d/stats.py:164
Functioncompute_query_to_response_separation_statistics
Compute the set of query to response separation in # frames.
VQ2D/vq2d/stats.py:236
Functioncompute_response_track_length_statistics
Compute the set of response track lengths in # frames.
VQ2D/vq2d/stats.py:259
Functioncompute_response_track_location_statistics
Compute the set of response track lengths in # frames.
VQ2D/vq2d/stats.py:282
Functioncompute_scenario_coverage_statistics
Compute the video hours annotated per scenario. Note: We only count scenarios for which we have more than `thresh` hours of annotations.
VQ2D/vq2d/stats.py:185
Functioncompute_university_coverage_statistics
Compute the video hours annotated per university.
VQ2D/vq2d/stats.py:213
Methodconcat
(self, other)
EgoTracks/tracking/utils/tensor.py:188
Functionconvert_2d_to_3d
(u, v, z, K)
VQ3D/camera_pose_estimation/utils.py:225
Functionconvert_annot_to_bbox
(annot: Dict[str, Any])
VQ2D/vq2d/baselines/utils.py:31
Functionconvert_image_np2torch
image - (H, W, 3) numpy array
VQ2D/vq2d/baselines/utils.py:19
Methodcopy
(self)
EgoTracks/tracking/utils/tensor.py:191
Methodcreate_traj_azure
(self, output_traj, K, Ci_T_G=None)
VQ3D/VQ3D/API/get_query_3d_ground_truth.py:52
Functioncxy_wh_2_rect
(pos, sz)
EgoTracks/tracking/utils/bbox_helper.py:50
Functioncxywh_2_xywh
Center x, y to top-left x, y. Args: bbox: bounding box with [x, y, w, h], x,y are center position of the box. context_ratio:
EgoTracks/tracking/utils/bbox_helper.py:198
Functioneval_ego4d_lt_tracking
(model, cfg)
EgoTracks/tools/eval_datasets/eval_ego4d_lt_tracking.py:35
Methodevaluate
Evaluates a prediction df. For the detection task we measure the interpolated mean average precision to measure the performance of a m
MQ/Evaluation/ego4d/eval_action_detection.py:73
Functionevaluation_detection
(opt)
MQ/Evaluation/ego4d/get_detect_performance.py:19
Functionevaluation_retrieval
(opt)
MQ/Evaluation/ego4d/get_retrieval_performance.py:146
Methodextract_features
(self, textual_input, textual_mask, visual_input)
NLQ/2D-TAN/lib/models/tan.py:28
Functionextract_frames
(frame_numbers: List, container: av.container.Container)
EgoTracks/tracking/utils/utils.py:48
Functionextract_frames_by_range
( start: int, end: int, container: av.container.Container )
EgoTracks/tracking/utils/utils.py:58
Functionextract_frames_for_sfm
(inputs)
VQ3D/camera_pose_estimation/extract_frames_for_colmap.py:12
Methodextract_index
(start_logits, end_logits)
NLQ/VSLNet/model/layers.py:548
Methodforward
(self, kpts, scores)
VQ3D/camera_pose_estimation/SuperGlueMatching/models/superglue.py:80
Methodforward
(self, query, key, value)
VQ3D/camera_pose_estimation/SuperGlueMatching/models/superglue.py:102
Methodforward
(self, x, source)
VQ3D/camera_pose_estimation/SuperGlueMatching/models/superglue.py:117
Methodforward
(self, desc0, desc1)
VQ3D/camera_pose_estimation/SuperGlueMatching/models/superglue.py:130
Methodforward
Run SuperGlue on a pair of keypoints and descriptors
VQ3D/camera_pose_estimation/SuperGlueMatching/models/superglue.py:228
Methodforward
Run SuperPoint (optionally) and SuperGlue SuperPoint is skipped if ['keypoints0', 'keypoints1'] exist in input Args: data:
VQ3D/camera_pose_estimation/SuperGlueMatching/models/matching.py:56
Methodforward
Compute keypoints, scores, descriptors for image
VQ3D/camera_pose_estimation/SuperGlueMatching/models/superpoint.py:145
Methodforward
(self, loc_box, feat_frmlvl)
MQ/Models/BoundaryAdjust.py:28
Methodforward
(self, input, num_frms, gt_action= None, gt_start= None, gt_end= None, gt_bbox = None, num_gt = None)
MQ/Models/VSGN.py:66
Methodforward
(self, input, num_frms)
MQ/Models/XGPN.py:99
Methodforward
(self, x)
MQ/Models/Head.py:56
Methodforward
(self, feat_prop, neigh_idx)
MQ/Models/GCNs.py:57
Methodforward
(self, x, num_frms)
MQ/Models/GCNs.py:100
Methodforward
Forward pass with input x.
EgoTracks/tracking/models/stark_tracker/head.py:85
Methodforward
Forward pass with input x.
EgoTracks/tracking/models/stark_tracker/head.py:176
Methodforward
(self, x)
EgoTracks/tracking/models/stark_tracker/head.py:238
Methodforward
( self, img=None, seq_dict=None, mode="backbone", run_box_head=False,
EgoTracks/tracking/models/stark_tracker/stark_st.py:40
Methodforward
( self, img=None, seq_dict=None, mode="backbone", run_box_head=True,
EgoTracks/tracking/models/stark_tracker/stark_s.py:50
Methodforward
:param feat: (H1W1+H2W2, bs, C) :param mask: (bs, H1W1+H2W2) :param query_embed: (N, C) or (N, B, C) :param pos_embe
EgoTracks/tracking/models/stark_tracker/transformer.py:101
Methodforward
( self, src, mask: Optional[Tensor] = None, src_key_padding_mask: Optional[Ten
EgoTracks/tracking/models/stark_tracker/transformer.py:158
Methodforward
(self, seq_dict, return_intermediate=False, part_att=False)
EgoTracks/tracking/models/stark_tracker/transformer.py:208
Methodforward
( self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: O
EgoTracks/tracking/models/stark_tracker/transformer.py:237
Methodforward
( self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: Optional
EgoTracks/tracking/models/stark_tracker/transformer.py:350
Methodforward
(self, seq_dict, part_att=False)
EgoTracks/tracking/models/stark_tracker/transformer.py:433
Methodforward
( self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: O
EgoTracks/tracking/models/stark_tracker/transformer.py:551
Methodforward
(self, x)
EgoTracks/tracking/models/stark_tracker/backbone.py:55
Methodforward
(self, tensor_list: NestedTensor)
EgoTracks/tracking/models/stark_tracker/backbone.py:107
Methodforward
(self, tensor_list: NestedTensor, mode=None)
EgoTracks/tracking/models/stark_tracker/backbone.py:156
Methodforward
(self, x)
EgoTracks/tracking/models/stark_tracker/resnet.py:182
Methodforward
(self, tensor_list: NestedTensor)
EgoTracks/tracking/models/stark_tracker/position_encoding.py:30
Methodforward
(self, tensor_list: NestedTensor)
EgoTracks/tracking/models/stark_tracker/position_encoding.py:86
Methodforward
(self, tensor_list: NestedTensor)
EgoTracks/tracking/models/stark_tracker/position_encoding.py:117
Methodforward
(self, textual_input, textual_mask, visual_input)
NLQ/2D-TAN/lib/models/tan.py:18
← previousnext →801–900 of 1,142, ranked by callers