MCPcopy Create free account

hub / github.com/MiliLab/UniGeoSeg / functions

Functions763 in github.com/MiliLab/UniGeoSeg

↓ 1 callersMethodget_batch_size
(self, batch_records)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/dataset.py:143
↓ 1 callersFunctionget_bounding_box_mask
从二值掩码生成最大连通区域的外接矩形掩码 mask: 输入二值掩码 (H, W),0为背景,1为目标 return: 矩形掩码 (H, W),目标区域为最大连通区域的外接框
unigeoseg/eval_and_test/eval.py:174
↓ 1 callersFunctionget_colormap
Get the defined colormap. :return: A mapping from the class names to the respective RGB values.
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/color_map.py:4
↓ 1 callersMethodget_data_aug_pipe
(self)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/dataset.py:129
↓ 1 callersMethodget_ego_pose_data
Return the ego pose data of the <= 13 sample_datas associated with this sample. The method return translation, rotation, rotation_rat
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/nuimages.py:464
↓ 1 callersFunctionget_extensions
()
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops/setup.py:26
↓ 1 callersFunctionget_font
Check if there is a desired font present in the user's system. If there is, use that font; otherwise, use a default font. :param fonts_va
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/utils.py:48
↓ 1 callersMethodget_images
(self, return_pil=False)
unigeoseg/conversation.py:109
↓ 1 callersMethodget_img_ratio
(self, img_size, target_size)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/dataset.py:95
↓ 1 callersFunctionget_largest_connected_component
从二值掩码中获取最大的连通区域 mask: 输入二值掩码 (H, W),0为背景,1为目标 return: 只保留最大连通区域的掩码
unigeoseg/eval_and_test/eval.py:145
↓ 1 callersFunctionget_mask_config
(config='./unigeoseg/mask_config/maskformer2_swin_base_384_bs16_50ep.yaml')
unigeoseg/mask_config/config.py:157
↓ 1 callersFunctionget_model_name_from_path
(model_path)
unigeoseg/mm_utils.py:40
↓ 1 callersMethodget_num_layers
(self)
unigeoseg/model/multimodal_encoder/hieradet.py:341
↓ 1 callersMethodget_num_layers
(self)
unigeoseg/model/multimodal_encoder/sam2/modeling/backbones/hieradet.py:316
↓ 1 callersMethodget_reference_points
(spatial_shapes, valid_ratios, device)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/msdeformattn.py:76
↓ 1 callersMethodget_trajectory
Get the trajectory of the ego vehicle and optionally rotate and center it. :param sample_token: Sample token. :param rotation
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/nuimages.py:502
↓ 1 callersMethodget_valid_ratio
(self, mask)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/msdeformattn.py:127
↓ 1 callersMethodget_vision_tower
(self)
unigeoseg/model/llava_arch.py:55
↓ 1 callersMethodgetind
This returns the index of the record in a table in constant runtime. :param table_name: Table name. :param token: Token of th
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/nuimages.py:104
↓ 1 callersMethodimage_preprocess
(self, img)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/Segmentation.py:103
↓ 1 callersMethodimg_transform
(self, img)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/Segmentation.py:60
↓ 1 callersMethodimg_transform
(self, img)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/dataset.py:79
↓ 1 callersFunctionimgaug_boxes
:param img: H W C int :param mask: H W int :param seq: aug_pipe :return:img H W C int mask H W int
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/aug_strategy.py:129
↓ 1 callersMethodinit_state
Initialize an inference state.
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor.py:42
↓ 1 callersFunctioninit_t_xy
(end_x: int, end_y: int)
unigeoseg/model/multimodal_encoder/sam2/modeling/position_encoding.py:185
↓ 1 callersMethodinitial_mask_module
(self, pretrained_path=None, model_args=None)
unigeoseg/model/language_model/llava_phi.py:239
↓ 1 callersFunctioninsert_separator
(X, sep)
unigeoseg/mm_utils.py:21
↓ 1 callersFunctionintersectionAndUnionGPU
(output, target, K, ignore_index=255)
unigeoseg/eval_and_test/eval.py:131
↓ 1 callersFunctionis_box_near_crop_edge
Filter masks at the edge of a crop, but not at the edge of the original image.
unigeoseg/model/multimodal_encoder/sam2/utils/amg.py:80
↓ 1 callersMethodloadRefs
(self, ref_ids=[])
unigeoseg/eval_and_test/refer.py:207
↓ 1 callersMethodload_model
(self, pretrain_weights)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/Segmentation.py:49
↓ 1 callersMethodload_model
(self, pretrain_weights)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/maskformer_train.py:105
↓ 1 callersFunctionload_pretrained_model
(model_path, model_base, model_name, model_args, mask_config='./unigeoseg/mask_config/maskformer2_swin_base_38
unigeoseg/model/builder.py:24
↓ 1 callersFunctionload_video_frames_from_jpg_images
Load the video frames from a directory of JPEG files ("<frame_index>.jpg" format). The frames are resized to image_size x image_size and are
unigeoseg/model/multimodal_encoder/sam2/utils/misc.py:213
↓ 1 callersFunctionload_video_frames_from_video_file
Load the video frames from a video file.
unigeoseg/model/multimodal_encoder/sam2/utils/misc.py:280
↓ 1 callersMethodloss_SEG_labels
(self, outputs, targets, indices, num_masks)
unigeoseg/model/mask_decoder/mask_criterion/pretrain_criterion.py:140
↓ 1 callersMethodloss_SEG_labels_focal
(self, outputs, targets, indices, num_masks)
unigeoseg/model/mask_decoder/mask_criterion/pretrain_criterion.py:160
↓ 1 callersMethodloss_masks_refer
(self, outputs, targets)
unigeoseg/model/mask_decoder/mask_criterion/refcoco.py:32
↓ 1 callersFunctionmake_delta
(base_model_path, target_model_path, delta_path, hub_repo_id)
unigeoseg/model/make_delta.py:9
↓ 1 callersMethodmask_decoder_training_init
(self, cfg)
unigeoseg/model/language_model/llava_phi.py:299
↓ 1 callersFunctionmask_to_box
compute bounding box given an input mask Inputs: - masks: [B, 1, H, W] masks, dtype=torch.Tensor Returns: - box_coords: [B, 1,
unigeoseg/model/multimodal_encoder/sam2/utils/misc.py:66
↓ 1 callersFunctionmaybe_add_gradient_clipping
If gradient clipping is enabled through config options, wraps the existing optimizer type to become a new dynamically created class Optimizer
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/solver.py:75
↓ 1 callersMethodmemory_efficient_forward
More memory-friendly matching
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/matcher.py:125
↓ 1 callersMethodmm_conv_prepare_inputs_labels_for_multimodal
( self, input_ids, attention_mask, past_key_values, labels, images )
unigeoseg/model/language_model/llava_phi.py:1072
↓ 1 callersFunctionnested_tensor_from_tensor_list
(tensor_list: List[Tensor])
unigeoseg/model/mask_decoder/mask_criterion/refcoco.py:71
↓ 1 callersMethodoutput_shape
(self)
unigeoseg/model/language_model/llava_phi.py:393
↓ 1 callersMethodoutput_shape
(self)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/backbone/swin.py:760
↓ 1 callersMethodparse_input_list
(self, odgt, max_sample=-1, start_idx=-1, end_idx=-1)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/dataset.py:64
↓ 1 callersMethodpixel_decoder_init
(self, cfg, input_shape)
unigeoseg/model/language_model/llava_phi.py:419
↓ 1 callersMethodpixel_decoder_init
(self, cfg, input_shape)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/MaskFormerModel.py:29
↓ 1 callersMethodpostprocess
(self, pred_mask, transformer_info, target_size)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/Segmentation.py:124
↓ 1 callersMethodpostprocess_masks
Perform PostProcessing on output masks.
unigeoseg/model/multimodal_encoder/sam2/utils/transforms.py:76
↓ 1 callersMethodpredict_masks
Predicts masks. See 'forward' for more details.
unigeoseg/model/multimodal_encoder/sam2/modeling/sam/mask_decoder.py:168
↓ 1 callersMethodpredictor_init
(self, cfg)
unigeoseg/model/language_model/llava_phi.py:358
↓ 1 callersMethodpredictor_init
(self, cfg)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/MaskFormerModel.py:50
↓ 1 callersMethodpropagate_in_video
Propagate the input points across frames to track in the entire video.
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor.py:546
↓ 1 callersMethodpropagate_in_video_preflight
Prepare inference_state and consolidate temporary outputs before tracking.
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor_legacy.py:593
↓ 1 callersMethodpropagate_in_video_preflight
Prepare inference_state and consolidate temporary outputs before tracking.
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor.py:480
↓ 1 callersMethodreduce_mean
(self, tensor, nprocs)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/maskformer_train.py:159
↓ 1 callersMethodrefine_with_m2m
(self, points, point_labels, low_res_masks, points_per_batch)
unigeoseg/model/multimodal_encoder/sam2/automatic_mask_generator.py:437
↓ 1 callersMethodreset
(self)
unigeoseg/eval_and_test/eval.py:35
↓ 1 callersMethodreset_state
Remove all input points or mask in all frames throughout the video.
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor_legacy.py:848
↓ 1 callersMethodreset_state
Remove all input points or mask in all frames throughout the video.
unigeoseg/model/multimodal_encoder/sam2/sam2_video_predictor.py:676
↓ 1 callersFunctionreshape_for_broadcast
(freqs_cis: torch.Tensor, x: torch.Tensor)
unigeoseg/model/multimodal_encoder/sam2/modeling/position_encoding.py:204
↓ 1 callersMethodresize_padding
(self, img, outsize, Interpolation=Image.BILINEAR)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/Segmentation.py:81
↓ 1 callersFunctionsample_one_point_from_error_center
Sample 1 random point (along with its label) from the center of each error region, that is, the point with the largest distance to the bounda
unigeoseg/model/multimodal_encoder/sam2/modeling/sam2_utils.py:252
↓ 1 callersFunctionsample_random_points_from_errors
Sample `num_pt` random points (along with their labels) independently from the error regions. Inputs: - gt_masks: [B, 1, H_im, W_im] mas
unigeoseg/model/multimodal_encoder/sam2/modeling/sam2_utils.py:202
↓ 1 callersFunctionselect_closest_cond_frames
Select up to `max_cond_frame_num` conditioning frames from `cond_frame_outputs` that are temporally closest to the current frame at `frame_id
unigeoseg/model/multimodal_encoder/sam2/modeling/sam2_utils.py:19
↓ 1 callersMethodsemantic_inference
(self, mask_cls, mask_pred)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/Segmentation.py:118
↓ 1 callersMethodsemantic_inference
(self, mask_cls, mask_pred)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/maskformer_train.py:307
↓ 1 callersMethodset_image
Calculates the image embeddings for the provided image, allowing masks to be predicted with the 'predict' method. Arguments:
unigeoseg/model/multimodal_encoder/sam2/sam2_image_predictor.py:86
↓ 1 callersMethodshowRef
(self, ref, seg_box='seg')
unigeoseg/eval_and_test/refer.py:236
↓ 1 callersMethodshow_result
(self, img, pred, save_path=None)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/misc.py:242
↓ 1 callersFunctionsigmoid_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/mask_criterion/pretrain_criterion.py:78
↓ 1 callersMethodtext
(self)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/configs/config.py:127
↓ 1 callersMethodto_numpy
(self)
unigeoseg/model/multimodal_encoder/sam2/utils/amg.py:74
↓ 1 callersMethodtrain_epoch
(self,data_loader, epoch)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/maskformer_train.py:190
↓ 1 callersFunctiontrain_nuimages
()
unigeoseg/model/mask_decoder/Mask2Former_Simplify/main.py:89
↓ 1 callersMethodtransform_boxes
Expects a tensor of shape Bx4. The coordinates can be in absolute image or normalized coordinates, if the coords are in absolute imag
unigeoseg/model/multimodal_encoder/sam2/utils/transforms.py:66
↓ 1 callersFunctionuncrop_masks
( masks: torch.Tensor, crop_box: List[int], orig_h: int, orig_w: int )
unigeoseg/model/multimodal_encoder/sam2/utils/amg.py:257
↓ 1 callersFunctionuncrop_points
(points: torch.Tensor, crop_box: List[int])
unigeoseg/model/multimodal_encoder/sam2/utils/amg.py:248
↓ 1 callersFunctionwindow_reverse
Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image
unigeoseg/model/multimodal_encoder/swin_trans.py:52
↓ 1 callersFunctionwindow_reverse
Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/backbone/swin.py:59
↓ 1 callersMethodwith_pos_embed
(tensor, pos)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/msdeformattn.py:48
↓ 1 callersMethodwrite
(self, buf)
unigeoseg/utils.py:73
FunctionPYBIND11_MODULE
unigeoseg/model/mask_decoder/Mask2Former_Simplify/modeling/pixel_decoder/ops/src/vision.cpp:18
MethodPyramidPoolAgg
(self, image_features)
unigeoseg/model/language_model/llava_phi.py:905
Method__call__
(self, output_ids: torch.LongTensor, scores: torch.FloatTensor, **kwargs)
unigeoseg/mm_utils.py:61
Method__call__
(self, x)
unigeoseg/model/multimodal_encoder/sam2/utils/transforms.py:37
Method__call__
Same input/output format as :meth:`SemanticSegmentor.forward`
unigeoseg/model/mask_decoder/Mask2Former_Simplify/test_time_augmentation.py:49
Method__call__
(self, data_dicts)
unigeoseg/eval_and_test/eval_dataset/RS_val_dataset.py:549
Method__delitem__
(self, key: str)
unigeoseg/model/multimodal_encoder/sam2/utils/amg.py:37
Method__exit__
(self, exc_type, exc_val, exc_tb)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/utils/summary.py:120
Method__getattr__
(self, attr)
unigeoseg/utils.py:70
Method__getattr__
(self, name)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/configs/config.py:16
Method__getattr__
Implement lazy loading for the database tables. Otherwise throw the default error. :param attr_name: The name of the variable to look
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/NuImages/nuimages.py:82
Method__getattr__
(self, name)
unigeoseg/mask_config/config.py:16
Method__getitem__
(self, key: str)
unigeoseg/model/multimodal_encoder/sam2/utils/amg.py:40
Method__getitem__
(self, name)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/configs/config.py:140
Method__getitem__
(self, index)
unigeoseg/model/mask_decoder/Mask2Former_Simplify/dataset/dataset.py:170
← previousnext →301–400 of 763, ranked by callers