MCPcopy Create free account

hub / github.com/IRVLUTD/UnseenObjectsWithMeanShift / functions

Functions680 in github.com/IRVLUTD/UnseenObjectsWithMeanShift

Method__init__
NOTE: this interface is experimental. Args: in_channels: channels of the input features mask_classification:
MSMFormer/meanshiftformer/modeling/transformer_decoder/meanshiftformer_transformer_decoder.py:1079
Method__init__
(self, num_pos_feats=64, temperature=10000, normalize=False, scale=None)
MSMFormer/meanshiftformer/modeling/transformer_decoder/position_encoding.py:18
Method__init__
(self, d_model, nhead, dropout=0.0, activation="relu", normalize_before=False)
MSMFormer/meanshiftformer/modeling/transformer_decoder/mask2former_transformer_decoder.py:19
Method__init__
(self, d_model, nhead, dropout=0.0, activation="relu", normalize_before=False)
MSMFormer/meanshiftformer/modeling/transformer_decoder/mask2former_transformer_decoder.py:77
Method__init__
(self, d_model, dim_feedforward=2048, dropout=0.0, activation="relu", normalize_before=False)
MSMFormer/meanshiftformer/modeling/transformer_decoder/mask2former_transformer_decoder.py:140
Method__init__
NOTE: this interface is experimental. Args: in_channels: channels of the input features mask_classification:
MSMFormer/meanshiftformer/modeling/transformer_decoder/mask2former_transformer_decoder.py:236
Method__init__
( self, dim, window_size, num_heads, qkv_bias=True, qk_scale=N
MSMFormer/meanshiftformer/modeling/backbone/swin.py:87
Method__init__
( self, dim, num_heads, window_size=7, shift_size=0, mlp_ratio
MSMFormer/meanshiftformer/modeling/backbone/swin.py:191
Method__init__
(self, dim, norm_layer=nn.LayerNorm)
MSMFormer/meanshiftformer/modeling/backbone/swin.py:305
Method__init__
( self, dim, depth, num_heads, window_size=7, mlp_ratio=4.0,
MSMFormer/meanshiftformer/modeling/backbone/swin.py:358
Method__init__
(self, patch_size=4, in_chans=3, embed_dim=96, norm_layer=None)
MSMFormer/meanshiftformer/modeling/backbone/swin.py:465
Method__init__
( self, pretrain_img_size=224, patch_size=4, in_chans=3, embed_dim=96,
MSMFormer/meanshiftformer/modeling/backbone/swin.py:526
Method__init__
(self, cfg, input_shape)
MSMFormer/meanshiftformer/modeling/backbone/swin.py:688
Method__init__
NOTE: this interface is experimental. Args: input_shape: shapes (channels and stride) of the input features n
MSMFormer/meanshiftformer/modeling/meta_arch/mask_former_head.py:48
Method__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:176
Method__init__
NOTE: this interface is experimental. Args: input_shape: shapes (channels and stride) of the input features t
MSMFormer/meanshiftformer/modeling/meta_arch/per_pixel_baseline.py:153
Method__init__
NOTE: this interface is experimental. Args: input_shape: shapes (channels and stride) of the input features c
MSMFormer/meanshiftformer/modeling/pixel_decoder/fpn.py:164
Method__init__
( self, d_model=512, nhead=8, num_encoder_layers=6, dim_feedforward=20
MSMFormer/meanshiftformer/modeling/pixel_decoder/fpn.py:292
Method__init__
NOTE: this interface is experimental. Args: input_shape: shapes (channels and stride) of the input features t
MSMFormer/meanshiftformer/modeling/pixel_decoder/fpn.py:336
Method__init__
(self, d_model=256, nhead=8, num_encoder_layers=6, dim_feedforward=1024, dropout=0.1,
MSMFormer/meanshiftformer/modeling/pixel_decoder/msdeformattn.py:24
Method__init__
(self, d_model=256, d_ffn=1024, dropout=0.1, activation="relu",
MSMFormer/meanshiftformer/modeling/pixel_decoder/msdeformattn.py:93
Method__init__
(self, encoder_layer, num_layers)
MSMFormer/meanshiftformer/modeling/pixel_decoder/msdeformattn.py:135
Method__init__
Multi-Scale Deformable Attention Module :param d_model hidden dimension :param n_levels number of feature levels
MSMFormer/meanshiftformer/modeling/pixel_decoder/ops/modules/ms_deform_attn.py:35
Method__init__
(self, tensors, mask: Optional[Tensor])
MSMFormer/meanshiftformer/utils/misc.py:26
Method__init__
NOTE: this interface is experimental. Args: is_train: for training or inference augmentations: a list of augm
MSMFormer/meanshiftformer/data/dataset_mappers/mask_former_semantic_dataset_mapper.py:33
Method__init__
NOTE: this interface is experimental. Args: is_train: for training or inference augmentations: a list of augm
MSMFormer/meanshiftformer/data/dataset_mappers/mask_former_panoptic_dataset_mapper.py:33
Method__init__
NOTE: this interface is experimental. Args: is_train: for training or inference augmentations: a list of augm
MSMFormer/meanshiftformer/data/dataset_mappers/coco_panoptic_new_baseline_dataset_mapper.py:67
Method__init__
Args: crop_instance (bool): if False, extend cropping boxes to avoid cropping instances
MSMFormer/meanshiftformer/data/dataset_mappers/unseen_instance_dataset_mapper.py:108
Method__init__
Args: h, w (int): original image size new_h, new_w (int): new image size interp: PIL interpolation method
MSMFormer/meanshiftformer/data/dataset_mappers/unseen_instance_dataset_mapper.py:131
Method__init__
NOTE: this interface is experimental. Args: is_train: for training or inference augmentations: a list of augm
MSMFormer/meanshiftformer/data/dataset_mappers/unseen_instance_dataset_mapper.py:235
Method__init__
NOTE: this interface is experimental. Args: is_train: for training or inference augmentations: a list of augm
MSMFormer/meanshiftformer/data/dataset_mappers/coco_instance_new_baseline_dataset_mapper.py:86
Method__init__
NOTE: this interface is experimental. Args: is_train: for training or inference augmentations: a list of augm
MSMFormer/meanshiftformer/data/dataset_mappers/mask_former_instance_dataset_mapper.py:33
Method__len__
(self)
lib/datasets/ocid_object.py:119
Method__len__
(self)
lib/datasets/ocid_dataset.py:132
Method__len__
(self)
lib/datasets/tabletop_dataset.py:409
Method__len__
(self)
lib/datasets/tabletop_object.py:315
Method__len__
(self)
lib/datasets/load_OCID_UOAIS.py:145
Method__len__
(self)
lib/datasets/pushing_dataset.py:417
Method__len__
(self)
lib/datasets/mixture_dataset.py:47
Method__len__
(self)
lib/datasets/uoais_dataset.py:301
Method__len__
(self)
lib/datasets/osd_object.py:112
Method__len__
(self)
lib/datasets/load_OSD_UOAIS.py:214
Method__repr__
(self)
lib/fcn/train.py:33
Method__repr__
(self)
lib/fcn/test_dataset.py:40
Method__repr__
(self, _repr_indent=4)
MSMFormer/meanshiftformer/modeling/matcher.py:181
Method__repr__
(self)
MSMFormer/meanshiftformer/modeling/criterion.py:249
Method__repr__
(self, _repr_indent=4)
MSMFormer/meanshiftformer/modeling/transformer_decoder/position_encoding.py:54
Method__repr__
(self)
MSMFormer/meanshiftformer/utils/misc.py:44
Method__step1
For each row of the matrix, find the smallest element and subtract it from every element in its row. Go to Step 2.
lib/utils/munkres.py:385
Method__step2
Find a zero (Z) in the resulting matrix. If there is no starred zero in its row or column, star Z. Repeat for each element in the
lib/utils/munkres.py:401
Method__step3
Cover each column containing a starred zero. If K columns are covered, the starred zeros describe a complete set of unique as
lib/utils/munkres.py:420
Method__step4
Find a noncovered zero and prime it. If there is no starred zero in the row containing this primed zero, Go to Step 5. Otherwise,
lib/utils/munkres.py:441
Method__step5
Construct a series of alternating primed and starred zeros as follows. Let Z0 represent the uncovered primed zero found in Step 4.
lib/utils/munkres.py:474
Method__step6
Add the value found in Step 4 to every element of each covered row, and subtract it from every element of each uncovered column.
lib/utils/munkres.py:510
Method_build_background_images
(self)
lib/datasets/imdb.py:93
Method_build_uniform_poses
(self)
lib/datasets/imdb.py:75
Method_eval_predictions
Evaluate predictions. Fill self._results with the metrics of the tasks.
MSMFormer/meanshiftformer/evaluation/instance_evaluation.py:43
Method_init_weights
(m)
MSMFormer/meanshiftformer/modeling/backbone/swin.py:642
Method_load_from_state_dict
( self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs )
MSMFormer/meanshiftformer/modeling/transformer_decoder/meanshiftformer_transformer_decoder.py:348
Method_load_from_state_dict
( self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs )
MSMFormer/meanshiftformer/modeling/transformer_decoder/meanshiftformer_transformer_decoder.py:702
Method_load_from_state_dict
( self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs )
MSMFormer/meanshiftformer/modeling/transformer_decoder/meanshiftformer_transformer_decoder.py:1055
Method_load_from_state_dict
( self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs )
MSMFormer/meanshiftformer/modeling/transformer_decoder/mask2former_transformer_decoder.py:212
Method_load_from_state_dict
( self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs )
MSMFormer/meanshiftformer/modeling/meta_arch/mask_former_head.py:23
Method_load_from_state_dict
( self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs )
MSMFormer/meanshiftformer/modeling/meta_arch/meanshift_former_head.py:23
Method_load_from_state_dict
( self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs )
MSMFormer/meanshiftformer/modeling/meta_arch/meanshift_former_head.py:150
Method_load_from_state_dict
( self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs )
MSMFormer/meanshiftformer/modeling/meta_arch/per_pixel_baseline.py:22
Method_load_from_state_dict
( self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs )
MSMFormer/meanshiftformer/modeling/meta_arch/per_pixel_baseline.py:128
Method_maybe_read_image
(dataset_dict)
MSMFormer/meanshiftformer/test_time_augmentation.py:54
Function_process_panoptic_to_semantic
(input_panoptic, output_semantic, segments, id_map)
MSMFormer/datasets/prepare_coco_semantic_annos_from_panoptic_annos.py:18
Function_vis_features
(features, labels, rgb, intial_labels, selected_pixels=None)
lib/fcn/test_common.py:39
Functionadd_gaussian_noise_cuda
(image, level = 0.1)
lib/utils/blob.py:147
Functionadd_maskformer2_config
Add config for MASK_FORMER.
MSMFormer/meanshiftformer/config.py:137
Functionadd_noise_cuda
(image, level = 0.1)
lib/utils/blob.py:157
Functionadd_noise_depth
(image, level = 0.1)
lib/utils/blob.py:132
Functionadd_noise_depth_cuda
(image, level = 0.1)
lib/utils/blob.py:141
Functionadd_noise_to_depth
Distort depth image with multiplicative gamma noise. This is adapted from the DexNet 2.0 codebase. Their code: https://github.com/Ber
lib/utils/augmentation.py:58
Functionadd_noise_to_xyz
Add (approximate) Gaussian Process noise to ordered point cloud. This is adapted from the DexNet 2.0 codebase. @param xyz_img: a [H
lib/utils/augmentation.py:73
Functionallocentric2egocentric
(qt, T)
lib/utils/se3.py:40
Methodapply_coords
(self, coords)
MSMFormer/meanshiftformer/data/dataset_mappers/unseen_instance_dataset_mapper.py:187
Functionarray_to_tensor
Converts a numpy.ndarray (N x H x W x C) to a torch.FloatTensor of shape (N x C x H x W) OR converts a nump.ndarray (H x W x C) to a
lib/utils/augmentation.py:19
Functionarray_to_tensor
Converts a numpy.ndarray (N x H x W x C) to a torch.FloatTensor of shape (N x C x H x W) OR converts a nump.ndarray (H x W x C) to a
lib/datasets/load_OSD_UOAIS.py:61
Methodbackproject
(self, depth_cv, intrinsic_matrix, factor)
lib/datasets/imdb.py:47
Functionbatch_dice_loss
Compute the DICE loss, similar to generalized IOU for masks Args: inputs: A float tensor of arbitrary shape. The pred
MSMFormer/meanshiftformer/modeling/matcher.py:15
Functionbatch_sigmoid_ce_loss
Args: inputs: A float tensor of arbitrary shape. The predictions for each example. targets: A float tensor with t
MSMFormer/meanshiftformer/modeling/matcher.py:38
Methodbuild_lr_scheduler
It now calls :func:`detectron2.solver.build_lr_scheduler`. Overwrite it if you'd like a different scheduler.
MSMFormer/tabletop_train_net_pretrained.py:105
Methodbuild_optimizer
(cls, cfg, model)
MSMFormer/tabletop_train_net_pretrained.py:113
Methodbuild_train_loader
(cls, cfg)
MSMFormer/tabletop_train_net_pretrained.py:92
Methodcache_path
(self)
lib/datasets/imdb.py:39
Methodcallback
(self, rgb, depth)
ros/collect_images_realsense.py:65
Methodcallback_rgbd
(self, rgb, depth)
ros/test_images_segmentation_transformer.py:119
Methodcallback_rgbd
(self, rgb, depth)
ros/test_images_segmentation.py:116
Methodclass_colors
(self)
lib/datasets/imdb.py:35
Methodclasses
(self)
lib/datasets/imdb.py:31
Functioncombine_masks
Combine several bit masks [N, H, W] into a mask [H,W], e.g. 8*480*640 tensor becomes a numpy array of 480*640. [[1,0,0], [0,1,0]] = > [1,
MSMFormer/meanshiftformer/pretrained_meanshiftformer_model.py:31
Functioncombine_masks
Combine several bit masks [N, H, W] into a mask [H,W], e.g. 8*480*640 tensor becomes a numpy array of 480*640. [[1,0,0], [0,1,0]] = > [1,
MSMFormer/meanshiftformer/meanshiftformer_model.py:22
Functionconcatenate_spatial_coordinates
Adds x,y coordinates as channels to feature map @param feature_map: a [T x C x H x W] torch tensor
lib/networks/utils.py:38
Methoddevice
(self)
MSMFormer/meanshiftformer/pretrained_meanshiftformer_model.py:241
Functiondice_loss
Compute the DICE loss, similar to generalized IOU for masks Args: inputs: A float tensor of arbitrary shape. The pred
MSMFormer/meanshiftformer/modeling/criterion.py:21
Functiondropout_random_ellipses
Randomly drop a few ellipses in the image for robustness. This is adapted from the DexNet 2.0 codebase. Their code: https://github.co
lib/utils/augmentation.py:92
Functionegocentric2allocentric
(qt, T)
lib/utils/se3.py:32
← previousnext →401–500 of 680, ranked by callers