MCPcopy Create free account

hub / github.com/NVlabs/prismer / functions

Functions669 in github.com/NVlabs/prismer

↓ 1 callersFunctionget_metadata
()
experts/segmentation/mask2former/data/datasets/register_mapillary_vistas_panoptic.py:445
↓ 1 callersFunctionget_metadata
()
experts/segmentation/mask2former/data/datasets/register_ade20k_panoptic.py:326
↓ 1 callersFunctionget_metadata
()
experts/segmentation/mask2former/data/datasets/register_coco_panoptic_annos_semseg.py:31
↓ 1 callersMethodget_reference_points
(spatial_shapes, valid_ratios, device)
experts/segmentation/mask2former/modeling/pixel_decoder/msdeformattn.py:141
↓ 1 callersMethodget_valid_ratio
(self, mask)
experts/segmentation/mask2former/modeling/pixel_decoder/msdeformattn.py:52
↓ 1 callersFunctionhourglass88
()
experts/ocr_detection/charnet/modeling/backbone/hourglass.py:102
↓ 1 callersFunctionimage_normalization
This is a typical image normalization function where the minimum and maximum of the image is needed source: https://en.wikipedia.org/wiki/Norm
experts/edge/images.py:6
↓ 1 callersMethodinference
(self, batched_inputs, detected_instances=None, do_postprocess=True)
experts/obj_detection/unidet/modeling/meta_arch/split_rcnn.py:53
↓ 1 callersMethodinference
(self, batched_inputs, detected_instances=None, do_postprocess=True)
experts/obj_detection/unidet/modeling/meta_arch/unified_rcnn.py:65
↓ 1 callersFunctionis_dist_avail_and_initialized
()
experts/segmentation/mask2former/utils/misc.py:106
↓ 1 callersFunctioniter_annotations
()
experts/obj_detection/datasets/prepare_panoptic_fpn.py:62
↓ 1 callersFunctionkappa_to_alpha
(pred_kappa)
experts/normal/utils/utils.py:123
↓ 1 callersMethodlayers
(self, features, mask=None)
experts/segmentation/mask2former/modeling/meta_arch/mask_former_head.py:118
↓ 1 callersMethodlayers
(self, features)
experts/segmentation/mask2former/modeling/meta_arch/per_pixel_baseline.py:109
↓ 1 callersMethodlayers
(self, features)
experts/segmentation/mask2former/modeling/meta_arch/per_pixel_baseline.py:231
↓ 1 callersFunctionload_ade20k_panoptic_json
Args: image_dir (str): path to the raw dataset. e.g., "~/coco/train2017". gt_dir (str): path to the raw annotations. e.g., "~/coc
experts/segmentation/mask2former/data/datasets/register_ade20k_panoptic.py:217
↓ 1 callersFunctionload_char_dict
(path, seperator=chr(31))
experts/ocr_detection/charnet/modeling/postprocessing.py:29
↓ 1 callersFunctionload_coco_json_mem_efficient
Actually not mem efficient
experts/obj_detection/unidet/data/datasets/register_oid.py:43
↓ 1 callersFunctionload_coco_panoptic_json
Args: image_dir (str): path to the raw dataset. e.g., "~/coco/train2017". gt_dir (str): path to the raw annotations. e.g., "~/coc
experts/segmentation/mask2former/data/datasets/register_coco_panoptic_annos_semseg.py:75
↓ 1 callersFunctionload_decoder
(name: str, config: RobertaConfig)
model/modules/roberta.py:433
↓ 1 callersFunctionload_encoder
(name: str, experts: dict, image_resolution: int)
model/modules/vit.py:175
↓ 1 callersFunctionload_lexicon
(path)
experts/ocr_detection/charnet/modeling/postprocessing.py:18
↓ 1 callersFunctionload_mapillary_vistas_panoptic_json
Args: image_dir (str): path to the raw dataset. e.g., "~/coco/train2017". gt_dir (str): path to the raw annotations. e.g., "~/coc
experts/segmentation/mask2former/data/datasets/register_mapillary_vistas_panoptic.py:338
↓ 1 callersMethodmake_deconv_layers
(self, in_features, up_scale)
experts/edge/model.py:95
↓ 1 callersFunctionmake_stage
Create a resnet stage by creating many blocks. Args: block_class (class): a subclass of ResNetBlockBase num_blocks (int):
experts/obj_detection/unidet/modeling/backbone/resnest.py:448
↓ 1 callersFunctionmakedir
(dirpath)
experts/normal/utils/utils.py:94
↓ 1 callersMethodmemory_efficient_forward
More memory-friendly matching
experts/segmentation/mask2former/modeling/matcher.py:96
↓ 1 callersFunctionnested_tensor_from_tensor_list
(tensor_list: List[Tensor])
experts/segmentation/mask2former/utils/misc.py:48
↓ 1 callersFunctionnms_with_char_cls
(boxes, char_scores, overlapThresh, neighbourThresh=0.5, minScore=0, num_neig=0)
experts/ocr_detection/charnet/modeling/rotated_nms.py:58
↓ 1 callersFunctionnorm_to_rgb
(norm)
experts/normal/utils/utils.py:131
↓ 1 callersFunctionobj_detection_prettify
(file_name)
demo_vis.py:34
↓ 1 callersFunctionocr_detection_prettify
(file_name)
demo_vis.py:86
↓ 1 callersMethodoutput_shape
(self)
experts/segmentation/mask2former/modeling/backbone/swin.py:760
↓ 1 callersMethodoutput_shape
(self)
experts/obj_detection/unidet/modeling/backbone/resnest.py:602
↓ 1 callersMethodparse_char
( self, pred_word_fg, pred_char_fg, pred_char_tblr, pred_char_cls, scale_w
experts/ocr_detection/charnet/modeling/postprocessing.py:116
↓ 1 callersMethodparse_word_bboxes
( self, pred_word_fg, pred_word_tblr, pred_word_orient, scale_w, scale_h,
experts/ocr_detection/charnet/modeling/postprocessing.py:90
↓ 1 callersMethodparse_words
(self, word_bboxes, char_bboxes, char_scores, char_dict)
experts/ocr_detection/charnet/modeling/postprocessing.py:218
↓ 1 callersMethodprepare_targets
(self, targets, images)
experts/segmentation/mask2former/maskformer_model.py:264
↓ 1 callersMethodprepare_to_train
(self, mode='none')
model/prismer.py:39
↓ 1 callersFunctionregister_ade20k_panoptic
Register a "standard" version of ADE20k panoptic segmentation dataset named `name`. The dictionaries in this registered dataset follows detec
experts/segmentation/mask2former/data/datasets/register_ade20k_panoptic.py:271
↓ 1 callersFunctionregister_all_ade20k_full
(root)
experts/segmentation/mask2former/data/datasets/register_ade20k_full.py:944
↓ 1 callersFunctionregister_all_ade20k_instance
(root)
experts/segmentation/mask2former/data/datasets/register_ade20k_instance.py:41
↓ 1 callersFunctionregister_all_ade20k_panoptic
(root)
experts/segmentation/mask2former/data/datasets/register_ade20k_panoptic.py:370
↓ 1 callersFunctionregister_all_coco_panoptic_annos_sem_seg
(root)
experts/segmentation/mask2former/data/datasets/register_coco_panoptic_annos_semseg.py:160
↓ 1 callersFunctionregister_all_coco_stuff_10k
(root)
experts/segmentation/mask2former/data/datasets/register_coco_stuff_10k.py:200
↓ 1 callersFunctionregister_all_mapillary_vistas
(root)
experts/segmentation/mask2former/data/datasets/register_mapillary_vistas.py:487
↓ 1 callersFunctionregister_all_mapillary_vistas_panoptic
(root)
experts/segmentation/mask2former/data/datasets/register_mapillary_vistas_panoptic.py:489
↓ 1 callersFunctionregister_coco_panoptic_annos_sem_seg
( name, metadata, image_root, panoptic_root, panoptic_json, sem_seg_root, instances_json )
experts/segmentation/mask2former/data/datasets/register_coco_panoptic_annos_semseg.py:129
↓ 1 callersFunctionregister_mapillary_vistas_panoptic
Register a "standard" version of ADE20k panoptic segmentation dataset named `name`. The dictionaries in this registered dataset follows detec
experts/segmentation/mask2former/data/datasets/register_mapillary_vistas_panoptic.py:392
↓ 1 callersFunctionresize
(im)
experts/ocr_detection/generate_dataset.py:36
↓ 1 callersFunctionresnet50
()
experts/ocr_detection/charnet/modeling/backbone/resnet.py:364
↓ 1 callersFunctionseg_prettify
(file_name)
demo_vis.py:59
↓ 1 callersFunctionseparate_coco_semantic_from_panoptic
Create semantic segmentation annotations from panoptic segmentation annotations, to be used by PanopticFPN. It maps all thing categories
experts/obj_detection/datasets/prepare_panoptic_fpn.py:29
↓ 1 callersMethodsummarize
Compute and display summary metrics for evaluation results.
experts/obj_detection/unidet/evaluation/oideval.py:498
↓ 1 callersFunctionto_numpy_or_none
(*tensors)
experts/ocr_detection/charnet/modeling/model.py:32
↓ 1 callersFunctionunnormalize
(img_in)
experts/normal/utils/utils.py:113
↓ 1 callersMethodupdate_keys_to_ignore
Remove some keys from ignore list
model/modules/roberta.py:266
↓ 1 callersFunctionwarmup_lr_schedule
Warmup the learning rate
utils.py:20
↓ 1 callersFunctionwindow_reverse
Args: windows: (num_windows*B, window_size, window_size, C) window_size (int): Window size H (int): Height of image
experts/segmentation/mask2former/modeling/backbone/swin.py:58
↓ 1 callersMethodwith_pos_embed
(tensor, pos)
experts/segmentation/mask2former/modeling/pixel_decoder/msdeformattn.py:113
↓ 1 callersMethodwrite_as_cityscapes
(coco_results, all_file_names, temp_dir, mapped=False, thing_classes=None, ext='_pred.txt',
experts/obj_detection/unidet/evaluation/multi_dataset_evaluator.py:297
FunctionAutoContrast
(pair, _)
dataset/randaugment.py:88
FunctionBrightness
(pair, v)
dataset/randaugment.py:140
FunctionColor
(pair, v)
dataset/randaugment.py:134
FunctionContrast
(pair, v)
dataset/randaugment.py:128
FunctionCutout
(pair, v)
dataset/randaugment.py:152
FunctionEqualize
(pair, _)
dataset/randaugment.py:98
FunctionFlip
(pair, _)
dataset/randaugment.py:103
FunctionIdentity
(pair, v)
dataset/randaugment.py:182
FunctionInvert
(pair, _)
dataset/randaugment.py:93
FunctionPYBIND11_MODULE
experts/segmentation/mask2former/modeling/pixel_decoder/ops/src/vision.cpp:18
FunctionPosterize
(pair, v)
dataset/randaugment.py:114
FunctionPosterize2
(pair, v)
dataset/randaugment.py:121
FunctionRotate
(pair, v)
dataset/randaugment.py:76
FunctionSharpness
(pair, v)
dataset/randaugment.py:146
FunctionShearX
(pair, v)
dataset/randaugment.py:30
FunctionShearY
(pair, v)
dataset/randaugment.py:37
FunctionSolarize
(pair, v)
dataset/randaugment.py:108
FunctionTranslateX
(pair, v)
dataset/randaugment.py:44
FunctionTranslateXAbs
(pair, v)
dataset/randaugment.py:62
FunctionTranslateY
(pair, v)
dataset/randaugment.py:53
FunctionTranslateYAbs
(pair, v)
dataset/randaugment.py:69
Method__call__
(self, image, labels)
dataset/utils.py:30
Method__call__
(self, img)
dataset/randaugment.py:215
Method__call__
(self, img)
dataset/randaugment.py:235
Method__call__
(self, img, label)
dataset/randaugment.py:259
Method__call__
Same input/output format as :meth:`SemanticSegmentor.forward`
experts/segmentation/mask2former/test_time_augmentation.py:49
Method__call__
Args: dataset_dict (dict): Metadata of one image, in Detectron2 Dataset format. Returns: dict: a format that
experts/segmentation/mask2former/data/dataset_mappers/mask_former_semantic_dataset_mapper.py:98
Method__call__
Args: dataset_dict (dict): Metadata of one image, in Detectron2 Dataset format. Returns: dict: a format that
experts/segmentation/mask2former/data/dataset_mappers/mask_former_panoptic_dataset_mapper.py:59
Method__call__
Args: dataset_dict (dict): Metadata of one image, in Detectron2 Dataset format. Returns: dict: a format that
experts/segmentation/mask2former/data/dataset_mappers/coco_panoptic_new_baseline_dataset_mapper.py:105
Method__call__
Args: dataset_dict (dict): Metadata of one image, in Detectron2 Dataset format. Returns: dict: a format that
experts/segmentation/mask2former/data/dataset_mappers/coco_instance_new_baseline_dataset_mapper.py:121
Method__call__
Args: dataset_dict (dict): Metadata of one image, in Detectron2 Dataset format. Returns: dict: a format that
experts/segmentation/mask2former/data/dataset_mappers/mask_former_instance_dataset_mapper.py:87
Method__call__
(self, image)
experts/obj_detection/unidet/predictor.py:215
Method__getitem__
(self, index)
dataset/caption_dataset.py:43
Method__getitem__
(self, index)
dataset/pretrain_dataset.py:39
Method__getitem__
(self, index)
dataset/vqa_dataset.py:32
Method__getitem__
(self, index)
dataset/classification_dataset.py:38
Method__getitem__
(self, index)
experts/ocr_detection/generate_dataset.py:27
Method__getitem__
(self, index)
experts/edge/generate_dataset.py:27
Method__getitem__
(self, index)
experts/depth/generate_dataset.py:27
← previousnext →201–300 of 669, ranked by callers