MCPcopy Create free account

hub / github.com/ai4ce/NYU-VPR / functions

Functions2,323 in github.com/ai4ce/NYU-VPR

↓ 1 callersFunctionget_seg_model
( cfg, criterion, n_classes: int, load_imagenet_model: bool = False, imagenet_ckpt_fpath:
segmentation/mseg-semantic/mseg_semantic/model/seg_hrnet.py:545
↓ 1 callersMethodget_segment_mask
seq_id is only provided so that all other datasets can share a common API.
segmentation/mseg-api/mseg/dataset_apis/SunrgbdImageLevelDataset.py:94
↓ 1 callersMethodget_segment_mask
Args: - segmentid: integer representing segment unique ID - fname_stem - split: dataset split, i.e. 'train' or 'val' Returns: - se
segmentation/mseg-api/mseg/dataset_apis/Ade20kMaskLevelDataset.py:166
↓ 1 callersMethodget_semantic_annotations
Get COCO Panoptic semantic annotations from .json file Args: - split: string representing training, validation, or testing split of the data
segmentation/mseg-api/mseg/dataset_apis/COCOSemanticAPI.py:38
↓ 1 callersFunctiongetting_VLAD_for_Query_image
(path, visualDictionary)
test/vlad/query_image_closest_image_generation.py:42
↓ 1 callersFunctiongetting_VLAD_for_Query_image
(path, visualDictionary)
test/vlad_SP/find_closest.py:58
↓ 1 callersFunctiongps
(query)
test/DBow3/utils/score_mix.py:111
↓ 1 callersFunctionharmonic_mean
1. Take the reciprocal of all numbers in the dataset 2. Find the arithmetic mean of those reciprocals 3. Take the reciprocal of that number
segmentation/mseg-semantic/mseg_semantic/scripts/collect_results.py:134
↓ 1 callersMethodhasNVTX
(marker)
segmentation/apex/apex/pyprof/prof/utility.py:45
↓ 1 callersFunctionhasTileSize
(name)
segmentation/apex/apex/pyprof/prof/recurrentCell.py:5
↓ 1 callersMethodheader
(self)
segmentation/apex/apex/pyprof/prof/output.py:99
↓ 1 callersFunctionhstack_img_with_palette
Horizontally stack a left image with a palette image on the right.
segmentation/mseg-api/mseg/utils/mask_utils.py:704
↓ 1 callersMethodimage
(self)
test/posenet/solver.py:387
↓ 1 callersFunctionimread_rgb
Returns: - RGB 3 channel nd-array with shape H * W * 3
segmentation/mseg-semantic/mseg_semantic/tool/inference_task.py:169
↓ 1 callersFunctioninit
(enabled=True, loss_scale="dynamic", enable_caching=True, verbose=False, allow_banned=False)
segmentation/apex/apex/amp/amp.py:68
↓ 1 callersMethodinit_inference
init_inference()
segmentation/apex/apex/RNN/RNNBackend.py:225
↓ 1 callersMethodinit_outf
segmentation/mseg-semantic/mseg_semantic/utils/cv2_video_utils.py:27
↓ 1 callersMethodinit_params
(self, clsts, traindescs)
test/netvlad/netvlad.py:35
↓ 1 callersMethodinit_weights
For training, we use a model pretrained on ImageNet. Irrelevant at inference. Args: - pretrained_fpath: str representing pa
segmentation/mseg-semantic/mseg_semantic/model/seg_hrnet.py:511
↓ 1 callersMethodinstance_masks
(self)
segmentation/mseg-api/mseg/utils/mask_utils_detectron2.py:186
↓ 1 callersFunctionis_iterable
(maybe_iterable)
segmentation/apex/apex/RNN/RNNBackend.py:10
↓ 1 callersMethodisleaf
test/DBow3/tests/test_fbow.cpp:69
↓ 1 callersFunctionjit_dropout_add
(x, residual, prob, is_training)
segmentation/apex/apex/contrib/multihead_attn/encdec_multihead_attn.py:19
↓ 1 callersFunctionjit_dropout_add
(x, residual, prob, is_training)
segmentation/apex/apex/contrib/multihead_attn/self_multihead_attn.py:19
↓ 1 callersFunctionkMeansDictionary
(training, k, kmeans)
test/vlad/vlad_codebook_generation.py:63
↓ 1 callersFunctionkMeansDictionary
(training, k, kmeans)
test/vlad_SP/main.py:15
↓ 1 callersMethodkdtree_distance
Returns the L2 distance between the vector "p1[0:size-1]" and the data point with index "idx_p2" stored in the class:
test/DBow3/tests/nanoflann.hpp:1363
↓ 1 callersMethodkdtree_get_bbox
test/DBow3/tests/nanoflann.hpp:1382
↓ 1 callersMethodknnSearch
* Find the "num_closest" nearest neighbors to the \a query_point[0:dim-1]. Their indices are stored inside * the result object. * \sa radiusSea
test/DBow3/tests/nanoflann.hpp:923
↓ 1 callersMethodl2norm
(self, sizea, sizeb, applier, repeat_tensors, in_type, per_tensor)
segmentation/apex/tests/L0/run_amp/test_multi_tensor_l2norm.py:35
↓ 1 callersMethodlabelrgb_to_label
Args: - label_img_rgb: Returns: - label_img
segmentation/mseg-api/mseg/dataset_apis/MapillaryMaskDataset.py:48
↓ 1 callersFunctionloadFeatures
test/DBow3/utils/demo_general.cpp:52
↓ 1 callersFunctionloadFeatures
test/DBow3/utils/create_voc_step0.cpp:46
↓ 1 callersFunctionloadFeatures
test/DBow3/tests/test_fbow.cpp:20
↓ 1 callersFunctionloadFeatures
test/DBow3/tests/test_bigvoc.cpp:21
↓ 1 callersFunctionloadFeatures
test/DBow3/tests/test_flann.cpp:21
↓ 1 callersFunctionload_csv
()
test/vlad/vlad_codebook_generation.py:130
↓ 1 callersFunctionload_csv
()
test/vlad_SP/find_closest.py:143
↓ 1 callersFunctionload_csv
()
test/vlad_SP/main.py:103
↓ 1 callersFunctionload_csv
()
test/DBow3/anony_side_view/non_paths.py:3
↓ 1 callersMethodload_model
Load Pytorch pre-trained model from disk of type torch.nn.DataParallel. Note that `args.num_model_classes` will be size of logits output.
segmentation/mseg-semantic/mseg_semantic/tool/inference_task.py:292
↓ 1 callersMethodload_pretrained_model
(self)
test/posenet/solver.py:64
↓ 1 callersMethodload_state_dict
Loads a state_dict created by an earlier call to state_dict(). If ``fp16_optimizer_instance`` was constructed from some ``init_optim
segmentation/apex/apex/fp16_utils/fp16_optimizer.py:230
↓ 1 callersFunctionmain
(config)
test/posenet/train.py:8
↓ 1 callersFunctionmain
(config)
test/posenet/single_test.py:8
↓ 1 callersFunctionmain
(config)
test/posenet/image_test.py:8
↓ 1 callersFunctionmain
We use the MSeg dataroot explicitly, as specified in mseg/utils/dataset_config.py
segmentation/mseg-api/tests/verify_all_relabeled_segments.py:353
↓ 1 callersFunctionmain
Given the full tsv with all dataset->universal mappings, and given relabeled update records, we form new "relabeled taxonomies" for each dataset, a
segmentation/mseg-api/tests/find_relabeled_taxonomies.py:138
↓ 1 callersFunctionmain
We use the MSeg dataroot explicitly, as specified in mseg/utils/dataset_config.py
segmentation/mseg-api/mseg/label_preparation/mseg_write_relabeled_segments.py:450
↓ 1 callersFunctionmain
Convert a json file path e.g. gtFine/train/1/210316_gtFine_polygons.json to: gtFine/train/1/210316_gtFine_labelids.png
segmentation/mseg-api/mseg/label_preparation/dump_idd_semantic_labels.py:192
↓ 1 callersFunctionmain
Visualize masks of a chosen category from BDD. Usage: python BDDImageLevelDataset.py \ --dataroot /Users/johnlamb/Downloads/bdd100k --classname
segmentation/mseg-api/mseg/dataset_apis/BDDImageLevelDataset.py:155
↓ 1 callersFunctionmain
Visualize masks of a chosen category from SUN RGB-D. Usage: python SunrgbdImageLevelDataset.py \ --dataroot /Users/johnlamb/Downloads/SUNRGBD-37
segmentation/mseg-api/mseg/dataset_apis/SunrgbdImageLevelDataset.py:130
↓ 1 callersFunctionmain
Example Use: python MapillaryMaskDataset.py -classname="Ground Animal" --folder_prefix mseg_mapillary_water_2019_04_19 --num_processes 4 --dataroo
segmentation/mseg-api/mseg/dataset_apis/MapillaryMaskDataset.py:206
↓ 1 callersFunctionmain
Visualize masks of a chosen category from COCO Panoptic. Interesting choices: 'floor','curtain','mountain','rug','hill' 'light', 'lamp', 'chan
segmentation/mseg-api/mseg/dataset_apis/Ade20kMaskLevelDataset.py:264
↓ 1 callersFunctionmain
Visualize masks of a chosen category from COCO Panoptic. Interesting choices: 'cabinet-merged', 'rug-merged', 'road', 'bus', 'truck', 'cup','
segmentation/mseg-api/mseg/dataset_apis/COCOPanopticJsonMaskDataset.py:148
↓ 1 callersFunctionmain
segmentation/mseg-api/download_scripts/quick_diff.py:25
↓ 1 callersFunctionmain
(args)
segmentation/apex/apex/contrib/sparsity/test/toy_problem.py:38
↓ 1 callersFunctionmain
(args)
segmentation/apex/apex/contrib/sparsity/test/checkpointing_test_part1.py:38
↓ 1 callersFunctionmain
(args)
segmentation/apex/apex/contrib/sparsity/test/checkpointing_test_reference.py:42
↓ 1 callersFunctionmain
(step, args, model_state_dict, optimizer_state_dict)
segmentation/apex/apex/contrib/sparsity/test/checkpointing_test_part2.py:38
↓ 1 callersFunctionmain
()
segmentation/apex/apex/pyprof/prof/prof.py:171
↓ 1 callersFunctionmain
()
segmentation/apex/apex/pyprof/parse/parse.py:25
↓ 1 callersFunctionmain
()
segmentation/apex/apex/pyprof/examples/imagenet/imagenet.py:89
↓ 1 callersFunctionmain
()
segmentation/apex/examples/imagenet/main_amp.py:95
↓ 1 callersFunctionmake_cast_wrapper
(orig_fn, cast_fn, handle, try_caching=False)
segmentation/apex/apex/amp/wrap.py:10
↓ 1 callersFunctionmake_dataset
Args: - split: string representing split of data set to use, must be either 'train','val','test' - data_root: path to whe
segmentation/mseg-semantic/mseg_semantic/utils/dataset.py:24
↓ 1 callersFunctionmake_promote_wrapper
(orig_fn, cast_fn, handle=None)
segmentation/apex/apex/amp/wrap.py:44
↓ 1 callersMethodmalloc
* Returns a pointer to a piece of new memory of the given size in bytes * allocated from the pool. */
test/DBow3/tests/nanoflann.hpp:515
↓ 1 callersFunctionmask_belongs_single_semantic_class
Args: - segment_mask - label_img Returns: - is_single_class - sem_class_ids
segmentation/mseg-api/mseg/dataset_apis/MapillaryMaskDataset.py:189
↓ 1 callersFunctionmix
(train, test, index)
test/DBow3/utils/score_mix.py:45
↓ 1 callersFunctionmn_1d_best
(matrix, m, n)
segmentation/apex/apex/contrib/sparsity/sparse_masklib.py:37
↓ 1 callersFunctionmn_2d_best
(matrix, m, n)
segmentation/apex/apex/contrib/sparsity/sparse_masklib.py:122
↓ 1 callersFunctionmn_2d_greedy
(matrix, m, n)
segmentation/apex/apex/contrib/sparsity/sparse_masklib.py:67
↓ 1 callersFunctionmodMarker
Returns the stringified extra_repr() of a module.
segmentation/apex/apex/pyprof/nvtx/nvmarker.py:56
↓ 1 callersFunctionmodel_parser
(model, fixed_weight=False, dropout_rate=0.0, bayesian = False)
test/posenet/model.py:18
↓ 1 callersFunctionndarray
(arg, name="")
segmentation/apex/apex/pyprof/nvtx/nvmarker.py:121
↓ 1 callersMethodnet_process
Feed input through the network. In addition to running a crop through the network, we can flip the crop horizontally, run both crops through t
segmentation/mseg-semantic/mseg_semantic/tool/inference_task.py:642
↓ 1 callersMethodnon_empty_mask
Returns: (H, W) array, a mask for all pixels that have a prediction
segmentation/mseg-api/mseg/utils/mask_utils_detectron2.py:162
↓ 1 callersFunctionnormalize_img
Pass in by reference Torch tensor, and normalize its values. Args: - input: Torch tensor of shape (3,M,N), must be in this order,
segmentation/mseg-semantic/mseg_semantic/utils/normalization_utils.py:25
↓ 1 callersMethodnum_flat_features
(self, x)
segmentation/apex/apex/pyprof/examples/lenet.py:35
↓ 1 callersMethodop
(self)
segmentation/apex/apex/pyprof/prof/linear.py:145
↓ 1 callersMethodop
(self)
segmentation/apex/apex/pyprof/prof/blas.py:336
↓ 1 callersFunctionopenFile
(f)
segmentation/apex/apex/pyprof/prof/usage.py:17
↓ 1 callersMethodops
(input, weight0, weight1)
segmentation/apex/tests/L0/run_amp/test_fused_sgd.py:34
↓ 1 callersMethodops
(input, weight0, weight1)
segmentation/apex/tests/L0/run_amp/test_multiple_models_optimizers_losses.py:24
↓ 1 callersMethodops
(input, weight)
segmentation/apex/tests/L0/run_amp/test_cache.py:30
↓ 1 callersMethodops
(input, weight)
segmentation/apex/tests/L0/run_amp/test_cache.py:43
↓ 1 callersMethodops
(input, weight)
segmentation/apex/tests/L0/run_amp/test_cache.py:56
↓ 1 callersMethodops
(input, weight0, weight1)
segmentation/apex/tests/L0/run_amp/test_add_param_group.py:24
↓ 1 callersMethodparams
(self)
segmentation/apex/apex/pyprof/prof/conv.py:180
↓ 1 callersFunctionparse
()
segmentation/apex/examples/imagenet/main_amp.py:44
↓ 1 callersFunctionparseArgs
Print usage and parse arguments.
segmentation/apex/apex/pyprof/prof/usage.py:4
↓ 1 callersFunctionparseArgs
()
segmentation/apex/apex/pyprof/parse/parse.py:15
↓ 1 callersFunctionparseArgs
()
segmentation/apex/apex/pyprof/examples/imagenet/imagenet.py:17
↓ 1 callersFunctionparse_entry
Parse an entry from dataset element, return list of classes. Args: - entry: string, representing cell in taxonomy spreadsheet Returns: - cl
segmentation/mseg-api/tests/find_relabeled_taxonomies.py:26
↓ 1 callersFunctionparse_test_entry
Can be a blank string, e.g. ''. We ensure no additional whitespace present. Args: - test_entry Returns: - test_entry
segmentation/mseg-api/mseg/taxonomy/taxonomy_converter.py:322
↓ 1 callersFunctionparse_uentry
Args: - uentry: string, representing TSV entry from `Universal` taxonomy column Returns: - level: integer, representing depth of node in tre
segmentation/mseg-api/tests/find_relabeled_taxonomies.py:53
↓ 1 callersFunctionpatchClass
(cls)
segmentation/apex/apex/pyprof/nvtx/nvmarker.py:201
↓ 1 callersFunctionpatch_step
(opt, loss_scaler, loss_id)
segmentation/apex/apex/amp/handle.py:133
↓ 1 callersFunctionpatch_step
(old_step)
segmentation/apex/apex/amp/_initialize.py:239
← previousnext →601–700 of 2,323, ranked by callers