MCPcopy Create free account

hub / github.com/NVlabs/prismer / functions

Functions669 in github.com/NVlabs/prismer

Method__init__
(self, skip_input, output_features)
experts/normal/models/submodules/submodules.py:29
Method__init__
(self, data_path, depth_path, transform)
experts/obj_detection/generate_dataset.py:18
Method__init__
Args: cfg (CfgNode): instance_mode (ColorMode): parallel (bool): whether to run the model in different pr
experts/obj_detection/unidet/predictor.py:19
Method__init__
(self, cfg, task_queue, result_queue)
experts/obj_detection/unidet/predictor.py:148
Method__init__
The `__init__` method of any subclass should also contain these arguments. Args: in_channels (int): out_chann
experts/obj_detection/unidet/modeling/backbone/resnest.py:37
Method__init__
The standard block type for ResNet18 and ResNet34. Args: in_channels (int): Number of input channels. out_cha
experts/obj_detection/unidet/modeling/backbone/resnest.py:58
Method__init__
Args: norm (str or callable): a callable that takes the number of channels and return a `nn.Module`, or a pre-def
experts/obj_detection/unidet/modeling/backbone/resnest.py:123
Method__init__
Similar to :class:`BottleneckBlock`, but with deformable conv in the 3x3 convolution.
experts/obj_detection/unidet/modeling/backbone/resnest.py:277
Method__init__
Args: norm (str or callable): a callable that takes the number of channels and return a `nn.Module`, or a pre-def
experts/obj_detection/unidet/modeling/backbone/resnest.py:468
Method__init__
(self, in_channels, out_channels)
experts/obj_detection/unidet/modeling/backbone/resnest.py:776
Method__init__
(self, in_channels, channels, kernel_size, stride=(1, 1), padding=(0, 0), dilation=(1, 1), gr
experts/obj_detection/unidet/modeling/backbone/splat.py:32
Method__init__
(self, radix, cardinality)
experts/obj_detection/unidet/modeling/backbone/splat.py:97
Method__init__
(self, in_channels, channels, kernel_size, stride=(1, 1), padding=(0, 0), dilation=(1, 1), gr
experts/obj_detection/unidet/modeling/backbone/splat.py:117
Method__init__
(self, in_channels, out_channels)
experts/obj_detection/unidet/modeling/backbone/fpn_p5.py:21
Method__init__
( self, cfg, num_classes_list, input_shape: ShapeSpec, **kwargs )
experts/obj_detection/unidet/modeling/roi_heads/multi_dataset_fast_rcnn.py:13
Method__init__
( self, cfg, input_shape: ShapeSpec, **kwargs )
experts/obj_detection/unidet/modeling/roi_heads/custom_fast_rcnn.py:76
Method__init__
(self, cfg)
experts/obj_detection/unidet/modeling/meta_arch/split_rcnn.py:21
Method__init__
(self, cfg)
experts/obj_detection/unidet/modeling/meta_arch/unified_rcnn.py:21
Method__init__
experts/obj_detection/unidet/data/multi_dataset_dataloader.py:125
Method__init__
experts/obj_detection/unidet/data/multi_dataset_dataloader.py:207
Method__init__
experts/obj_detection/unidet/data/custom_dataset_dataloader.py:68
Method__init__
Constructor for OIDEval. Args: lvis_gt (LVIS class instance, or str containing path of annotation file) lvis_dt (LVISR
experts/obj_detection/unidet/evaluation/oideval.py:81
Method__init__
(self, iou_type)
experts/obj_detection/unidet/evaluation/oideval.py:544
Method__init__
(self, dataset_name, cfg, distributed, output_dir=None)
experts/obj_detection/unidet/evaluation/oideval.py:566
Method__init__
( self, unified_label_file, dataset_name, cfg, distributed, output_dir=None)
experts/obj_detection/unidet/evaluation/multi_dataset_evaluator.py:82
Method__init__
( self, unified_label_file, dataset_name, cfg, distributed, output_dir=None)
experts/obj_detection/unidet/evaluation/multi_dataset_evaluator.py:162
Method__iter__
(self)
experts/obj_detection/unidet/data/multi_dataset_dataloader.py:169
Method__iter__
(self)
experts/obj_detection/unidet/data/multi_dataset_dataloader.py:215
Method__iter__
(self)
experts/obj_detection/unidet/data/custom_dataset_dataloader.py:82
Method__len__
(self)
dataset/caption_dataset.py:40
Method__len__
(self)
dataset/pretrain_dataset.py:36
Method__len__
(self)
dataset/vqa_dataset.py:29
Method__len__
(self)
dataset/classification_dataset.py:35
Method__len__
(self)
experts/ocr_detection/generate_dataset.py:24
Method__len__
(self)
experts/edge/generate_dataset.py:24
Method__len__
(self)
experts/depth/generate_dataset.py:24
Method__len__
(self)
experts/segmentation/generate_dataset.py:26
Method__len__
(self)
experts/normal/generate_dataset.py:24
Method__len__
(self)
experts/obj_detection/generate_dataset.py:26
Method__len__
(self)
experts/obj_detection/unidet/predictor.py:212
Method__repr__
(self, _repr_indent=4)
experts/segmentation/mask2former/modeling/matcher.py:181
Method__repr__
(self)
experts/segmentation/mask2former/modeling/criterion.py:249
Method__repr__
(self, _repr_indent=4)
experts/segmentation/mask2former/modeling/transformer_decoder/position_encoding.py:54
Method__repr__
(self)
experts/segmentation/mask2former/utils/misc.py:44
Method_eval_predictions
Evaluate predictions. Fill self._results with the metrics of the tasks.
experts/segmentation/mask2former/evaluation/instance_evaluation.py:43
Method_eval_predictions
(self, tasks, predictions)
experts/obj_detection/unidet/evaluation/multi_dataset_evaluator.py:111
Method_eval_predictions
(self, tasks, predictions)
experts/obj_detection/unidet/evaluation/multi_dataset_evaluator.py:218
Method_freeze_backbone
(self, freeze_at)
experts/ocr_detection/charnet/modeling/backbone/resnet.py:125
Method_init_box_head
(self, cfg, input_shape)
experts/obj_detection/unidet/modeling/roi_heads/custom_roi_heads.py:36
Method_init_box_head
(self, cfg, input_shape)
experts/obj_detection/unidet/modeling/roi_heads/unified_roi_heads.py:20
Method_init_box_head
(self, cfg, input_shape)
experts/obj_detection/unidet/modeling/roi_heads/split_roi_heads.py:20
Method_init_weights
Initialize the weights
model/modules/roberta.py:248
Method_init_weights
(m)
experts/segmentation/mask2former/modeling/backbone/swin.py:642
Method_load_from_state_dict
( self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs )
experts/segmentation/mask2former/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 )
experts/segmentation/mask2former/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 )
experts/segmentation/mask2former/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 )
experts/segmentation/mask2former/modeling/meta_arch/per_pixel_baseline.py:128
Function_make_pretrained_deitb16_384
( pretrained, use_readout="ignore", hooks=None, enable_attention_hooks=False )
experts/depth/vit.py:546
Function_make_pretrained_deitb16_distil_384
( pretrained, use_readout="ignore", hooks=None, enable_attention_hooks=False )
experts/depth/vit.py:561
Method_maybe_read_image
(dataset_dict)
experts/segmentation/mask2former/test_time_augmentation.py:54
Function_process_panoptic_to_semantic
(input_panoptic, output_semantic, segments, id_map)
experts/obj_detection/datasets/prepare_panoptic_fpn.py:18
Function_resize_pos_embed
(self, posemb, gs_h, gs_w)
experts/depth/vit.py:148
Method_set_gradient_checkpointing
(self, module, value=False)
model/modules/roberta.py:262
Method_tie_weights
(self)
model/modules/roberta.py:428
Methodbackward
(ctx, grad_output)
experts/segmentation/mask2former/modeling/pixel_decoder/ops/functions/ms_deform_attn_func.py:43
Functionbatch_dice_loss
Compute the DICE loss, similar to generalized IOU for masks Args: inputs: A float tensor of arbitrary shape. The pred
experts/segmentation/mask2former/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
experts/segmentation/mask2former/modeling/matcher.py:38
Functionbuild_custom_train_loader
Modified from detectron2.data.build.build_custom_train_loader, but supports different samplers
experts/obj_detection/unidet/data/custom_dataset_dataloader.py:23
Functionbuild_multi_dataset_train_loader
Modified from detectron2.data.build.build_custom_train_loader, but supports different samplers
experts/obj_detection/unidet/data/multi_dataset_dataloader.py:63
Functionbuild_p67_resnest_fpn_backbone
Args: cfg: a detectron2 CfgNode Returns: backbone (Backbone): backbone module, must be a subclass of :class:`Backbone`.
experts/obj_detection/unidet/modeling/backbone/resnest.py:791
Functionbuild_p67_resnet_fpn_backbone
Args: cfg: a detectron2 CfgNode Returns: backbone (Backbone): backbone module, must be a subclass of :class:`Backbone`.
experts/obj_detection/unidet/modeling/backbone/fpn_p5.py:37
Functionbuild_resnest_fpn_backbone
Args: cfg: a detectron2 CfgNode Returns: backbone (Backbone): backbone module, must be a subclass of :class:`Backbone`.
experts/obj_detection/unidet/modeling/backbone/resnest.py:725
Functionbuild_retinanet_resnest_fpn_backbone
Args: cfg: a detectron2 CfgNode Returns: backbone (Backbone): backbone module, must be a subclass of :class:`Backbone`.
experts/obj_detection/unidet/modeling/backbone/resnest.py:748
Functioncollate_fn
(batch)
experts/segmentation/generate_dataset.py:40
Functioncollate_fn
(batch)
experts/obj_detection/generate_dataset.py:51
Methodcompute_match_ioa
(ioa)
experts/obj_detection/unidet/evaluation/oideval.py:340
Methodcompute_match_iou
(iou)
experts/obj_detection/unidet/evaluation/oideval.py:327
Functioncompute_normal_errors
(total_normal_errors)
experts/normal/utils/utils.py:63
Functionconcat_image
(image_path_list, concat_image_path)
experts/normal/utils/utils.py:31
Functionconvert_arg_line_to_args
(arg_line)
experts/normal/utils/utils.py:16
Methoddefault_buffer_size
(self)
experts/obj_detection/unidet/predictor.py:224
Methoddfs
(hierarchy, cur_id)
experts/obj_detection/unidet/evaluation/oideval.py:118
Functiondice_loss
Compute the DICE loss, similar to generalized IOU for masks Args: inputs: A float tensor of arbitrary shape. The pred
experts/segmentation/mask2former/modeling/criterion.py:21
Functiondownload_checkpoints
( download_experts: bool = False, download_models: Union[bool, List] = False, hide_tqd
download_checkpoints.py:36
Methodevaluate
(self)
experts/obj_detection/unidet/evaluation/oideval.py:593
Methodevaluate
(self)
experts/obj_detection/unidet/evaluation/multi_dataset_evaluator.py:357
Methodfilter_and_correct
(word_ins, lexicon)
experts/ocr_detection/charnet/modeling/postprocessing.py:169
Methodforward
(self, experts, caption=None, answer=None, train=True, prefix='', inference='generate', k_test=32)
model/prismer_caption.py:15
Methodforward
(self, experts, question, answer=None, weights=None, train=True, inference='rank', k_test=128)
model/prismer_vqa.py:16
Methodforward
(self, x: torch.Tensor, mode='attention')
model/modules/vit.py:55
Methodforward
(self, x: torch.Tensor)
model/modules/vit.py:70
Methodforward
(self, x: dict)
model/modules/vit.py:133
Methodforward
(self, x: torch.Tensor)
model/modules/utils.py:16
Methodforward
(self, x: torch.Tensor)
model/modules/utils.py:24
Methodforward
(self, x: torch.Tensor)
model/modules/utils.py:29
Methodforward
(self, hidden_states: torch.Tensor)
model/modules/utils.py:60
Methodforward
(self, input_ids=None)
model/modules/roberta.py:66
Methodforward
(self, hidden_states: torch.Tensor, attention_mask: Optional[torch.FloatTensor
model/modules/roberta.py:95
Methodforward
(self, hidden_states: torch.Tensor, input_tensor: torch.Tensor)
model/modules/roberta.py:136
Methodforward
( self, hidden_states: torch.Tensor, attention_mask: Optional[torch.FloatT
model/modules/roberta.py:149
← previousnext →401–500 of 669, ranked by callers