MCPcopy Create free account

hub / github.com/NVlabs/prismer / functions

Functions669 in github.com/NVlabs/prismer

↓ 1 callersFunction_convert_category_id
(segment_info, meta)
experts/segmentation/mask2former/data/datasets/register_mapillary_vistas_panoptic.py:349
↓ 1 callersFunction_convert_category_id
(segment_info, meta)
experts/segmentation/mask2former/data/datasets/register_ade20k_panoptic.py:228
↓ 1 callersFunction_convert_category_id
(segment_info, meta)
experts/segmentation/mask2former/data/datasets/register_coco_panoptic_annos_semseg.py:86
↓ 1 callersMethod_frame_from_video
(self, video)
experts/obj_detection/unidet/predictor.py:73
↓ 1 callersFunction_get_ade20k_full_meta
()
experts/segmentation/mask2former/data/datasets/register_ade20k_full.py:926
↓ 1 callersFunction_get_ade_instances_meta
()
experts/segmentation/mask2former/data/datasets/register_ade20k_instance.py:28
↓ 1 callersMethod_get_augmented_inputs
(self, input)
experts/segmentation/mask2former/test_time_augmentation.py:100
↓ 1 callersFunction_get_builtin_metadata
()
experts/obj_detection/unidet/data/datasets/scannet.py:26
↓ 1 callersFunction_get_builtin_metadata
()
experts/obj_detection/unidet/data/datasets/kitti.py:15
↓ 1 callersFunction_get_builtin_metadata
()
experts/obj_detection/unidet/data/datasets/crowdhuman.py:9
↓ 1 callersFunction_get_builtin_metadata
()
experts/obj_detection/unidet/data/datasets/cityscapes_cocoformat.py:10
↓ 1 callersFunction_get_builtin_metadata
()
experts/obj_detection/unidet/data/datasets/voc_cocoformat.py:27
↓ 1 callersFunction_get_builtin_metadata
()
experts/obj_detection/unidet/data/datasets/wilddash.py:20
↓ 1 callersFunction_get_builtin_metadata
()
experts/obj_detection/unidet/data/datasets/objects365.py:372
↓ 1 callersFunction_get_builtin_metadata
()
experts/obj_detection/unidet/data/datasets/viper.py:18
↓ 1 callersFunction_get_builtin_metadata
()
experts/obj_detection/unidet/data/datasets/mapillary.py:86
↓ 1 callersMethod_get_class_balance_factor
(self, dataset_dicts, l=1.)
experts/obj_detection/unidet/data/custom_dataset_dataloader.py:98
↓ 1 callersMethod_get_class_balance_factor_per_dataset
(self, dataset_dicts, l=1.)
experts/obj_detection/unidet/data/multi_dataset_dataloader.py:190
↓ 1 callersFunction_get_coco_stuff_meta
()
experts/segmentation/mask2former/data/datasets/register_coco_stuff_10k.py:182
↓ 1 callersFunction_get_mapillary_vistas_meta
()
experts/segmentation/mask2former/data/datasets/register_mapillary_vistas.py:473
↓ 1 callersMethod_get_tgt_permutation_idx
(self, indices)
experts/segmentation/mask2former/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
experts/segmentation/mask2former/test_time_augmentation.py:71
↓ 1 callersMethod_infinite_indices
(self)
experts/obj_detection/unidet/data/multi_dataset_dataloader.py:175
↓ 1 callersMethod_infinite_indices
(self)
experts/obj_detection/unidet/data/custom_dataset_dataloader.py:88
↓ 1 callersMethod_init_box_head
(self, cfg, input_shape)
experts/obj_detection/unidet/modeling/roi_heads/custom_roi_heads.py:27
↓ 1 callersFunction_is_power_of_2
(n)
experts/segmentation/mask2former/modeling/pixel_decoder/ops/modules/ms_deform_attn.py:28
↓ 1 callersFunction_load_class_freq
(cfg)
experts/obj_detection/unidet/modeling/roi_heads/custom_fast_rcnn.py:22
↓ 1 callersFunction_load_class_hierarchy
(cfg)
experts/obj_detection/unidet/modeling/roi_heads/custom_fast_rcnn.py:45
↓ 1 callersFunction_make_encoder
( backbone, features, use_pretrained, groups=1, expand=False, exportable=True, hoo
experts/depth/blocks.py:12
↓ 1 callersFunction_make_layer_revr
(in_channels, out_channels, num_blocks, **kwargs)
experts/ocr_detection/charnet/modeling/backbone/hourglass.py:24
↓ 1 callersFunction_make_pretrained_resnext101_wsl
(use_pretrained)
experts/depth/blocks.py:133
↓ 1 callersFunction_make_pretrained_vitb16_384
( pretrained, use_readout="ignore", hooks=None, enable_attention_hooks=False )
experts/depth/vit.py:531
↓ 1 callersFunction_make_pretrained_vitb_rn50_384
( pretrained, use_readout="ignore", hooks=None, use_vit_only=False, enable_attention_hooks
experts/depth/vit.py:494
↓ 1 callersFunction_make_pretrained_vitl16_384
( pretrained, use_readout="ignore", hooks=None, enable_attention_hooks=False )
experts/depth/vit.py:515
↓ 1 callersFunction_make_resnet_backbone
(resnet)
experts/depth/blocks.py:120
↓ 1 callersFunction_make_vit_b_rn50_backbone
( model, features=[256, 512, 768, 768], size=[384, 384], hooks=[0, 1, 8, 11], vit_features
experts/depth/vit.py:351
↓ 1 callersFunction_max_by_axis
(the_list)
experts/segmentation/mask2former/utils/misc.py:16
↓ 1 callersFunction_onnx_nested_tensor_from_tensor_list
(tensor_list: List[Tensor])
experts/segmentation/mask2former/utils/misc.py:76
↓ 1 callersFunction_output_size
(dim)
experts/ocr_detection/charnet/modeling/layers/misc.py:98
↓ 1 callersMethod_prepare
Prepare self._gts and self._dts for evaluation based on params.
experts/obj_detection/unidet/evaluation/oideval.py:168
↓ 1 callersMethod_reset_parameters
(self)
experts/segmentation/mask2former/modeling/transformer_decoder/transformer.py:56
↓ 1 callersMethod_reset_parameters
(self)
experts/segmentation/mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:32
↓ 1 callersMethod_reset_parameters
(self)
experts/segmentation/mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:90
↓ 1 callersMethod_reset_parameters
(self)
experts/segmentation/mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:155
↓ 1 callersMethod_reset_parameters
(self)
experts/segmentation/mask2former/modeling/pixel_decoder/fpn.py:186
↓ 1 callersMethod_reset_parameters
(self)
experts/segmentation/mask2former/modeling/pixel_decoder/msdeformattn.py:43
↓ 1 callersMethod_reset_parameters
(self)
experts/segmentation/mask2former/modeling/pixel_decoder/ops/modules/ms_deform_attn.py:66
↓ 1 callersMethod_run_stage
Map back labels
experts/obj_detection/unidet/modeling/roi_heads/unified_roi_heads.py:115
↓ 1 callersMethod_run_stage
support dataset_source
experts/obj_detection/unidet/modeling/roi_heads/split_roi_heads.py:135
↓ 1 callersMethod_set_aux_loss
(self, outputs_class, outputs_seg_masks)
experts/segmentation/mask2former/modeling/transformer_decoder/maskformer_transformer_decoder.py:161
↓ 1 callersMethod_set_aux_loss
(self, outputs_class, outputs_seg_masks)
experts/segmentation/mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:451
↓ 1 callersMethod_summarize
(self, summary_type)
experts/obj_detection/unidet/evaluation/oideval.py:489
↓ 1 callersMethodaccumulate
Accumulate per image evaluation results and store the result in self.eval.
experts/obj_detection/unidet/evaluation/oideval.py:386
↓ 1 callersFunctionadd_maskformer2_config
Add config for MASK_FORMER.
experts/segmentation/mask2former/config.py:6
↓ 1 callersFunctionadd_unidet_config
(cfg)
experts/obj_detection/unidet/config.py:3
↓ 1 callersMethodattention
(self, q: torch.Tensor, kv: torch.Tensor)
model/modules/resampler.py:30
↓ 1 callersFunctionaugment_list
()
dataset/randaugment.py:186
↓ 1 callersFunctionbuild_transform_gen
Create a list of default :class:`Augmentation` from config. Now it includes resizing and flipping. Returns: list[Augmentation]
experts/segmentation/mask2former/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]
experts/segmentation/mask2former/data/dataset_mappers/coco_instance_new_baseline_dataset_mapper.py:37
↓ 1 callersFunctionbuild_transformer_decoder
Build a instance embedding branch from `cfg.MODEL.INS_EMBED_HEAD.NAME`.
experts/segmentation/mask2former/modeling/transformer_decoder/maskformer_transformer_decoder.py:22
↓ 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:
experts/segmentation/mask2former/modeling/criterion.py:73
↓ 1 callersFunctioncheck_forward_equal_with_pytorch_double
()
experts/segmentation/mask2former/modeling/pixel_decoder/ops/test.py:35
↓ 1 callersFunctioncheck_forward_equal_with_pytorch_float
()
experts/segmentation/mask2former/modeling/pixel_decoder/ops/test.py:51
↓ 1 callersFunctioncheck_gradient_numerical
(channels=4, grad_value=True, grad_sampling_loc=True, grad_attn_weight=True)
experts/segmentation/mask2former/modeling/pixel_decoder/ops/test.py:66
↓ 1 callersFunctioncocofy_lvis
Filter LVIS instance segmentation annotations to remove all categories that are not included in COCO. The new json files can be used to evalu
experts/obj_detection/datasets/prepare_cocofied_lvis.py:96
↓ 1 callersFunctioncompute_average_precision
Compute Average Precision according to the definition in VOCdevkit. Precision is modified to ensure that it does not decrease as recall decrease.
experts/obj_detection/unidet/evaluation/oideval.py:31
↓ 1 callersMethodcompute_iou
(self, img_id, cat_id)
experts/obj_detection/unidet/evaluation/oideval.py:262
↓ 1 callersMethodcompute_out_features
(self, idx, up_scale)
experts/edge/model.py:109
↓ 1 callersFunctionconvert
(input, output)
experts/obj_detection/datasets/prepare_ade20k_sem_seg.py:11
↓ 1 callersFunctionconvert_coco_poly_to_mask
(segmentations, height, width)
experts/segmentation/mask2former/data/dataset_mappers/coco_instance_new_baseline_dataset_mapper.py:20
↓ 1 callersFunctioncreate_ade20k_label_colormap
Creates a label colormap used in ADE20K segmentation benchmark. Returns: A colormap for visualizing segmentation results.
utils.py:44
↓ 1 callersFunctioncreate_position_ids_from_input_ids
Replace non-padding symbols with their position numbers. Position numbers begin at padding_idx + 1. Padding symbols are ignored. This is modi
model/modules/roberta.py:38
↓ 1 callersMethoddecompose
(self)
experts/segmentation/mask2former/utils/misc.py:41
↓ 1 callersFunctiondepth_prettify
(file_name)
demo_vis.py:27
↓ 1 callersMethodevaluate_img_google
(self, img_id, cat_id, area_rng)
experts/obj_detection/unidet/evaluation/oideval.py:289
↓ 1 callersMethodfilter_word_instances
(self, word_instances, lexicon)
experts/ocr_detection/charnet/modeling/postprocessing.py:156
↓ 1 callersMethodforward
(self, x)
experts/edge/model.py:212
↓ 1 callersMethodforward
(self, x)
experts/depth/models.py:68
↓ 1 callersMethodforward
(self, x)
experts/segmentation/mask2former/modeling/backbone/swin.py:35
↓ 1 callersMethodforward_features
(self, features)
experts/segmentation/mask2former/modeling/pixel_decoder/fpn.py:284
↓ 1 callersMethodforward_ffn
(self, src)
experts/segmentation/mask2former/modeling/pixel_decoder/msdeformattn.py:116
↓ 1 callersMethodforward_post
( self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: Optional
experts/segmentation/mask2former/modeling/transformer_decoder/transformer.py:182
↓ 1 callersMethodforward_post
( self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: O
experts/segmentation/mask2former/modeling/transformer_decoder/transformer.py:261
↓ 1 callersMethodforward_post
(self, tgt, tgt_mask: Optional[Tensor] = None, tgt_key_padding_mask:
experts/segmentation/mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:40
↓ 1 callersMethodforward_post
(self, tgt, memory, memory_mask: Optional[Tensor] = None, memory_key
experts/segmentation/mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:98
↓ 1 callersMethodforward_post
(self, tgt)
experts/segmentation/mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:163
↓ 1 callersMethodforward_pre
( self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: Optional
experts/segmentation/mask2former/modeling/transformer_decoder/transformer.py:200
↓ 1 callersMethodforward_pre
( self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: O
experts/segmentation/mask2former/modeling/transformer_decoder/transformer.py:292
↓ 1 callersMethodforward_pre
(self, tgt, tgt_mask: Optional[Tensor] = None, tgt_key_padding_mask: O
experts/segmentation/mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:52
↓ 1 callersMethodforward_pre
(self, tgt, memory, memory_mask: Optional[Tensor] = None, memory_key_p
experts/segmentation/mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:112
↓ 1 callersMethodforward_pre
(self, tgt)
experts/segmentation/mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:169
↓ 1 callersFunctionforward_vit
(pretrained, x)
experts/depth/vit.py:104
↓ 1 callersMethodfrom_config
(cls, cfg, input_shape: Dict[str, ShapeSpec])
experts/segmentation/mask2former/modeling/meta_arch/per_pixel_baseline.py:83
↓ 1 callersMethodfrom_config
(cls, cfg, input_shape: Dict[str, ShapeSpec])
experts/segmentation/mask2former/modeling/pixel_decoder/fpn.py:126
↓ 1 callersFunctionfuse_edge
(pred)
experts/edge/images.py:26
↓ 1 callersFunctionget_detection_dataset_dicts_with_source
Similar to detectron2.data.build.get_detection_dataset_dicts, but also returns the dataset source.
experts/obj_detection/unidet/data/multi_dataset_dataloader.py:25
↓ 1 callersFunctionget_extensions
()
experts/segmentation/mask2former/modeling/pixel_decoder/ops/setup.py:26
↓ 1 callersMethodget_ignored_modules
(self, mode='none')
model/prismer.py:61
↓ 1 callersFunctionget_label
(w, h, word_instances)
experts/generate_ocrdet.py:47
↓ 1 callersFunctionget_mask_labels
(depth, instance_boxes, instance_id)
experts/generate_objdet.py:44
← previousnext →101–200 of 669, ranked by callers