MCPcopy Create free account

hub / github.com/NVlabs/prismer / functions

Functions669 in github.com/NVlabs/prismer

Methodforward_R
(self, norm_out, gt_norm, gt_norm_mask)
experts/normal/utils/losses.py:31
Methodforward_UG
(self, pred_list, coord_list, gt_norm, gt_norm_mask)
experts/normal/utils/losses.py:91
Methodforward_features
(self, features)
experts/segmentation/mask2former/modeling/pixel_decoder/msdeformattn.py:315
Functionforward_flex
(self, x)
experts/depth/vit.py:165
Methodfrom_config
(cls, cfg)
experts/segmentation/mask2former/maskformer_model.py:97
Methodfrom_config
(cls, cfg, in_channels, mask_classification)
experts/segmentation/mask2former/modeling/transformer_decoder/maskformer_transformer_decoder.py:108
Methodfrom_config
(cls, cfg, in_channels, mask_classification)
experts/segmentation/mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:337
Methodfrom_config
(cls, cfg, input_shape: Dict[str, ShapeSpec])
experts/segmentation/mask2former/modeling/meta_arch/mask_former_head.py:88
Methodfrom_config
(cls, cfg, input_shape: Dict[str, ShapeSpec])
experts/segmentation/mask2former/modeling/meta_arch/per_pixel_baseline.py:195
Methodfrom_config
(cls, cfg, input_shape: Dict[str, ShapeSpec])
experts/segmentation/mask2former/modeling/pixel_decoder/fpn.py:273
Methodfrom_config
(cls, cfg, input_shape: Dict[str, ShapeSpec])
experts/segmentation/mask2former/modeling/pixel_decoder/msdeformattn.py:295
Methodfrom_config
(cls, cfg, is_train=True)
experts/segmentation/mask2former/data/dataset_mappers/mask_former_semantic_dataset_mapper.py:62
Methodfrom_config
(cls, cfg, is_train=True)
experts/segmentation/mask2former/data/dataset_mappers/coco_panoptic_new_baseline_dataset_mapper.py:94
Methodfrom_config
(cls, cfg, is_train=True)
experts/segmentation/mask2former/data/dataset_mappers/coco_instance_new_baseline_dataset_mapper.py:110
Methodfrom_config
(cls, cfg, is_train=True)
experts/segmentation/mask2former/data/dataset_mappers/mask_former_instance_dataset_mapper.py:59
Methodget_10x_lr_params
(self)
experts/normal/models/NNET.py:18
Methodget_10x_lr_params
(self)
experts/normal/models/baseline.py:28
Methodget_1x_lr_params
(self)
experts/normal/models/NNET.py:15
Methodget_1x_lr_params
(self)
experts/normal/models/baseline.py:25
Methodget_input_embeddings
(self)
model/modules/roberta.py:288
Functionget_mean_attention_map
(attn, token, shape)
experts/depth/vit.py:45
Methodget_output_embeddings
(self)
model/modules/roberta.py:352
Functionget_unified_evaluator
( evaluator_type, dataset_name, cfg, distributed, output_dir)
experts/obj_detection/unidet/evaluation/multi_dataset_evaluator.py:35
Functionhook
(model, input, output)
experts/depth/vit.py:13
Methodinit_weights
Initialize the weights in backbone. Args: pretrained (str, optional): Path to pre-trained weights. Defaults to Non
experts/segmentation/mask2former/modeling/backbone/swin.py:635
Methodinstance_inference
(self, mask_cls, mask_pred)
experts/segmentation/mask2former/maskformer_model.py:344
Functioninterpolate
( input, size=None, scale_factor=None, mode="nearest", align_corners=None )
experts/ocr_detection/charnet/modeling/layers/misc.py:75
Functionload_checkpoint
(fpath, model)
experts/normal/utils/utils.py:47
Functionlog_normal_errors
(metrics, where_to_write, first_line)
experts/normal/utils/utils.py:78
Methodloss_labels
Classification loss (NLL) targets dicts must contain the key "labels" containing a tensor of dim [nb_target_boxes]
experts/segmentation/mask2former/modeling/criterion.py:122
Methodloss_masks
Compute the losses related to the masks: the focal loss and the dice loss. targets dicts must contain the key "masks" containing a tensor of d
experts/segmentation/mask2former/modeling/criterion.py:140
Methodlosses
(self, predictions, proposals, dataset_source)
experts/obj_detection/unidet/modeling/roi_heads/multi_dataset_fast_rcnn.py:46
Functionmake_dir_from_list
(dirpath_list)
experts/normal/utils/utils.py:100
Methodmatch
(word_bbox, word_poly, char_bbox, char_poly)
experts/ocr_detection/charnet/modeling/postprocessing.py:219
Methodmatch_lexicon
(text, lexicon)
experts/ocr_detection/charnet/modeling/postprocessing.py:157
Functionms_deform_attn_backward
experts/segmentation/mask2former/modeling/pixel_decoder/ops/src/ms_deform_attn.h:46
Functionms_deform_attn_cpu_backward
experts/segmentation/mask2former/modeling/pixel_decoder/ops/src/cpu/ms_deform_attn_cpu.cpp:34
Functionms_deform_attn_cpu_forward
experts/segmentation/mask2former/modeling/pixel_decoder/ops/src/cpu/ms_deform_attn_cpu.cpp:22
Functionms_deform_attn_forward
experts/segmentation/mask2former/modeling/pixel_decoder/ops/src/ms_deform_attn.h:25
Functionnms_poly
(polys, scores, overlapThresh, neighbourThresh=0.5, minScore=0, num_neig=0)
experts/ocr_detection/charnet/modeling/rotated_nms.py:166
Methodnms_word_instances
(self, word_instances, h, w, edst=False)
experts/ocr_detection/charnet/modeling/postprocessing.py:200
Methodout_channels
(self)
experts/obj_detection/unidet/modeling/backbone/resnest.py:522
Methodpanoptic_inference
(self, mask_cls, mask_pred)
experts/segmentation/mask2former/maskformer_model.py:286
Methodprepare_inputs_for_generation
(self, input_ids, attention_mask=None, encoder_hidden_states=None, **model_kwargs)
model/modules/roberta.py:401
Methodprocess
(self, inputs, outputs)
experts/obj_detection/unidet/evaluation/oideval.py:585
Methodprocess
Args: inputs: the inputs to a COCO model (e.g., GeneralizedRCNN). It is a list of dict. Each dict corresponds to
experts/obj_detection/unidet/evaluation/multi_dataset_evaluator.py:192
Methodprocess_predictions
(frame, predictions)
experts/obj_detection/unidet/predictor.py:94
Methodrecog
(word_bbox, char_bboxes, char_scores)
experts/ocr_detection/charnet/modeling/postprocessing.py:239
Methodreset
(self)
experts/obj_detection/unidet/evaluation/oideval.py:581
Methodrun
(self)
experts/obj_detection/unidet/predictor.py:154
Methodrun_on_image
Args: image (np.ndarray): an image of shape (H, W, C) (in BGR order). This is the format used by OpenCV.
experts/obj_detection/unidet/predictor.py:42
Methodrun_on_video
Visualizes predictions on frames of the input video. Args: video (cv2.VideoCapture): a :class:`VideoCapture` object, who
experts/obj_detection/unidet/predictor.py:81
Functionsave_args
(args, filename)
experts/normal/utils/utils.py:24
Methodsemantic_inference
(self, mask_cls, mask_pred)
experts/segmentation/mask2former/maskformer_model.py:280
Methodset_eval_dataset
(self, dataset_name)
experts/obj_detection/unidet/modeling/meta_arch/split_rcnn.py:69
Methodset_eval_dataset
(self, dataset_name)
experts/obj_detection/unidet/modeling/meta_arch/unified_rcnn.py:85
Methodset_input_embeddings
(self, value)
model/modules/roberta.py:291
Methodset_output_embeddings
(self, new_embeddings)
model/modules/roberta.py:355
Functionsetup_cfg
(args)
experts/obj_detection/utils.py:5
Methodshutdown
(self)
experts/obj_detection/unidet/predictor.py:219
Functionsigmoid_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/criterion.py:48
Methodsize_divisibility
(self)
experts/segmentation/mask2former/modeling/backbone/swin.py:769
Methodslice
(self, tensor, slice_shape)
experts/edge/model.py:201
Functionsoftnms
(boxes, box_scores, char_scores=None, overlapThresh=0.3, threshold=0.8, neighbourThr
experts/ocr_detection/charnet/modeling/rotated_nms.py:105
Functionstep_lr_schedule
Decay the learning rate
utils.py:27
Methodstride
(self)
experts/obj_detection/unidet/modeling/backbone/resnest.py:529
Functionvisualize
(args, img, gt_norm, gt_norm_mask, norm_out_list, total_iter)
experts/normal/utils/utils.py:140
Functionweight_init
(m)
experts/edge/model.py:6
Methodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
experts/segmentation/mask2former/modeling/transformer_decoder/mask2former_transformer_decoder.py:160
← previous601–669 of 669, ranked by callers