MCPcopy Create free account

hub / github.com/Seung-Hun-Lee/LOMM / functions

Functions839 in github.com/Seung-Hun-Lee/LOMM

↓ 90 callersMethodcat
Concatenates a list of Boxes into a single Boxes Arguments: boxes_list (list[Boxes]) Returns: Boxes:
lomm/data_video/utils.py:175
↓ 88 callersMethodget
(self)
demo_video/predictor.py:314
↓ 41 callersMethodto
(self, device)
mask2former/utils/misc.py:30
↓ 38 callersMethodto
(self, device: torch.device)
lomm/data_video/utils.py:96
↓ 30 callersMethodclone
Clone the Boxes. Returns: Boxes
lomm/data_video/utils.py:86
↓ 28 callersMethodinfo
Print information about the annotation file. :return:
mask2former_video/data_video/datasets/ytvis_api/ytvos.py:99
↓ 20 callersMethodinfo
Print information about the annotation file. :return:
lomm/data_video/datasets/ytvis_api/ytvos.py:105
↓ 11 callersFunctionretry_if_cuda_oom
Makes a function retry itself after encountering pytorch's CUDA OOM error. It will first retry after calling `torch.cuda.empty_cache()`.
mask2former_video/utils/memory.py:27
↓ 10 callersFunction_isArrayLike
(obj)
lomm/data_video/datasets/ytvis_api/ytvos.py:50
↓ 10 callersFunction_isArrayLike
(obj)
mask2former_video/data_video/datasets/ytvis_api/ytvos.py:44
↓ 10 callersMethodarea
Computes the area of all the boxes. Returns: torch.Tensor: a vector with areas of each box.
lomm/data_video/utils.py:101
↓ 9 callersMethod__init__
(self, dim=768)
mask2former/modeling/backbones_samAdapter/adapter.py:84
↓ 9 callersMethod__init__
(self, dim=768)
mask2former/modeling/backbones_vitAdapter/adapter.py:84
↓ 7 callersMethod__init__
( self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.0 )
mask2former/modeling/backbone/swin.py:24
↓ 7 callersMethoddevice
(self)
lomm/lomm.py:166
↓ 6 callersMethod__init__
Args: prob (float): probability of flip. horizontal (boolean): whether to apply horizontal flipping verti
lomm/data_video/augmentation.py:381
↓ 6 callersMethodoutput_shape
(self)
mask2former/modeling/backbone/swin.py:760
↓ 6 callersMethodprepare_tokens_with_masks
(self, x, masks=None, return_HW=False)
mask2former/modeling/backbones_samAdapter/backbones.py:208
↓ 6 callersMethodprepare_tokens_with_masks
(self, x, masks=None, return_HW=False)
mask2former/modeling/backbones_vitAdapter/backbones.py:205
↓ 6 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
mask2former/modeling/transformer_decoder/transformer.py:258
↓ 5 callersMethodmatch_embds
(self, ref_embds, cur_embds)
lomm/lomm_tracker.py:345
↓ 5 callersMethodmatch_embds
(self, ref_embds, cur_embds)
lomm/lomm_tracker.py:806
↓ 4 callersMethod__init__
( self, d_model=512, nhead=8, num_encoder_layers=6, num_decoder_layers
mask2former/modeling/transformer_decoder/transformer.py:20
↓ 4 callersMethod__init__
(self, input_dim, hidden_dim, output_dim, num_layers)
mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:195
↓ 4 callersMethod__init__
(self, input_dim, hidden_dim, output_dim, num_layers)
mask2former_video/modeling/transformer_decoder/video_mask2former_transformer_decoder.py:196
↓ 4 callersFunction_get_activation_fn
Return an activation function given a string
mask2former_video/modeling/transformer_decoder/video_mask2former_transformer_decoder.py:182
↓ 4 callersMethodforward_features
(self, features)
mask2former/modeling/pixel_decoder/fpn.py:136
↓ 4 callersMethodgetAnnIds
Get ann ids that satisfy given filter conditions. default skips that filter :param vidIds (int array) : get anns for given vids
lomm/data_video/datasets/ytvis_api/ytvos.py:113
↓ 4 callersMethodgetAnnIds
Get ann ids that satisfy given filter conditions. default skips that filter :param vidIds (int array) : get anns for given vids
mask2former_video/data_video/datasets/ytvis_api/ytvos.py:107
↓ 4 callersMethodget_loss
(self, loss, outputs, targets, indices, num_masks)
lomm/criterion_lomm.py:271
↓ 4 callersMethodget_loss
(self, loss, outputs, targets, indices, num_masks)
lomm/criterion_lomm_E.py:254
↓ 4 callersFunctionis_dist_avail_and_initialized
()
mask2former/utils/misc.py:106
↓ 4 callersMethodloadAnns
Load anns with the specified ids. :param ids (int array) : integer ids specifying anns :return: anns (object array) : l
lomm/data_video/datasets/ytvis_api/ytvos.py:184
↓ 4 callersMethodloadAnns
Load anns with the specified ids. :param ids (int array) : integer ids specifying anns :return: anns (object array) : l
mask2former_video/data_video/datasets/ytvis_api/ytvos.py:178
↓ 4 callersMethodlosses
(self, predictions, targets)
mask2former/modeling/meta_arch/per_pixel_baseline.py:114
↓ 4 callersMethodmatch_from_embds
(self, tgt_embds, cur_embds)
lomm/lomm.py:268
↓ 4 callersMethodnonempty
Find boxes that are non-empty. A box is considered empty, if either of its side is no larger than threshold. Returns:
lomm/data_video/utils.py:126
↓ 4 callersMethodprepare_targets
(self, targets, images)
lomm/lomm.py:399
↓ 4 callersFunctionregister_ytvis_instances
Register a dataset in YTVIS's json annotation format for instance tracking. Args: name (str): the name that identifies a dataset
lomm/data_video/datasets/ytvis.py:331
↓ 4 callersMethodtrain
Convert the model into training mode while keep layers freezed.
mask2former/modeling/backbone/swin.py:680
↓ 4 callersMethodwith_pos_embed
(self, tensor, pos)
lomm/lomm_tracker.py:34
↓ 4 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:95
↓ 4 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
mask2former_video/modeling/transformer_decoder/video_mask2former_transformer_decoder.py:96
↓ 3 callersMethod__init__
Args: backbone: a backbone module, must follow detectron2's backbone interface sem_seg_head: a module that predicts s
lomm/lomm.py:36
↓ 3 callersMethod__init__
NOTE: this interface is experimental. Args: input_shape: shapes (channels and stride) of the input features t
mask2former/modeling/pixel_decoder/msdeformattn.py:167
↓ 3 callersMethod__init__
Args: dim (int): Number of input channels. num_heads (int): Number of attention heads in each ViT block.
mask2former/modeling/backbones_samAdapter/sam_modeling/image_encoder.py:193
↓ 3 callersFunction_get_activation_fn
Return an activation function given a string
mask2former/modeling/transformer_decoder/transformer.py:361
↓ 3 callersFunction_get_activation_fn
Return an activation function given a string
mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:181
↓ 3 callersFunction_get_clones
(module, N)
mask2former/modeling/transformer_decoder/transformer.py:357
↓ 3 callersMethod_get_src_permutation_idx
(self, indices)
lomm/criterion_lomm.py:259
↓ 3 callersMethod_get_src_permutation_idx
(self, indices)
lomm/criterion_lomm_E.py:242
↓ 3 callersMethod_reset_parameters
(self)
mask2former/modeling/pixel_decoder/fpn.py:186
↓ 3 callersMethod_separate_heads
(self, x: Tensor, num_heads: int)
mask2former/modeling/backbones_samAdapter/sam_modeling/transformer.py:208
↓ 3 callersFunctionadd_dvis_config
(cfg)
lomm/config.py:26
↓ 3 callersFunctionadd_lomm_config
(cfg)
lomm/config.py:12
↓ 3 callersFunctionadd_maskformer2_config
Add config for MASK_FORMER.
mask2former/config.py:6
↓ 3 callersFunctionadd_maskformer2_video_config
(cfg)
mask2former_video/config.py:6
↓ 3 callersFunctionadd_minvis_config
(cfg)
lomm/config.py:19
↓ 3 callersFunctionms_deform_attn_core_pytorch
(value, value_spatial_shapes, sampling_locations, attention_weights)
mask2former/modeling/pixel_decoder/ops/functions/ms_deform_attn_func.py:52
↓ 3 callersMethodput
(self, image)
demo_video/predictor.py:310
↓ 3 callersMethodrun_on_video
Args: frames (List[np.ndarray]): a list of images of shape (H, W, C) (in BGR order). This is the format used by O
demo_video/predictor.py:120
↓ 2 callersMethod__init__
( self, hidden_channel=256, feedforward_channel=2048, num_head=8, deco
lomm/lomm_tracker.py:420
↓ 2 callersMethod__init__
NOTE: this interface is experimental. Args: input_shape: shapes (channels and stride) of the input features c
mask2former/modeling/pixel_decoder/fpn.py:40
↓ 2 callersMethod__init__
( self, embedding_dim: int, num_heads: int, downsample_rate: int = 1, )
mask2former/modeling/backbones_samAdapter/sam_modeling/transformer.py:191
↓ 2 callersMethod_derive_coco_results
Derive the desired score numbers from summarized COCOeval. Args: coco_eval (None or COCOEval): None represents no predict
mask2former_video/data_video/ytvis_eval.py:193
↓ 2 callersFunction_evaluate_predictions_on_coco
Evaluate the coco results using COCOEval API.
mask2former_video/data_video/ytvis_eval.py:296
↓ 2 callersMethod_freeze_stages
(self)
mask2former/modeling/backbone/swin.py:618
↓ 2 callersFunction_get_objects_from_outputs
(outputs)
demo_video/predictor.py:21
↓ 2 callersMethod_get_src_permutation_idx
(self, indices)
mask2former/modeling/criterion.py:192
↓ 2 callersMethod_get_src_permutation_idx
(self, indices)
mask2former_video/modeling/criterion.py:230
↓ 2 callersFunction_get_ytvis_2021_instances_meta
()
lomm/data_video/datasets/ytvis.py:176
↓ 2 callersFunction_ignore_torch_cuda_oom
A context which ignores CUDA OOM exception from pytorch.
mask2former_video/utils/memory.py:13
↓ 2 callersMethod_jitter
Randomly modifies given color to produce a slightly different color than the color given. Args: color (tuple[double]): a
demo_video/visualizer.py:25
↓ 2 callersMethod_pe_encoding
Positionally encode points that are normalized to [0,1].
mask2former/modeling/backbones_samAdapter/sam_modeling/prompt_encoder.py:185
↓ 2 callersMethod_set_aux_loss
(self, outputs_class, outputs_seg_masks)
lomm/lomm_tracker.py:822
↓ 2 callersMethod_set_aux_loss
(self, outputs_class, outputs_seg_masks)
lomm/refiner.py:162
↓ 2 callersMethodannToRLE
Convert annotation which can be polygons, uncompressed RLE to RLE. :return: binary mask (numpy 2D array)
lomm/data_video/datasets/ytvis_api/ytvos.py:268
↓ 2 callersMethodannToRLE
Convert annotation which can be polygons, uncompressed RLE to RLE. :return: binary mask (numpy 2D array)
mask2former_video/data_video/datasets/ytvis_api/ytvos.py:262
↓ 2 callersFunctionbuild_augmentation
(cfg, is_train)
mask2former_video/data_video/augmentation.py:115
↓ 2 callersFunctionbuild_detection_train_loader
Build a dataloader for object detection with some default features. This interface is experimental. Args: dataset (list or torch
lomm/data_video/build.py:146
↓ 2 callersFunctionbuild_pixel_decoder
Build a pixel decoder from `cfg.MODEL.MASK_FORMER.PIXEL_DECODER_NAME`.
mask2former/modeling/pixel_decoder/fpn.py:21
↓ 2 callersFunctioncalculate_uncertainty
We estimate uncerainty as L1 distance between 0.0 and the logit prediction in 'logits' for the foreground class in `classes`. Args:
mask2former_video/modeling/criterion.py:73
↓ 2 callersMethodconvert2ytvis
(self, dataset_dict)
lomm/data_video/dataset_mapper_vps.py:179
↓ 2 callersMethodconvert2ytvis
(self, dataset_dict)
lomm/data_video/dataset_mapper_vss.py:152
↓ 2 callersMethodcreateIndex
(self)
lomm/data_video/datasets/ytvis_api/ytvos.py:74
↓ 2 callersMethodcreateIndex
(self)
mask2former_video/data_video/datasets/ytvis_api/ytvos.py:68
↓ 2 callersMethoddraw_instance_predictions
Draw instance-level prediction results on an image. Args: predictions (Instances): the output of an instance detection/se
demo_video/visualizer.py:50
↓ 2 callersFunctiondrop_add_residual_stochastic_depth
( x: Tensor, residual_func: Callable[[Tensor], Tensor], sample_drop_ratio: float = 0.0, )
mask2former/modeling/backbones_samAdapter/layers/block.py:110
↓ 2 callersFunctiondrop_add_residual_stochastic_depth
( x: Tensor, residual_func: Callable[[Tensor], Tensor], sample_drop_ratio: float = 0.0, )
mask2former/modeling/backbones_vitAdapter/layers/block.py:110
↓ 2 callersFunctiondrop_add_residual_stochastic_depth_list
( x_list: List[Tensor], residual_func: Callable[[Tensor, Any], Tensor], sample_drop_ratio: float =
mask2former/modeling/backbones_samAdapter/layers/block.py:181
↓ 2 callersFunctiondrop_add_residual_stochastic_depth_list
( x_list: List[Tensor], residual_func: Callable[[Tensor, Any], Tensor], sample_drop_ratio: float =
mask2former/modeling/backbones_vitAdapter/layers/block.py:181
↓ 2 callersFunctionfilter_empty_instances
Filter out empty instances in an `Instances` object. Args: instances (Instances): by_box (bool): whether to filter out insta
lomm/data_video/dataset_mapper.py:36
↓ 2 callersFunctionfilter_empty_instances
Filter out empty instances in an `Instances` object. Args: instances (Instances): by_box (bool): whether to filter out insta
mask2former_video/data_video/dataset_mapper.py:27
↓ 2 callersMethodforward_prediction_heads
(self, output, mask_features, attn_mask_target_size)
lomm/video_mask2former_transformer_decoder.py:197
↓ 2 callersMethodforward_prediction_heads
(self, output, mask_features, attn_mask_target_size)
mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:433
↓ 2 callersMethodforward_prediction_heads
(self, output, mask_features, attn_mask_target_size)
mask2former_video/modeling/transformer_decoder/video_mask2former_transformer_decoder.py:444
↓ 2 callersMethodforward_with_coords
Positionally encode points that are not normalized to [0,1].
mask2former/modeling/backbones_samAdapter/sam_modeling/prompt_encoder.py:207
↓ 2 callersMethodframe_decoder_loss_reshape
(self, outputs, targets, image_outputs=None)
lomm/lomm.py:1988
↓ 2 callersMethodgetCatIds
filtering parameters. default skips that filter. :param catNms (str array) : get cats for given cat names :param supNms (str
lomm/data_video/datasets/ytvis_api/ytvos.py:141
↓ 2 callersMethodgetCatIds
filtering parameters. default skips that filter. :param catNms (str array) : get cats for given cat names :param supNms (str
mask2former_video/data_video/datasets/ytvis_api/ytvos.py:135
next →1–100 of 839, ranked by callers