MCPcopy Create free account

hub / github.com/MiliLab/UniGeoSeg / functions

Functions763 in github.com/MiliLab/UniGeoSeg

↓ 1 callersMethod_forward_sa
(self, tgt, query_pos)
unigeoseg/model/multimodal_encoder/sam2/modeling/memory_attention.py:58
↓ 1 callersMethod_generate_masks
(self, image: np.ndarray)
unigeoseg/model/multimodal_encoder/sam2/automatic_mask_generator.py:224
↓ 1 callersFunction_generate_optimizer_class_with_gradient_clipping
Dynamically creates a new type that inherits the type of a given instance and overrides the `step` method to add gradient clipping
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/solver.py:41
↓ 1 callersMethod_get_augmented_inputs
(self, input)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/test_time_augmentation.py:100
↓ 1 callersMethod_get_batch_size
Gets the batch size of the output given the batch size of the input prompts.
unigeoseg/model/multimodal_encoder/sam2/modeling/sam/prompt_encoder.py:139
↓ 1 callersMethod_get_binary_mask
(self, target)
unigeoseg/model/mask_decoder/mask_criterion/pretrain_criterion.py:266
↓ 1 callersMethod_get_binary_mask
(self, target)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/maskformer_train.py:300
↓ 1 callersMethod_get_binary_mask
(self, target)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/criterion.py:236
↓ 1 callersMethod_get_binary_mask
(self, target)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/criterion.py:406
↓ 1 callersMethod_get_device
(self)
unigeoseg/model/multimodal_encoder/sam2/modeling/sam/prompt_encoder.py:157
↓ 1 callersMethod_get_dice
(self, predict, target)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/maskformer_train.py:290
↓ 1 callersMethod_get_empty_mask_ptr
Get a dummy object pointer based on an empty mask on the current frame.
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor_legacy.py:556
↓ 1 callersMethod_get_pos_embed
(self, hw: Tuple[int, int])
unigeoseg/model/multimodal_encoder/hieradet.py:297
↓ 1 callersMethod_get_pos_embed
(self, hw: Tuple[int, int])
unigeoseg/model/multimodal_encoder/sam2/modeling/backbones/hieradet.py:273
↓ 1 callersMethod_get_stability_scores
Compute stability scores of the mask logits based on the IoU between upper and lower thresholds.
unigeoseg/model/multimodal_encoder/sam2/modeling/sam/mask_decoder.py:247
↓ 1 callersMethod_get_targets
(self, gt_masks, num_query, device)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/criterion.py:412
↓ 1 callersMethod_get_tgt_permutation_idx
(self, indices)
unigeoseg/model/mask_decoder/mask_criterion/pretrain_criterion.py:260
↓ 1 callersMethod_get_tgt_permutation_idx
(self, indices)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/criterion.py:230
↓ 1 callersMethod_inference_one_image
Args: input (dict): one dataset dict with "image" field being a CHW tensor Returns: dict: one output dict
unigeoseg/model/mask_decoder/Mask2Former_Simplify/test_time_augmentation.py:71
↓ 1 callersFunction_is_power_of_2
(n)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops/modules/ms_deform_attn.py:28
↓ 1 callersMethod_make_layer
(self, block, planes, blocks, stride=1, dcn=None)
unigeoseg/model/multimodal_projector/builder.py:350
↓ 1 callersMethod_obj_idx_to_id
Map model-side object index to client-side object id.
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor.py:152
↓ 1 callersMethod_prepare_memory_conditioned_features
Fuse the current frame's visual feature map with previous memory.
unigeoseg/model/multimodal_encoder/sam2/modeling/sam2_base.py:497
↓ 1 callersMethod_process_batch
( self, points: np.ndarray, im_size: Tuple[int, ...], crop_box: List[int],
unigeoseg/model/multimodal_encoder/sam2/automatic_mask_generator.py:294
↓ 1 callersMethod_process_crop
( self, image: np.ndarray, crop_box: List[int], crop_layer_idx: int, o
unigeoseg/model/multimodal_encoder/sam2/automatic_mask_generator.py:251
↓ 1 callersMethod_reset_parameters
(self)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/transformer.py:56
↓ 1 callersMethod_reset_parameters
(self)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:27
↓ 1 callersMethod_reset_parameters
(self)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:85
↓ 1 callersMethod_reset_parameters
(self)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:150
↓ 1 callersMethod_reset_parameters
(self)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/msdeformattn.py:118
↓ 1 callersMethod_reset_parameters
(self)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops/modules/ms_deform_attn.py:66
↓ 1 callersMethod_reset_tracking_results
Reset all tracking inputs and results across the videos.
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor.py:689
↓ 1 callersMethod_run_memory_encoder
Run the memory encoder on `high_res_masks`. This is usually after applying non-overlapping constraints to object scores. Since their
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor_legacy.py:980
↓ 1 callersMethod_run_memory_encoder
Run the memory encoder on `high_res_masks`. This is usually after applying non-overlapping constraints to object scores. Since their
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor.py:805
↓ 1 callersMethod_set_aux_loss
(self, outputs_class, outputs_seg_masks)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/maskformer_transformer_decoder.py:96
↓ 1 callersMethod_set_aux_loss
(self, outputs_class, outputs_seg_masks)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:381
↓ 1 callersMethod_track_step
( self, frame_idx, is_init_cond_frame, current_vision_feats, current_v
unigeoseg/model/multimodal_encoder/sam2/modeling/sam2_base.py:728
↓ 1 callersMethod_training_init
(self, cfg)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/maskformer_train.py:114
↓ 1 callersMethod_use_mask_as_output
Directly turn binary `mask_inputs` into a output mask logits without using SAM. (same input and output shapes as in _forward_sam_head
unigeoseg/model/multimodal_encoder/sam2/modeling/sam2_base.py:415
↓ 1 callersMethod_use_multimask
Whether to use multimask output in the SAM head.
unigeoseg/model/multimodal_encoder/sam2/modeling/sam2_base.py:881
↓ 1 callersFunctionadd_args
(parser, cfg, prefix='')
unigeoseg/model/mask_decoder/Mask2Former_Simplify/configs/config.py:29
↓ 1 callersFunctionadd_args
(parser, cfg, prefix='')
unigeoseg/mask_config/config.py:29
↓ 1 callersMethodadd_new_points_or_box
Add new points to a frame.
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor_legacy.py:173
↓ 1 callersMethodadd_scalar
(self, tag, scalar_value, global_step=None, walltime=None)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/summary.py:34
↓ 1 callersFunctionannotation_name
Returns the "name" of an annotation, optionally including the attributes. :param attributes: The attribute dictionary. :param category_na
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/utils.py:15
↓ 1 callersMethodappend_message
(self, role, message)
unigeoseg/conversation.py:106
↓ 1 callersFunctionapply_delta
(base_model_path, target_model_path, delta_path)
unigeoseg/model/apply_delta.py:12
↓ 1 callersFunctionapply_rotary_enc
( xq: torch.Tensor, xk: torch.Tensor, freqs_cis: torch.Tensor, repeat_freqs_k: bool = False, )
unigeoseg/model/multimodal_encoder/sam2/modeling/position_encoding.py:212
↓ 1 callersFunctionarea_from_rle
(rle: Dict[str, Any])
unigeoseg/model/multimodal_encoder/sam2/utils/amg.py:154
↓ 1 callersMethodbackward
(ctx, grad_output)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops/functions/ms_deform_attn_func.py:43
↓ 1 callersFunctionbbox_correct
(bbox, w, h)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/aug_strategy.py:163
↓ 1 callersMethodbinary_dice_loss
(self, inputs, targets)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/criterion.py:377
↓ 1 callersFunctionboxes_demo
(image, boxes, seq)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/aug_strategy.py:181
↓ 1 callersFunctionbuild_all_layer_point_grids
Generates point grids for all crop layers.
unigeoseg/model/multimodal_encoder/sam2/utils/amg.py:191
↓ 1 callersMethodbuild_backbone
(self, cfg)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/MaskFormerModel.py:86
↓ 1 callersMethodbuild_optimizer
(self)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/maskformer_train.py:72
↓ 1 callersFunctionbuild_point_grid
Generates a 2D grid of points evenly spaced in [0,1]x[0,1].
unigeoseg/model/multimodal_encoder/sam2/utils/amg.py:181
↓ 1 callersFunctionbuild_sam2
( config_file, ckpt_path=None, device="cuda", mode="eval", hydra_overrides_extra=[], a
unigeoseg/model/multimodal_encoder/sam2/build_sam.py:71
↓ 1 callersFunctionbuild_swin_l
(pretrain=None)
unigeoseg/model/multimodal_encoder/swin_trans.py:700
↓ 1 callersFunctioncalculate_uncertainty
We estimate uncerainty as L1 distance between 0.0 and the logit prediction in 'logits' for the foreground class in `classes`. Args:
unigeoseg/model/mask_decoder/mask_criterion/pretrain_criterion.py:106
↓ 1 callersFunctioncalculate_uncertainty
We estimate uncerainty as L1 distance between 0.0 and the logit prediction in 'logits' for the foreground class in `classes`. Args:
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/criterion.py:100
↓ 1 callersFunctioncheck_forward_equal_with_pytorch_double
()
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops/test.py:35
↓ 1 callersFunctioncheck_forward_equal_with_pytorch_float
()
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops/test.py:51
↓ 1 callersFunctioncheck_gradient_numerical
(channels=4, grad_value=True, grad_sampling_loc=True, grad_attn_weight=True)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops/test.py:66
↓ 1 callersMethodclear_all_prompts_in_frame
Remove all input points or mask in a specific frame for a given object.
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor_legacy.py:777
↓ 1 callersMethodclear_all_prompts_in_frame
Remove all input points or mask in a specific frame for a given object.
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor.py:633
↓ 1 callersFunctioncoco_encode_rle
(uncompressed_rle: Dict[str, Any])
unigeoseg/model/multimodal_encoder/sam2/utils/amg.py:296
↓ 1 callersMethodcolorEncode
(self, labelmap, colors, mode='RGB')
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/misc.py:226
↓ 1 callersFunctioncompute_metric
(intersection_meter, union_meter, acc_iou_meter, pr_meters, cur_res, gt, rsvg=False)
unigeoseg/eval_and_test/eval.py:89
↓ 1 callersFunctionconsolidate_ckpt
(src_path, dst_path)
unigeoseg/model/consolidate.py:7
↓ 1 callersMethodcreateIndex
(self)
unigeoseg/eval_and_test/refer.py:80
↓ 1 callersFunctioncreate_annotations
()
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/DataTools.py:16
↓ 1 callersFunctioncreate_summary
(distributed_rank=0, **kwargs)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/summary.py:58
↓ 1 callersMethoddice_loss
(self, predict, targets)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/criterion.py:386
↓ 1 callersFunctiondisable_torch_init
Disable the redundant torch default initialization to accelerate model creation.
unigeoseg/utils.py:93
↓ 1 callersMethodevaluate
(self, eval_loder)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/maskformer_train.py:247
↓ 1 callersMethodevaluate_sample
(self)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/maskformer_train.py:270
↓ 1 callersFunctionevaluation
()
unigeoseg/eval_and_test/eval.py:220
↓ 1 callersFunctionextract_intera_coords
从原始文本中提取intera任务的坐标(框/点) 返回:coord_tensor(框:[1,4];点:[n,2],n=1-3;无坐标:None)
unigeoseg/eval_and_test/eval_dataset/RS_val_dataset.py:43
↓ 1 callersMethodflush
(self)
unigeoseg/utils.py:87
↓ 1 callersMethodforward
(self, x)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/backbone/swin.py:36
↓ 1 callersMethodforward_batch
(self, img_list)
unigeoseg/model/multimodal_encoder/sam2/utils/transforms.py:41
↓ 1 callersMethodforward_concat
(self, x, mask_features, mask=None, seg_query=None, SEG_embedding=None, class_name_emb
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:507
↓ 1 callersMethodforward_ffn
(self, src)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/msdeformattn.py:51
↓ 1 callersMethodforward_image
Identical to the corresponding method in the parent (SAM2VideoPredictor), but cloning the backbone features and pos encoding to enabl
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor.py:1013
↓ 1 callersMethodforward_post
( self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: Optional
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/transformer.py:182
↓ 1 callersMethodforward_post
( self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: O
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/transformer.py:261
↓ 1 callersMethodforward_post
(self, tgt, tgt_mask: Optional[Tensor] = None, tgt_key_padding_mas
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:35
↓ 1 callersMethodforward_post
(self, tgt, memory, memory_mask: Optional[Tensor] = None, memory_k
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:93
↓ 1 callersMethodforward_post
(self, tgt)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:158
↓ 1 callersMethodforward_pre
( self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: Optional
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/transformer.py:200
↓ 1 callersMethodforward_pre
( self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: O
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/transformer.py:292
↓ 1 callersMethodforward_pre
(self, tgt, tgt_mask: Optional[Tensor] = None, tgt_key_padding_mask:
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:47
↓ 1 callersMethodforward_pre
(self, tgt, memory, memory_mask: Optional[Tensor] = None, memory_key
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:107
↓ 1 callersMethodforward_pre
(self, tgt)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:164
↓ 1 callersMethodforward_woconcat
(self, x, mask_features, mask=None, seg_query=None, SEG_embedding=None, class_name_e
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:604
↓ 1 callersFunctiongenerate_crop_boxes
Generates a list of crop boxes of different sizes. Each layer has (2**i)**2 boxes for the ith layer.
unigeoseg/model/multimodal_encoder/sam2/utils/amg.py:202
↓ 1 callersMethodgetRefBox
(self, ref_id)
unigeoseg/eval_and_test/refer.py:231
↓ 1 callersFunctionget_1d_sine_pe
Get 1D sine positional embedding as in the original Transformer paper.
unigeoseg/model/multimodal_encoder/sam2/modeling/sam2_utils.py:64
↓ 1 callersFunctionget_activation_fn
Return an activation function given a string
unigeoseg/model/multimodal_encoder/sam2/modeling/sam2_utils.py:77
← previousnext →201–300 of 763, ranked by callers