MCPcopy Create free account

hub / github.com/IRVLUTD/UnseenObjectsWithMeanShift / functions

Functions680 in github.com/IRVLUTD/UnseenObjectsWithMeanShift

↓ 1 callersMethod__find_smallest
Find the smallest uncovered value in the matrix.
lib/utils/munkres.py:526
↓ 1 callersMethod__find_star_in_col
Find the first starred element in the specified row. Returns the row index, or -1 if no starred element was found.
lib/utils/munkres.py:575
↓ 1 callersMethod__find_star_in_row
Find the first starred element in the specified row. Returns the column index, or -1 if no starred element was found.
lib/utils/munkres.py:562
↓ 1 callersMethod__init__
(self, input_dim, hidden_dim, output_dim, num_layers)
MSMFormer/meanshiftformer/modeling/transformer_decoder/maskformer_transformer_decoder.py:177
↓ 1 callersMethod__init__
NOTE: this interface is experimental. Args: input_shape: shapes (channels and stride) of the input features n
MSMFormer/meanshiftformer/modeling/meta_arch/meanshift_former_head.py:48
↓ 1 callersMethod__init__
NOTE: this interface is experimental. Args: input_shape: shapes (channels and stride) of the input features n
MSMFormer/meanshiftformer/modeling/meta_arch/per_pixel_baseline.py:48
↓ 1 callersFunction_convert_category_id
(segment_info, meta)
MSMFormer/meanshiftformer/data/datasets/register_mapillary_vistas_panoptic.py:349
↓ 1 callersFunction_convert_category_id
(segment_info, meta)
MSMFormer/meanshiftformer/data/datasets/register_ade20k_panoptic.py:228
↓ 1 callersFunction_convert_category_id
(segment_info, meta)
MSMFormer/meanshiftformer/data/datasets/register_coco_panoptic_annos_semseg.py:86
↓ 1 callersFunction_get_ade20k_full_meta
()
MSMFormer/meanshiftformer/data/datasets/register_ade20k_full.py:926
↓ 1 callersFunction_get_ade_instances_meta
()
MSMFormer/meanshiftformer/data/datasets/register_ade20k_instance.py:28
↓ 1 callersMethod_get_augmented_inputs
(self, input)
MSMFormer/meanshiftformer/test_time_augmentation.py:100
↓ 1 callersFunction_get_coco_stuff_meta
()
MSMFormer/meanshiftformer/data/datasets/register_coco_stuff_10k.py:182
↓ 1 callersMethod_get_default_path
Return the default path where ocid_object is expected to be installed.
lib/datasets/ocid_object.py:123
↓ 1 callersMethod_get_default_path
Return the default path where ocid_object is expected to be installed.
lib/datasets/ocid_dataset.py:136
↓ 1 callersMethod_get_default_path
Return the default path where tabletop_object is expected to be installed.
lib/datasets/tabletop_dataset.py:413
↓ 1 callersMethod_get_default_path
Return the default path where tabletop_object is expected to be installed.
lib/datasets/tabletop_object.py:319
↓ 1 callersMethod_get_default_path
Return the default path where ocid_object is expected to be installed.
lib/datasets/load_OCID_UOAIS.py:149
↓ 1 callersMethod_get_default_path
Return the default path where tabletop_object is expected to be installed.
lib/datasets/pushing_dataset.py:421
↓ 1 callersMethod_get_default_path
Return the default path where uoais_object is expected to be installed.
lib/datasets/uoais_dataset.py:295
↓ 1 callersMethod_get_default_path
Return the default path where osd_object is expected to be installed.
lib/datasets/osd_object.py:116
↓ 1 callersMethod_get_default_path
Return the default path where osd_object is expected to be installed.
lib/datasets/load_OSD_UOAIS.py:218
↓ 1 callersFunction_get_mapillary_vistas_meta
()
MSMFormer/meanshiftformer/data/datasets/register_mapillary_vistas.py:473
↓ 1 callersMethod_get_tgt_permutation_idx
(self, indices)
MSMFormer/meanshiftformer/modeling/criterion.py:198
↓ 1 callersMethod_inference_one_image
Args: input (dict): one dataset dict with "image" field being a CHW tensor Returns: dict: one output dict
MSMFormer/meanshiftformer/test_time_augmentation.py:71
↓ 1 callersMethod_initialize_weights
(self)
lib/networks/SEG.py:77
↓ 1 callersFunction_is_power_of_2
(n)
MSMFormer/meanshiftformer/modeling/pixel_decoder/ops/modules/ms_deform_attn.py:28
↓ 1 callersFunction_max_by_axis
(the_list)
MSMFormer/meanshiftformer/utils/misc.py:16
↓ 1 callersFunction_merge_a_into_b
Merge config dictionary a into config dictionary b, clobbering the options in b whenever they are also specified in a.
lib/fcn/config.py:408
↓ 1 callersMethod_normal_initialization
(self, layer)
lib/networks/resnet_dilated.py:73
↓ 1 callersMethod_normal_initialization
(self, layer)
lib/networks/resnet_dilated.py:112
↓ 1 callersMethod_normal_initialization
(self, layer)
lib/networks/resnet_dilated.py:156
↓ 1 callersMethod_normal_initialization
(self, layer)
lib/networks/resnet_dilated.py:194
↓ 1 callersMethod_normal_initialization
(self, layer)
lib/networks/resnet_dilated.py:233
↓ 1 callersMethod_normal_initialization
(self, layer)
lib/networks/resnet_dilated.py:271
↓ 1 callersMethod_normal_initialization
(self, layer)
lib/networks/resnet_dilated.py:310
↓ 1 callersMethod_normal_initialization
(self, layer)
lib/networks/resnet_dilated.py:347
↓ 1 callersMethod_normal_initialization
(self, layer)
lib/networks/resnet_dilated.py:380
↓ 1 callersMethod_normal_initialization
(self, layer)
lib/networks/resnet_dilated.py:418
↓ 1 callersMethod_normal_initialization
(self, layer)
lib/networks/resnet_dilated.py:455
↓ 1 callersMethod_normal_initialization
(self, layer)
lib/networks/resnet_dilated.py:495
↓ 1 callersFunction_onnx_nested_tensor_from_tensor_list
(tensor_list: List[Tensor])
MSMFormer/meanshiftformer/utils/misc.py:76
↓ 1 callersMethod_reset_parameters
(self)
MSMFormer/meanshiftformer/modeling/transformer_decoder/transformer.py:56
↓ 1 callersMethod_reset_parameters
(self)
MSMFormer/meanshiftformer/modeling/transformer_decoder/meanshiftformer_transformer_decoder.py:42
↓ 1 callersMethod_reset_parameters
(self)
MSMFormer/meanshiftformer/modeling/transformer_decoder/meanshiftformer_transformer_decoder.py:100
↓ 1 callersMethod_reset_parameters
(self)
MSMFormer/meanshiftformer/modeling/transformer_decoder/meanshiftformer_transformer_decoder.py:163
↓ 1 callersMethod_reset_parameters
(self)
MSMFormer/meanshiftformer/modeling/transformer_decoder/meanshiftformer_transformer_decoder.py:223
↓ 1 callersMethod_reset_parameters
(self)
MSMFormer/meanshiftformer/modeling/transformer_decoder/meanshiftformer_transformer_decoder.py:292
↓ 1 callersMethod_reset_parameters
(self)
MSMFormer/meanshiftformer/modeling/transformer_decoder/mask2former_transformer_decoder.py:32
↓ 1 callersMethod_reset_parameters
(self)
MSMFormer/meanshiftformer/modeling/transformer_decoder/mask2former_transformer_decoder.py:90
↓ 1 callersMethod_reset_parameters
(self)
MSMFormer/meanshiftformer/modeling/transformer_decoder/mask2former_transformer_decoder.py:155
↓ 1 callersMethod_reset_parameters
(self)
MSMFormer/meanshiftformer/modeling/pixel_decoder/fpn.py:315
↓ 1 callersMethod_reset_parameters
(self)
MSMFormer/meanshiftformer/modeling/pixel_decoder/msdeformattn.py:43
↓ 1 callersMethod_reset_parameters
(self)
MSMFormer/meanshiftformer/modeling/pixel_decoder/ops/modules/ms_deform_attn.py:66
↓ 1 callersMethod_set_aux_loss
(self, outputs_class, outputs_seg_masks)
MSMFormer/meanshiftformer/modeling/transformer_decoder/maskformer_transformer_decoder.py:161
↓ 1 callersMethod_set_aux_loss
(self, outputs_class, outputs_seg_masks)
MSMFormer/meanshiftformer/modeling/transformer_decoder/meanshiftformer_transformer_decoder.py:685
↓ 1 callersMethod_set_aux_loss
(self, outputs_class, outputs_seg_masks)
MSMFormer/meanshiftformer/modeling/transformer_decoder/meanshiftformer_transformer_decoder.py:1038
↓ 1 callersMethod_set_aux_loss
(self, outputs_class, outputs_seg_masks)
MSMFormer/meanshiftformer/modeling/transformer_decoder/meanshiftformer_transformer_decoder.py:1294
↓ 1 callersMethod_set_aux_loss
(self, outputs_class, outputs_seg_masks)
MSMFormer/meanshiftformer/modeling/transformer_decoder/mask2former_transformer_decoder.py:451
↓ 1 callersFunctionadd_path
(path)
tools/_init_paths.py:10
↓ 1 callersFunctionadd_path
(path)
ros/_init_paths.py:10
↓ 1 callersFunctionadjust_crop
(x0, y0, crop_size, instances, eps=1e-3)
MSMFormer/meanshiftformer/data/dataset_mappers/unseen_instance_dataset_mapper.py:73
↓ 1 callersFunctionball_kernel
Computes pairwise ball kernel (without normalizing constant) (note this is kernel as defined in non-parametric statistics, not a kernel as in
lib/utils/mean_shift.py:11
↓ 1 callersFunctionball_kernel
Computes pairwise ball kernel (without normalizing constant) (note this is kernel as defined in non-parametric statistics, not a kernel as in
MSMFormer/meanshiftformer/modeling/transformer_decoder/mean_shift.py:11
↓ 1 callersMethodbias_parameters
(self)
lib/networks/SEG.py:125
↓ 1 callersFunctionboundary_overlap
Compute true positives of overlapped masks, using dilated boundaries Arguments: predicted_mask (ndarray): binary segmentation image
lib/utils/evaluation.py:75
↓ 1 callersMethodbuild_evaluator
Create evaluator(s) for a given dataset. This uses the special metadata "evaluator_type" associated with each builtin dataset
MSMFormer/tabletop_train_net_pretrained.py:73
↓ 1 callersFunctionbuild_matrix_of_indices
Builds a [height, width, 2] numpy array containing coordinates. @return: 3d array B s.t. B[..., 0] contains y-coordinates, B[..., 1] contain
lib/utils/mask.py:41
↓ 1 callersMethodbuild_network
Build encoder network Uses a U-Net-like architecture
lib/networks/unets.py:149
↓ 1 callersMethodbuild_network
Build a decoder network Uses a U-Net-like architecture
lib/networks/unets.py:187
↓ 1 callersFunctionbuild_transform_gen
Create a list of default :class:`Augmentation` from config. Now it includes resizing and flipping. Returns: list[Augmentation]
MSMFormer/meanshiftformer/data/dataset_mappers/coco_panoptic_new_baseline_dataset_mapper.py:18
↓ 1 callersFunctionbuild_transform_gen
Create a list of default :class:`Augmentation` from config. Now it includes resizing and flipping. Returns: list[Augmentation]
MSMFormer/meanshiftformer/data/dataset_mappers/coco_instance_new_baseline_dataset_mapper.py:37
↓ 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:
MSMFormer/meanshiftformer/modeling/criterion.py:73
↓ 1 callersFunctioncheck_forward_equal_with_pytorch_double
()
MSMFormer/meanshiftformer/modeling/pixel_decoder/ops/test.py:35
↓ 1 callersFunctioncheck_forward_equal_with_pytorch_float
()
MSMFormer/meanshiftformer/modeling/pixel_decoder/ops/test.py:51
↓ 1 callersFunctioncheck_gradient_numerical
(channels=4, grad_value=True, grad_sampling_loc=True, grad_attn_weight=True)
MSMFormer/meanshiftformer/modeling/pixel_decoder/ops/test.py:66
↓ 1 callersFunctioncompute_cluster_mean
Computes the spherical mean of a set of unit vectors. This is a PyTorch implementation The definition of spherical mean is minimizes cosine s
lib/networks/embedding.py:23
↓ 1 callersFunctioncompute_cluster_mean
Computes the spherical mean of a set of unit vectors. This is a PyTorch implementation The definition of spherical mean is minimizes cosine s
MSMFormer/meanshiftformer/embedding.py:23
↓ 1 callersFunctioncompute_xyz
(depth_img, fx, fy, px, py, height, width)
tools/test_image_with_ms_transformer.py:106
↓ 1 callersFunctioncompute_xyz
(depth_img, fx, fy, px, py, height, width)
tools/get_network.py:115
↓ 1 callersFunctioncompute_xyz
(depth_img, fx, fy, px, py, height, width)
tools/test_npy.py:96
↓ 1 callersFunctioncompute_xyz
(depth_img, fx, fy, px, py, height, width)
tools/test_images.py:95
↓ 1 callersFunctioncompute_xyz
(depth_img, fx, fy, px, py, height, width)
lib/fcn/get_backbone.py:96
↓ 1 callersFunctioncompute_xyz
(depth_img, fx, fy, px, py, height, width)
lib/fcn/get_network_crop.py:114
↓ 1 callersFunctioncompute_xyz
Compute ordered point cloud from depth image and camera parameters. If focal lengths fx,fy are stored in the camera_params dictionary, use t
lib/datasets/tabletop_dataset.py:60
↓ 1 callersFunctioncompute_xyz
Compute ordered point cloud from depth image and camera parameters. If focal lengths fx,fy are stored in the camera_params dictionary, use t
lib/datasets/tabletop_object.py:57
↓ 1 callersFunctioncompute_xyz
(depth_img, fx, fy, px, py, height, width)
ros/test_images_segmentation_transformer.py:41
↓ 1 callersFunctioncompute_xyz
(depth_img, fx, fy, px, py, height, width)
ros/test_images_segmentation.py:39
↓ 1 callersFunctionconnected_components
For the connected components, we simply perform a nearest neighbor search in order: for each point, find the points that are up t
lib/utils/mean_shift.py:41
↓ 1 callersFunctionconnected_components
For the connected components, we simply perform a nearest neighbor search in order: for each point, find the points that are up t
MSMFormer/meanshiftformer/modeling/transformer_decoder/mean_shift.py:41
↓ 1 callersFunctionconvert
(input, output)
MSMFormer/datasets/prepare_ade20k_sem_seg.py:12
↓ 1 callersFunctionconvert_coco_poly_to_mask
(segmentations, height, width)
MSMFormer/meanshiftformer/data/dataset_mappers/coco_instance_new_baseline_dataset_mapper.py:20
↓ 1 callersMethoddecompose
(self)
MSMFormer/meanshiftformer/utils/misc.py:41
↓ 1 callersMethodforward
(self, x)
MSMFormer/meanshiftformer/modeling/backbone/swin.py:35
↓ 1 callersMethodforward_features
(self, features)
MSMFormer/meanshiftformer/modeling/pixel_decoder/fpn.py:261
↓ 1 callersMethodforward_features
(self, features)
MSMFormer/meanshiftformer/modeling/pixel_decoder/fpn.py:413
↓ 1 callersMethodforward_ffn
(self, src)
MSMFormer/meanshiftformer/modeling/pixel_decoder/msdeformattn.py:116
↓ 1 callersMethodforward_post
( self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: Optional
MSMFormer/meanshiftformer/modeling/transformer_decoder/transformer.py:182
↓ 1 callersMethodforward_post
( self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: O
MSMFormer/meanshiftformer/modeling/transformer_decoder/transformer.py:261
↓ 1 callersMethodforward_post
(self, tgt, tgt_mask: Optional[Tensor] = None, tgt_key_padding_mask:
MSMFormer/meanshiftformer/modeling/transformer_decoder/meanshiftformer_transformer_decoder.py:50
← previousnext →101–200 of 680, ranked by callers