MCPcopy Create free account

hub / github.com/MiliLab/UniGeoSeg / functions

Functions763 in github.com/MiliLab/UniGeoSeg

Functioninner
(tensors, pattern, **kwargs)
unigeoseg/model/language_model/einops_exts.py:18
Methodinstance_inference
(self, mask_cls, mask_pred)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/maskformer_train.py:314
Methodlist_anns
List all the annotations of a sample. :param sample_token: Sample token. :param verbose: Whether to print to stdout.
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/nuimages.py:328
Methodlist_attributes
List all attributes and the number of annotations with each attribute. :param sort_by: Sorting criteria, e.g. "name", "freq".
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/nuimages.py:208
Methodlist_cameras
List all cameras and the number of samples for each.
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/nuimages.py:242
Methodlist_categories
List all categories and the number of object_anns and surface_anns for them. :param sample_tokens: A list of sample tokens for which
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/nuimages.py:271
Methodlist_logs
List all logs and the number of samples per log.
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/nuimages.py:360
Methodlist_sample_content
List the sample_datas for a given sample. :param sample_token: Sample token.
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/nuimages.py:384
Methodlist_sample_data_histogram
Show a histogram of the number of sample_datas per sample.
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/nuimages.py:404
MethodloadAnns
(self, ann_ids=[])
unigeoseg/eval_and_test/refer.py:213
MethodloadCats
(self, cat_ids=[])
unigeoseg/eval_and_test/refer.py:225
MethodloadImgs
(self, image_ids=[])
unigeoseg/eval_and_test/refer.py:219
Functionload_image_from_base64
(image)
unigeoseg/mm_utils.py:10
Methodloss_SEG_labels_concat
(self, outputs, targets, indices, num_masks)
unigeoseg/model/mask_decoder/mask_criterion/pretrain_criterion.py:174
Methodloss_class_name_labels
(self, outputs, targets, indices, num_masks)
unigeoseg/model/mask_decoder/mask_criterion/pretrain_criterion.py:182
Methodloss_labels
(self, outputs, targets, indices)
unigeoseg/model/mask_decoder/mask_criterion/pretrain_criterion.py:137
Methodloss_labels
Classification loss (NLL) targets dicts must contain the key "labels" containing a tensor of dim [nb_target_boxes]
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/criterion.py:150
Methodloss_masks
(self, outputs, targets, indices, num_masks)
unigeoseg/model/mask_decoder/mask_criterion/pretrain_criterion.py:205
Methodloss_masks
Compute the losses related to the masks: the focal loss and the dice loss. targets dicts must contain the key "masks" containing a tensor of d
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/criterion.py:171
Functionmask_demo
(image, mask, seq)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/aug_strategy.py:172
Methodmaybe_add_full_model_gradient_clipping
(optim)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/maskformer_train.py:73
Methodmemory_efficient_forward
More memory-friendly matching
unigeoseg/model/mask_decoder/mask_criterion/pretrain_criterion.py:341
Functionms_deform_attn_backward
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops/src/ms_deform_attn.h:46
Functionms_deform_attn_cpu_backward
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops/src/cpu/ms_deform_attn_cpu.cpp:34
Functionms_deform_attn_cpu_forward
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops/src/cpu/ms_deform_attn_cpu.cpp:22
Functionms_deform_attn_forward
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops/src/ms_deform_attn.h:25
Functionoptimizer_wgc_step
(self, closure=None)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/solver.py:55
Methodpostprocess_small_regions
Removes small disconnected regions and holes in masks, then reruns box NMS to remove any new duplicates. Edits mask_data in
unigeoseg/model/multimodal_encoder/sam2/automatic_mask_generator.py:387
Methodpredict
Predict masks for the given input prompts, using the currently set image. Arguments: point_coords (np.ndarray or None): A
unigeoseg/model/multimodal_encoder/sam2/sam2_image_predictor.py:237
Methodpredict_batch
This function is very similar to predict(...), however it is used for batched mode, when the model is expected to generate predictions on multiple ima
unigeoseg/model/multimodal_encoder/sam2/sam2_image_predictor.py:175
Methodprepare_inputs_labels_for_multimodal
( self, input_ids, attention_mask, past_key_values, labels, images )
unigeoseg/model/llava_arch.py:63
Methodprepare_targets
(self, targets, images)
unigeoseg/model/language_model/llava_phi.py:439
Functionpretty_print_semaphore
(semaphore)
unigeoseg/utils.py:123
Functionprocess_images
(images, image_processor, model_cfg)
unigeoseg/mm_utils.py:14
Methodpropagate_in_video
Propagate the input points across frames to track in the entire video.
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor_legacy.py:663
Functionreduce_dict
Args: input_dict (dict): all the values will be reduced average (bool): whether to do average or sum Reduce the values in the
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/misc.py:37
Functionrefer_ce_loss
( inputs: torch.Tensor, targets: torch.Tensor, weight: torch.Tensor)
unigeoseg/model/mask_decoder/mask_criterion/refcoco.py:5
Methodremove_object
Remove an object id from the tracking state. If strict is True, we check whether the object id actually exists and raise an error if
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor_legacy.py:1042
Methodremove_object
Remove an object id from the tracking state. If strict is True, we check whether the object id actually exists and raise an error if
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor.py:867
Methodrender_image
(self, img, mask_img, output_path=None)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/Segmentation.py:163
Methodrender_image
Renders an image (sample_data), optionally with annotations overlaid. :param sd_token: The token of the sample_data to be rendered.
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/nuimages.py:629
Methodrender_trajectory
Render a plot of the trajectory for the clip surrounding the annotated keyframe. A red cross indicates the starting point, a green do
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/nuimages.py:750
Functionsample_box_points
Sample a noised version of the top left and bottom right corners of a given `bbox` Inputs: - masks: [B, 1, H,W] boxes, dtype=torch.Tenso
unigeoseg/model/multimodal_encoder/sam2/modeling/sam2_utils.py:156
Methodsegm_transform
(self, segm: np.ndarray)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/dataset.py:86
Functionsegmentation_test
()
unigeoseg/model/mask_decoder/Mask2Former_Simplify/main.py:124
Methodset_image_batch
Calculates the image embeddings for the provided image batch, allowing masks to be predicted with the 'predict_batch' method.
unigeoseg/model/multimodal_encoder/sam2/sam2_image_predictor.py:132
MethodshowMask
(self, ref_id)
unigeoseg/eval_and_test/refer.py:299
Functionsigmoid_ce_loss
Args: inputs: A float tensor of arbitrary shape. The predictions for each example. targets: A float tensor with t
unigeoseg/model/mask_decoder/mask_criterion/pretrain_criterion.py:54
Functionsigmoid_ce_loss
Args: inputs: A float tensor of arbitrary shape. The predictions for each example. targets: A float tensor with t
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/criterion.py:51
Functionsigmoid_focal_loss
Loss used in RetinaNet for dense detection: https://arxiv.org/abs/1708.02002. Args: inputs: A float tensor of arbitrary shape.
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/criterion.py:73
Methodsize_divisibility
(self)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/backbone/swin.py:767
Functionsub_with_anonymous_dims
(t)
unigeoseg/model/language_model/einops_exts.py:42
Methodsummary
(self)
unigeoseg/eval_and_test/eval.py:74
Methodtable_root
Returns the folder where the tables are stored for the relevant version.
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/nuimages.py:120
Methodtext
(self)
unigeoseg/mask_config/config.py:127
Methodto_gradio_chatbot
(self)
unigeoseg/conversation.py:158
Functiontokenizer_image_token
(prompt, tokenizer, image_token_index=IMAGE_TOKEN_INDEX, return_tensors=None)
unigeoseg/mm_utils.py:18
Methodtrain
Convert the model into training mode while keep layers freezed.
unigeoseg/model/multimodal_encoder/swin_trans.py:635
Methodtrain
Convert the model into training mode while keep layers freezed.
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/backbone/swin.py:681
Functiontrain_ade200k
()
unigeoseg/model/mask_decoder/Mask2Former_Simplify/main.py:61
Functionuser_scattered_collate
(batch)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/main.py:29
Functionviolates_moderation
Check whether the text violates OpenAI moderation API.
unigeoseg/utils.py:102
Methodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/transformer_decoder/mask2former_transformer_decoder.py:155
← previous701–763 of 763, ranked by callers