MCPcopy Create free account

hub / github.com/baaivision/Painter / functions

Functions326 in github.com/baaivision/Painter

⨍ Functions 326◇ Types & classes 47↳ Endpoints 2
↓ 1 callersFunctionget_args_parser
()
Painter/eval/sidd/painter_inference_sidd.py:39
↓ 1 callersFunctionget_args_parser
()
Painter/eval/nyuv2_depth/painter_inference_depth.py:77
↓ 1 callersFunctionget_args_parser
()
Painter/eval/derain/painter_inference_derain.py:37
↓ 1 callersFunctionget_args_parser
()
Painter/eval/mmpose_custom/painter_inference_pose.py:38
↓ 1 callersFunctionget_args_parser
()
Painter/eval/mmpose_custom/gen_json_coco_pose.py:17
↓ 1 callersFunctionget_args_parser
()
Painter/eval/lol/painter_inference_lol.py:80
↓ 1 callersFunctionget_args_parser
()
Painter/eval/coco_panoptic/COCOPanoSemSegEvaluatorCustom.py:139
↓ 1 callersFunctionget_args_parser
()
Painter/eval/coco_panoptic/COCOInstSegEvaluatorCustom.py:197
↓ 1 callersFunctionget_args_parser
()
Painter/eval/coco_panoptic/COCOCAInstSegEvaluatorCustom.py:30
↓ 1 callersFunctionget_args_parser
()
Painter/eval/coco_panoptic/painter_inference_pano_semseg.py:38
↓ 1 callersFunctionget_args_parser
()
Painter/eval/coco_panoptic/painter_inference_pano_inst.py:38
↓ 1 callersFunctionget_args_parser
()
Painter/eval/ade20k_semantic/painter_inference_segm.py:38
↓ 1 callersFunctionget_args_parser
()
Painter/eval/ade20k_semantic/ADE20kSemSegEvaluatorCustom.py:24
↓ 1 callersFunctionget_args_parser
()
Painter/util/get_toy_dataset.py:9
↓ 1 callersFunctionget_args_parser
()
Painter/util/painter_inference_demo.py:25
↓ 1 callersFunctionget_args_parser
()
SegGPT/SegGPT_inference/seggpt_inference.py:15
↓ 1 callersFunctionget_args_parser_pano_seg
()
Painter/eval/coco_panoptic/COCOPanoEvaluatorCustom.py:279
↓ 1 callersFunctionget_confirm_token
(response)
Painter/data/depth/download_from_gdrive.py:7
↓ 1 callersFunctionget_grad_norm_
(parameters, norm_type: float = 2.0)
Painter/util/misc.py:281
↓ 1 callersFunctionget_layer_id_for_vit
Assign a parameter with its layer id Following BEiT: https://github.com/microsoft/unilm/blob/master/beit/optim_factory.py#L33
Painter/util/lr_decay.py:64
↓ 1 callersFunctionget_loss_scale_for_deepspeed
(model)
Painter/engine_train.py:23
↓ 1 callersFunctionget_metadata
()
Painter/data/register_coco_panoptic_annos_semseg.py:31
↓ 1 callersFunctionget_parameter_groups
(model, weight_decay=1e-5, skip_list=(), get_num_layer=None, get_layer_scale=None)
Painter/util/misc.py:446
↓ 1 callersFunctionget_rank
()
Painter/util/ddp_utils.py:74
↓ 1 callersFunctionget_rank
()
Painter/util/misc.py:202
↓ 1 callersFunctioninference_image
(model, device, img_path, img2_paths, tgt2_paths, out_path)
SegGPT/SegGPT_inference/seggpt_engine.py:56
↓ 1 callersFunctioninference_video
(model, device, vid_path, num_frames, img2_paths, tgt2_paths, out_path)
SegGPT/SegGPT_inference/seggpt_engine.py:106
↓ 1 callersFunctioninit_random_seed
Initialize random seed. If the seed is not set, the seed will be automatically randomized, and then broadcast to all processes to prevent som
Painter/eval/mmpose_custom/apis/train.py:29
↓ 1 callersFunctioninterpolate_pos_embed
(model, checkpoint_model)
Painter/util/pos_embed.py:75
↓ 1 callersFunctionis_main_process
()
Painter/util/misc.py:208
↓ 1 callersFunctioniter_annotations
()
Painter/data/prepare_coco_semantic_annos_from_panoptic_annos.py:58
↓ 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., "~/
Painter/data/register_coco_panoptic_annos_semseg.py:75
↓ 1 callersFunctionload_image_with_retry
(image_path)
Painter/data/coco_semseg/gen_color_coco_panoptic_segm.py:57
↓ 1 callersFunctionmain
(args, ds_init)
Painter/main_train.py:178
↓ 1 callersFunctionmain
()
Painter/data/mmpose_custom/tools/train.py:86
↓ 1 callersFunctionmain
()
Painter/data/mmpose_custom/tools/test.py:91
↓ 1 callersFunctionmain
()
Painter/data/mmdet_custom/tools/train.py:112
↓ 1 callersFunctionmain
()
Painter/data/mmdet_custom/tools/test.py:123
↓ 1 callersFunctionmain
()
Painter/data/depth/eval_with_pngs.py:221
↓ 1 callersFunctionmain
()
Painter/eval/nyuv2_depth/eval_with_pngs.py:220
↓ 1 callersFunctionmain
()
Painter/eval/mmpose_custom/tools/train.py:86
↓ 1 callersFunctionmain
()
Painter/eval/mmpose_custom/tools/test.py:91
↓ 1 callersFunctionmerge_configs
(cfg1, cfg2)
Painter/data/mmpose_custom/tools/test.py:80
↓ 1 callersFunctionmerge_configs
(cfg1, cfg2)
Painter/eval/mmpose_custom/tools/test.py:80
↓ 1 callersMethodmerge_inst_semseg_result
(self, output)
Painter/eval/coco_panoptic/COCOInstSegEvaluatorCustom.py:103
↓ 1 callersMethodmerge_inst_semseg_result_to_instseg
label each instance via max vote Args: semseg_map: (h, w) semseg_dist: (h, w, num_cls) instance_s
Painter/eval/coco_panoptic/COCOInstSegEvaluatorCustom.py:169
↓ 1 callersMethodmerge_inst_semseg_result_to_instseg
label each instance via max vote Args: semseg_map: (h, w) semseg_dist: (h, w, num_cls) instance_s
Painter/eval/coco_panoptic/COCOPanoEvaluatorCustom.py:259
↓ 1 callersMethodmerge_inst_semseg_result_to_panoseg
(self, output)
Painter/eval/coco_panoptic/COCOPanoEvaluatorCustom.py:203
↓ 1 callersFunctionparse_args
()
Painter/data/mmpose_custom/tools/train.py:24
↓ 1 callersFunctionparse_args
()
Painter/data/mmpose_custom/tools/test.py:31
↓ 1 callersFunctionparse_args
()
Painter/data/mmdet_custom/tools/train.py:29
↓ 1 callersFunctionparse_args
()
Painter/data/mmdet_custom/tools/test.py:26
↓ 1 callersFunctionparse_args
()
Painter/eval/mmpose_custom/tools/train.py:24
↓ 1 callersFunctionparse_args
()
Painter/eval/mmpose_custom/tools/test.py:31
↓ 1 callersMethodpatchify
imgs: (N, 3, H, W) x: (N, L, patch_size**2 *3)
Painter/models_painter.py:355
↓ 1 callersMethodpatchify
imgs: (N, 3, H, W) x: (N, L, patch_size**2 *3)
SegGPT/SegGPT_inference/models_seggpt.py:361
↓ 1 callersMethodpost_process_segm_output
Post-processing to turn output segm image to class index map Args: segm: (H, W, 3) Returns: class_m
Painter/eval/ade20k_semantic/ADE20kSemSegEvaluatorCustom.py:114
↓ 1 callersFunctionprepare_model
(chkpt_dir, arch, args=None)
Painter/data/mmpose_custom/painter_inference_pose.py:55
↓ 1 callersFunctionprepare_model
(chkpt_dir, arch='painter_vit_large_patch16_input896x448_win_dec64_8glb_sl1')
Painter/eval/sidd/painter_inference_sidd.py:53
↓ 1 callersFunctionprepare_model
(chkpt_dir, arch='painter_vit_large_patch16_input896x448_win_dec64_8glb_sl1')
Painter/eval/nyuv2_depth/painter_inference_depth.py:40
↓ 1 callersFunctionprepare_model
(chkpt_dir, arch='painter_vit_large_patch16_input896x448_win_dec64_8glb_sl1')
Painter/eval/derain/painter_inference_derain.py:48
↓ 1 callersFunctionprepare_model
(chkpt_dir, arch, args=None)
Painter/eval/mmpose_custom/painter_inference_pose.py:55
↓ 1 callersFunctionprepare_model
(chkpt_dir, arch='painter_vit_large_patch16_input896x448_win_dec64_8glb_sl1')
Painter/eval/lol/painter_inference_lol.py:37
↓ 1 callersFunctionprepare_model
(chkpt_dir, arch='painter_vit_large_patch16_input896x448_win_dec64_8glb_sl1', args=None)
Painter/eval/coco_panoptic/painter_inference_pano_semseg.py:54
↓ 1 callersFunctionprepare_model
(chkpt_dir, arch='painter_vit_large_patch16_input896x448_win_dec64_8glb_sl1', args=None)
Painter/eval/coco_panoptic/painter_inference_pano_inst.py:54
↓ 1 callersFunctionprepare_model
(chkpt_dir, arch='painter_vit_large_patch16_input896x448_win_dec64_8glb_sl1', args=None)
Painter/eval/ade20k_semantic/painter_inference_segm.py:54
↓ 1 callersFunctionprepare_model
(chkpt_dir, arch='painter_vit_large_patch16_input896x448_win_dec64_8glb_sl1')
Painter/util/painter_inference_demo.py:36
↓ 1 callersFunctionprepare_model
(chkpt_dir, arch='seggpt_vit_large_patch16_input896x448', seg_type='instance')
SegGPT/SegGPT_inference/seggpt_inference.py:40
↓ 1 callersMethodprocess
Args: inputs: the inputs to a model. It is a list of dicts. Each dict corresponds to an image and
Painter/eval/coco_panoptic/COCOPanoSemSegEvaluatorCustom.py:67
↓ 1 callersMethodprocess
(self, inputs, outputs)
Painter/eval/coco_panoptic/COCOInstSegEvaluatorCustom.py:88
↓ 1 callersMethodprocess
Args: inputs: the inputs to a COCO model (e.g., GeneralizedRCNN). It is a list of dict. Each dict corresponds to
Painter/eval/coco_panoptic/COCOCAInstSegEvaluatorCustom.py:123
↓ 1 callersMethodprocess
(self, inputs, outputs)
Painter/eval/coco_panoptic/COCOPanoEvaluatorCustom.py:180
↓ 1 callersMethodprocess
Args: inputs: the inputs to a model. It is a list of dicts. Each dict corresponds to an image and
Painter/eval/ade20k_semantic/ADE20kSemSegEvaluatorCustom.py:75
↓ 1 callersFunctionregister_all_coco_panoptic_annos_sem_seg
(root)
Painter/data/register_coco_panoptic_annos_semseg.py:160
↓ 1 callersFunctionregister_coco_panoptic_annos_sem_seg
( name, metadata, image_root, panoptic_root, panoptic_json, sem_seg_root, instances_json )
Painter/data/register_coco_panoptic_annos_semseg.py:129
↓ 1 callersFunctionrun_one_image
(img, tgt, size, model, out_path, device)
Painter/data/mmpose_custom/painter_inference_pose.py:68
↓ 1 callersFunctionrun_one_image
(img, tgt, size, model, out_path, device)
Painter/eval/sidd/painter_inference_sidd.py:63
↓ 1 callersFunctionrun_one_image
(img, tgt, size, model, out_path, device)
Painter/eval/nyuv2_depth/painter_inference_depth.py:51
↓ 1 callersFunctionrun_one_image
(img, tgt, size, model, out_path, device)
Painter/eval/derain/painter_inference_derain.py:58
↓ 1 callersFunctionrun_one_image
(img, tgt, size, model, out_path, device)
Painter/eval/mmpose_custom/painter_inference_pose.py:68
↓ 1 callersFunctionrun_one_image
(img, tgt, size, model, out_path, device)
Painter/eval/lol/painter_inference_lol.py:47
↓ 1 callersFunctionrun_one_image
(img, tgt, size, model, out_path, device)
Painter/eval/coco_panoptic/painter_inference_pano_semseg.py:67
↓ 1 callersFunctionrun_one_image
(img, tgt, size, model, out_path, device)
Painter/eval/coco_panoptic/painter_inference_pano_inst.py:67
↓ 1 callersFunctionrun_one_image
(img, tgt, size, model, out_path, device)
Painter/eval/ade20k_semantic/painter_inference_segm.py:67
↓ 1 callersFunctionrun_one_image
(img, tgt, size, model, out_path, device)
Painter/util/painter_inference_demo.py:47
↓ 1 callersFunctionsave_on_master
(*args, **kwargs)
Painter/util/misc.py:212
↓ 1 callersFunctionsave_response_content
(response, destination)
Painter/data/depth/download_from_gdrive.py:14
↓ 1 callersFunctionseparate_coco_semantic_from_panoptic
Create semantic segmentation annotations from panoptic segmentation annotations, to be used by PanopticFPN. It maps all thing categories
Painter/data/prepare_coco_semantic_annos_from_panoptic_annos.py:29
↓ 1 callersFunctionsetup_for_distributed
This function disables printing when not in master process
Painter/util/ddp_utils.py:45
↓ 1 callersFunctionsimplify_color_dict
(color_dict, num_location_r=16, num_location_gb=20)
Painter/data/mmdet_custom/data/pipelines/transforms.py:58
↓ 1 callersFunctionsingle_gpu_test
Test model with a single gpu. This method tests model with a single gpu and displays test progress bar. Args: model (nn.Module): Mod
Painter/data/mmpose_custom/apis/test.py:13
↓ 1 callersFunctionsingle_gpu_test
Test model with a single gpu. This method tests model with a single gpu and displays test progress bar. Args: model (nn.Module): Mod
Painter/eval/mmpose_custom/apis/test.py:13
↓ 1 callersFunctiontest
()
Painter/data/depth/eval_with_pngs.py:75
↓ 1 callersFunctiontest
()
Painter/eval/nyuv2_depth/eval_with_pngs.py:74
↓ 1 callersFunctiontrain_detector
(model, dataset, cfg, distributed=False,
Painter/data/mmdet_custom/apis/train.py:22
↓ 1 callersFunctiontrain_model
Train model entry function. Args: model (nn.Module): The model to be trained. dataset (Dataset): Train dataset. cfg (dict
Painter/data/mmpose_custom/apis/train.py:62
↓ 1 callersFunctiontrain_model
Train model entry function. Args: model (nn.Module): The model to be trained. dataset (Dataset): Train dataset. cfg (dict
Painter/eval/mmpose_custom/apis/train.py:62
↓ 1 callersFunctiontrain_one_epoch
(model: torch.nn.Module, data_loader: Iterable, optimizer: torch.optim.Optimizer,
Painter/engine_train.py:34
↓ 1 callersFunctionunique
copied from https://github.com/CSAILVision/semantic-segmentation-pytorch/blob/master/mit_semseg/utils.py
Painter/data/ade20k/gen_color_ade20k_sem.py:22
↓ 1 callersFunctionwindow_partition
Partition into non-overlapping windows with padding if needed. Args: x (tensor): input tokens with [B, H, W, C]. window_size
Painter/util/vitdet_utils.py:16
← previousnext →101–200 of 326, ranked by callers