MCPcopy Create free account

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

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

Methodhas_class_info
(self)
EgoTracks/tracking/dataset/train_datasets/got10k.py:114
Methodhas_class_info
(self)
EgoTracks/tracking/dataset/train_datasets/coco_seq.py:99
Methodhas_class_info
(self)
EgoTracks/tracking/dataset/train_datasets/lasot.py:98
Methodhas_occlusion_info
(self)
EgoTracks/tracking/dataset/base_video_dataset.py:67
Methodhas_occlusion_info
(self)
EgoTracks/tracking/dataset/train_datasets/got10k.py:117
Methodhas_occlusion_info
(self)
EgoTracks/tracking/dataset/train_datasets/lasot.py:101
Methodhas_segmentation_info
(self)
EgoTracks/tracking/dataset/base_video_dataset.py:79
Methodhas_segmentation_info
(self)
EgoTracks/tracking/dataset/base_image_dataset.py:63
Methodhas_segmentation_info
(self)
EgoTracks/tracking/dataset/train_datasets/coco_seq.py:108
Functionheat_distance
(d)
VQ2D/vq2d/tracking/pfilter.py:13
Functionhist
( x, figsize=(7, 7), rotation=None, xlabel=None, ylabel=None, xlim=None, ylim=None
VQ2D/vq2d/stats.py:49
Functioninfer_v_asis
(*args, **kwargs)
MQ/Infer.py:82
Methodinference
(self, video, meta_data)
EgoTracks/tracking/models/multiple_object_tracker.py:17
Methodinference_img_sequence
Run tracking model on a video sequence. Args: video: a torch Tensor that contains a sequence of images, [N, 3, H, W]
EgoTracks/tracking/models/single_object_tracker.py:118
Methodinference_sequence
Run tracking model on a video sequence. Since some dataset stores video as individual frames and each sequence can be quite large, it
EgoTracks/tracking/models/single_object_tracker.py:36
Methodinference_video_handler
( self, video: av.container.Container, meta_data: Dict = None )
EgoTracks/tracking/models/single_object_tracker.py:31
Methodinit_bbox
(self, frame_num=0)
EgoTracks/tracking/dataset/eval_datasets/base_dataset.py:132
Methodinit_info
(self)
EgoTracks/tracking/dataset/eval_datasets/base_dataset.py:124
Methodinit_mask
(self, frame_num=0)
EgoTracks/tracking/dataset/eval_datasets/base_dataset.py:135
Methodinit_tracker
(self, img, meta)
EgoTracks/tracking/models/multiple_object_tracker.py:29
Methodinit_tracker
This function is used to initilize the tracking model, typically the starting frame. This is mostly designed for SOT, but can also be
EgoTracks/tracking/models/tracker.py:235
Methodinit_tracker
This function is used to initilize SOT with first frame annotation Args: image: this is one image where we used to initi
EgoTracks/tracking/models/stark_tracker/stark_tracker.py:67
Methodinit_weights
(m)
NLQ/VSLNet/model/VSLNet.py:104
Functioniou_with_anchors
Compute jaccard score between a box and the anchors.
MQ/Utils/dataset.py:192
Methodis_synthetic_video_dataset
Returns whether the dataset contains real videos or synthetic returns: bool - True if a video dataset
EgoTracks/tracking/dataset/base_video_dataset.py:41
Methodis_video_sequence
Returns whether the dataset is a video dataset or an image dataset returns: bool - True if a video dataset
EgoTracks/tracking/dataset/base_video_dataset.py:33
Functionjittered_center_crop
For each frame in frames, extracts a square crop centered at box_extract, of area search_area_factor^2 times box_extract area. The extracted crops
EgoTracks/tracking/dataset/processing/processing_utils.py:124
Methodlength
(self)
VQ2D/vq2d/structures.py:54
Methodlist
(self)
EgoTracks/tracking/utils/tensor.py:206
Functionload_lines
(filename)
NLQ/VSLNet/utils/data_util.py:25
Functionload_pretrain
(model, pretrained_path)
EgoTracks/tracking/utils/load_helper.py:38
Functionload_text
(path, delimiter=" ", dtype=np.float32, backend="numpy")
EgoTracks/tracking/utils/load_text.py:51
Methodlost_track
(self)
VQ2D/vq2d/tracking/kys.py:61
Functionltr_collate
Puts each data field into a tensor with outer dimension batch size
EgoTracks/tracking/dataset/dataloader.py:22
Functionltr_collate_stack1
Puts each data field into a tensor. The tensors are stacked at dim=1 to form the batch
EgoTracks/tracking/dataset/dataloader.py:83
FunctionmIoU
(y_pred, y_gt)
EgoTracks/tracking/metrics/miou.py:19
Functionmain
(args)
VQ2D/train_siam_rcnn.py:227
Functionmake_heat_adjusted
(sigma)
VQ2D/vq2d/tracking/pfilter.py:12
Functionmove_forward
(vis)
VQ3D/camera_pose_estimation/Visualization/visualize_render_images.py:69
Functionmultinomial_resample
(weights)
VQ2D/vq2d/tracking/pfilter.py:70
Functionnetwork
(sample)
NLQ/2D-TAN/moment_localization/train.py:153
Functionnetwork
(sample)
NLQ/2D-TAN/moment_localization/test.py:195
Methodnext_frame
Return the next frame, and increment internal counter. Returns image: Next H x W image. status: True or False depen
VQ3D/camera_pose_estimation/SuperGlueMatching/models/utils.py:175
Functionno_processing
(data)
EgoTracks/tracking/dataset/trackingdataset.py:7
Functionobserve
create observation hypothesis given a particle and the current observed frame. each row in x contains one particle info. One row of x = [
VQ2D/vq2d/tracking/particle_filter.py:19
Functionon_end
(state)
NLQ/2D-TAN/moment_localization/train.py:261
Functionon_forward
(state)
NLQ/2D-TAN/moment_localization/train.py:194
Functionon_start
(state)
NLQ/2D-TAN/moment_localization/train.py:186
Functionon_test_end
(state)
NLQ/2D-TAN/moment_localization/train.py:289
Functionon_test_end
(state)
NLQ/2D-TAN/moment_localization/test.py:262
Functionon_test_forward
(state)
NLQ/2D-TAN/moment_localization/train.py:279
Functionon_test_forward
(state)
NLQ/2D-TAN/moment_localization/test.py:252
Functionon_test_start
(state)
NLQ/2D-TAN/moment_localization/train.py:266
Functionon_test_start
(state)
NLQ/2D-TAN/moment_localization/test.py:245
Functionon_update
(state)
NLQ/2D-TAN/moment_localization/train.py:198
Functionoplist
(*args, **kwargs)
EgoTracks/tracking/utils/tensor.py:236
Functionpad_bboxes
(frame_bbox_dict, frame_numbers)
EgoTracks/tracking/utils/utils.py:69
Functionparameters
(yaml_name: str)
EgoTracks/tracking/models/stark_tracker/params.py:30
Functionparse_VQ3D_queries
(filename: str)
VQ3D/depth_estimation/prepare_inputs_for_depth_estimation.py:38
Functionparse_opt
()
MQ/Utils/opts.py:3
Functionrank
(pred, gt)
NLQ/2D-TAN/lib/core/eval.py:29
Functionread_command_line
()
NLQ/VSLNet/options.py:11
Functionrefine_poses
(inputs)
VQ3D/camera_pose_estimation/sfm_api_wsuperglue.py:286
Functionresample
(weights)
VQ2D/vq2d/tracking/pfilter.py:75
Functionresidual_resample
(weights)
VQ2D/vq2d/tracking/pfilter.py:33
Functionresnet101
r"""ResNet-101 model from `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_ Args: pretrained (b
EgoTracks/tracking/models/stark_tracker/resnet.py:235
Functionresnet152
r"""ResNet-152 model from `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_ Args: pretrained (b
EgoTracks/tracking/models/stark_tracker/resnet.py:247
Functionresnet18
r"""ResNet-18 model from `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_ Args: pretrained (bo
EgoTracks/tracking/models/stark_tracker/resnet.py:205
Functionresnet34
r"""ResNet-34 model from `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_ Args: pretrained (bo
EgoTracks/tracking/models/stark_tracker/resnet.py:215
Functionresnet50
r"""ResNet-50 model from `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_ Args: pretrained (bo
EgoTracks/tracking/models/stark_tracker/resnet.py:225
Functionresnext101_32x8d
r"""ResNeXt-101 32x8d model from `"Aggregated Residual Transformation for Deep Neural Networks" <https://arxiv.org/pdf/1611.05431.pdf>`_ Args:
EgoTracks/tracking/models/stark_tracker/resnet.py:273
Functionresnext50_32x4d
r"""ResNeXt-50 32x4d model from `"Aggregated Residual Transformation for Deep Neural Networks" <https://arxiv.org/pdf/1611.05431.pdf>`_ Args:
EgoTracks/tracking/models/stark_tracker/resnet.py:259
Functionrestore_from
(model, optimizer, ckpt_path)
EgoTracks/tracking/utils/load_helper.py:63
Methodroll
(self)
EgoTracks/tracking/dataset/transforms.py:228
Methodroll
(self)
EgoTracks/tracking/dataset/transforms.py:279
Methodroll
(self)
EgoTracks/tracking/dataset/transforms.py:309
Functionrun
Runs a function from a child process. Args: local_rank (int): rank of the current process on the current machine. num_proc (i
EgoTracks/tracking/utils/multiprocessing.py:12
Methodrun_model
(self, img)
EgoTracks/tracking/models/multiple_object_tracker.py:21
Methodrun_model
Most tracking models run frame by frame. This function runs a single forward of the tracking model. Args: image: this is
EgoTracks/tracking/models/tracker.py:205
Methodrun_model
Run one step of inference from a given search image. Args: img: search image in cv2 format. multiscale_list:
EgoTracks/tracking/models/stark_tracker/stark_tracker.py:138
Functionrun_single_process
(clip_uid: str, params: ExtractFramesWorkflowParams)
EgoTracks/tools/preprocess/extract_ego4d_clip_frames.py:29
Functionrun_train
( local_rank, main_func, params, num_machines, num_gpus_per_machine, machine_rank,
EgoTracks/tools/train_net.py:162
Functionsample_fn
(n)
VQ2D/vq2d/tracking/pfilter.py:154
Functionsave_lines
(data, filename)
NLQ/VSLNet/utils/data_util.py:30
Methodsave_off
(self, filename: str)
VQ3D/annotation_API/API/bounding_box.py:106
Functionsave_scores
(scores, data, dataset_name, split)
NLQ/2D-TAN/moment_localization/test.py:70
Functionscale_im_height
(image, H)
VQ3D/VQ3D/scripts/run.py:15
Methodscore
(self)
VQ2D/vq2d/structures.py:58
Methodset_default_values
(self, default_vals: Dict)
EgoTracks/tracking/models/stark_tracker/params.py:9
Functionsetup_environment
()
EgoTracks/tracking/utils/env.py:8
Functionsquared_error
RBF kernel, supporting masked values in the observation Parameters: ----------- x : array (N,D) array of values y : array (N,D) a
VQ2D/vq2d/tracking/pfilter.py:91
Functionstack_tensors
(x)
EgoTracks/tracking/dataset/processing/stark_processing.py:8
Functionstratified_resample
(weights)
VQ2D/vq2d/tracking/pfilter.py:27
Functionsystematic_resample
(weights)
VQ2D/vq2d/tracking/pfilter.py:21
Methodtarget_class
(self, frame_num=None)
EgoTracks/tracking/dataset/eval_datasets/base_dataset.py:171
Methodtemporal_extent
(self)
VQ2D/vq2d/structures.py:46
Functiontensor_operation
(op)
EgoTracks/tracking/utils/tensor.py:231
Functiontest_collate_fn
(data)
NLQ/VSLNet/utils/data_loader.py:80
Methodto_json
(self)
VQ2D/vq2d/structures.py:95
Functiontrain_collate_fn
(data)
NLQ/VSLNet/utils/data_loader.py:26
← previousnext →1,001–1,100 of 1,142, ranked by callers