MCPcopy Create free account

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

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

↓ 3 callersMethod_clear_cache
(self)
segmentation/apex/apex/amp/handle.py:226
↓ 3 callersFunction_decorator_helper
(orig_fn, cast_fn, wrap_fn)
segmentation/apex/apex/amp/amp.py:18
↓ 3 callersMethod_make_stage
(self, layer_config, num_inchannels, multi_scale_output=True)
segmentation/mseg-semantic/mseg_semantic/model/seg_hrnet.py:411
↓ 3 callersMethod_make_transition_layer
Use 3x3 convolutions, with stride 2 and padding 1.
segmentation/mseg-semantic/mseg_semantic/model/seg_hrnet.py:350
↓ 3 callersMethod_master_params_to_model_params
(self)
segmentation/apex/apex/fp16_utils/fp16_optimizer.py:160
↓ 3 callersMethodaddIfNotExist
test/DBow3/src/BowVector.cpp:50
↓ 3 callersMethodallreduce_maybe_retain
(self, bucket, bucket_idx, force_default_stream=False)
segmentation/apex/apex/parallel/distributed.py:478
↓ 3 callersFunctioncolormap
Create an array of visually distinctive RGB values. Args: - rgb: boolean, whether to return in RGB or BGR order. BGR corresponds to
segmentation/mseg-api/mseg/utils/colormap.py:12
↓ 3 callersMethodcompute_metrics
Args: - save_vis: whether to save visualize examplars
segmentation/mseg-semantic/mseg_semantic/tool/accuracy_calculator.py:114
↓ 3 callersMethodcompute_sparse_masks
Call this method to enable sparsity. If init(...) was called with allow_recompute_mask=False AND sparsity is disabled, pruned field can be Non
segmentation/apex/apex/contrib/sparsity/asp.py:155
↓ 3 callersFunctionconv1x1
1x1 convolution
segmentation/apex/apex/pyprof/examples/user_annotation/resnet.py:20
↓ 3 callersMethodend
test/DBow3/tests/nanoflann.hpp:624
↓ 3 callersMethodexecute
Execute the demo, i.e. feed all of the desired input through the network and obtain predictions. Gracefully handles .txt, or video file (.mp4,
segmentation/mseg-semantic/mseg_semantic/tool/inference_task.py:339
↓ 3 callersMethodfindNeighbors
test/DBow3/tests/nanoflann.hpp:901
↓ 3 callersMethodfind_inf
(self, sizea, sizeb, applier, repeat_tensors, in_type, out_type, t, ind, val, inplace=False)
segmentation/apex/tests/L0/run_amp/test_multi_tensor_scale.py:55
↓ 3 callersFunctionfoo
(x, y)
segmentation/apex/apex/pyprof/examples/jit/jit_trace_function.py:7
↓ 3 callersFunctionform_vstacked_imgs
Concatenate images along a vertical axis and save them. Accept RGB images, and convert to BGR for OpenCV to save them. Args: - img_list: list
segmentation/mseg-api/mseg/utils/cv2_utils.py:101
↓ 3 callersMethodgen_grad
(self, ref_param, tst_param)
segmentation/apex/tests/L0/run_optimizers/test_lamb.py:170
↓ 3 callersMethodgen_param_optim
(self, tensors, lamb_option)
segmentation/apex/tests/L0/run_optimizers/test_lamb.py:158
↓ 3 callersMethodgen_single_type_test
(self, param_type=torch.float, device="cuda")
segmentation/apex/tests/L0/run_optimizers/test_lamb.py:193
↓ 3 callersMethodgetString
Get the string associated with an id.
segmentation/apex/apex/pyprof/parse/nvvp.py:36
↓ 3 callersFunctionget_ade20k_instance_label_masks
ADE20K non-Scene-Parsing-Challenge data provides instance masks, and stores the instance IDs in the "B" channel of an RGB image. However, the no
segmentation/mseg-api/mseg/dataset_apis/Ade20kMaskLevelDataset.py:200
↓ 3 callersFunctionget_cuda_bare_metal_version
(cuda_dir)
segmentation/apex/setup.py:13
↓ 3 callersFunctionget_loader
(model, image_path, metadata_path, mode, batch_size, is_shuffle=False, num_val=100)
test/posenet/data_loader.py:81
↓ 3 callersMethodget_max_diff
(self, ref_param, tst_param)
segmentation/apex/tests/L0/run_optimizers/test_lamb.py:182
↓ 3 callersFunctionget_most_populous_class
Args: - segment_mask - label_map Returns: - class_mode_idx: integer representing most populous class index
segmentation/mseg-api/mseg/utils/mask_utils.py:1069
↓ 3 callersFunctionintersectionAndUnionGPU
Note output and target sizes are N or N * L or N * H * W Args: - output: Pytorch tensor represeting predicted label map,
segmentation/mseg-semantic/mseg_semantic/utils/iou.py:60
↓ 3 callersFunctionis_fp_tensor
(x)
segmentation/apex/apex/amp/utils.py:14
↓ 3 callersFunctionisfunc
(mod, f)
segmentation/apex/apex/pyprof/nvtx/nvmarker.py:27
↓ 3 callersFunctionmain
()
segmentation/apex/tests/L1/common/main_amp.py:117
↓ 3 callersMethodmask_to_polygons
(self, mask)
segmentation/mseg-api/mseg/utils/mask_utils_detectron2.py:116
↓ 3 callersFunctionmax_pool
(x)
test/vlad_SP/superpoint.py:51
↓ 3 callersMethodmaybe_print
(self, msg)
segmentation/apex/apex/fp16_utils/fp16_optimizer.py:110
↓ 3 callersFunctionnormalize
segmentation/apex/apex/contrib/csrc/groupbn/nhwc_batch_norm_kernel.h:340
↓ 3 callersFunctionparse_dbStruct
(path)
test/netvlad/pittsburgh.py:61
↓ 3 callersFunctionparse_result_file
segmentation/mseg-semantic/mseg_semantic/scripts/collect_results.py:94
↓ 3 callersFunctionpost_backward_models_are_masters
(scaler, params, stashed_grads, scale_override=None)
segmentation/apex/apex/amp/_process_optimizer.py:93
↓ 3 callersFunctionqlz_size_header
test/DBow3/src/quicklz.c:195
↓ 3 callersFunctionquat_to_euler
(q, is_degree=False)
test/posenet/pose_utils.py:5
↓ 3 callersFunctionread_image
(path, device)
test/vlad_SP/main.py:43
↓ 3 callersFunctionreset_table_compress
test/DBow3/src/quicklz.c:64
↓ 3 callersFunctionreset_table_decompress
test/DBow3/src/quicklz.c:77
↓ 3 callersFunctionrgb_img_to_obj_cls_img
Any unmapped pixels (given no corresponding RGB values) will default to zero'th-class. Args: - label_img_rgb: Numpy array of shape (M,N,3) - d
segmentation/mseg-api/mseg/utils/mask_utils.py:359
↓ 3 callersMethodrun_binary_promote_test
(self, fns, input_shape, x_inplace=False)
segmentation/apex/tests/L0/run_amp/test_promotion.py:20
↓ 3 callersMethodrun_cell_test
(self, cell, state_tuple=False)
segmentation/apex/tests/L0/run_amp/test_rnn.py:18
↓ 3 callersMethodrun_rnn_test
(self, rnn, layers, bidir, state_tuple=False)
segmentation/apex/tests/L0/run_amp/test_rnn.py:62
↓ 3 callersFunctionsave_classnames_in_image_sufficientpx
Write a classname over each connected component of a label map as long as the connected component has a sufficiently large number of pixels (sp
segmentation/mseg-api/mseg/utils/mask_utils.py:89
↓ 3 callersFunctionsave_prediction_visualization
Args: - pred_folder - image_path - pred - target_img - id_to_class_name_map Return
segmentation/mseg-semantic/mseg_semantic/tool/accuracy_calculator.py:321
↓ 3 callersFunctionscipy_conn_comp
labelsndarray of dtype int Labeled array, where all connected regions are assigned the same integer value. numint, optional Number of labe
segmentation/mseg-api/mseg/utils/conn_comp.py:14
↓ 3 callersMethodtest
(self)
test/posenet/solver.py:244
↓ 3 callersFunctiontrain_loop
(args, model, optimizer, step, num_steps)
segmentation/apex/apex/contrib/sparsity/test/checkpointing_test_reference.py:35
↓ 3 callersMethodtransform_predictions_test
Function to be called outside. Explicitly for inference on our test datasets. Suppose universal taxonomy has N_u classes, and test taxonomy
segmentation/mseg-api/mseg/taxonomy/taxonomy_converter.py:267
↓ 3 callersFunctionupdate_hash_upto
test/DBow3/src/quicklz.c:245
↓ 3 callersMethodupdate_scale
(self)
segmentation/apex/apex/amp/scaler.py:197
↓ 2 callersMethod__init__
(self, structFile, input_transform=None, onlyDB=False)
test/netvlad/pittsburgh.py:80
↓ 2 callersMethod__init__
(self, *args, **kwargs)
segmentation/apex/tests/L0/run_optimizers/test_fused_optimizer.py:247
↓ 2 callersMethod__init__
(self, dtype)
segmentation/apex/tests/L0/run_amp/test_cache.py:51
↓ 2 callersMethod__init__
(self)
segmentation/apex/tests/L0/run_fp16util/test_fp16util.py:21
↓ 2 callersMethod__init__
(self, gate_multiplier, input_size, hidden_size, cell, n_hidden_states = 2, bias = False, output_size = None)
segmentation/apex/apex/RNN/RNNBackend.py:242
↓ 2 callersMethod__init__
(self, block, layers, num_classes=1000, deep_base=True)
segmentation/mseg-semantic/mseg_semantic/model/resnet.py:104
↓ 2 callersMethod__launch_step_kernel
(self, p, p_copy, m, v, g)
segmentation/apex/apex/contrib/optimizers/distributed_fused_adam.py:355
↓ 2 callersMethod__launch_step_kernel
(self, p, p_copy, m, v, g)
segmentation/apex/apex/contrib/optimizers/distributed_fused_adam_v2.py:406
↓ 2 callersMethod_flatten_grad_mt
(self, scale)
segmentation/apex/apex/contrib/optimizers/distributed_fused_adam_v3.py:237
↓ 2 callersMethod_get_flush_block
(self)
segmentation/apex/apex/contrib/optimizers/distributed_fused_adam_v3.py:202
↓ 2 callersMethod_get_flush_block
(self)
segmentation/apex/apex/contrib/optimizers/distributed_fused_adam.py:303
↓ 2 callersMethod_get_flush_block
(self)
segmentation/apex/apex/contrib/optimizers/distributed_fused_adam_v2.py:354
↓ 2 callersMethod_get_flush_block
(self)
segmentation/apex/apex/contrib/optimizers/distributed_fused_lamb.py:401
↓ 2 callersMethod_init_everything
(self)
segmentation/apex/apex/contrib/optimizers/distributed_fused_lamb.py:390
↓ 2 callersMethod_init_mappings
Populate the train->universal, and universal->test mappings. Args: - None Returns: - None
segmentation/mseg-api/mseg/taxonomy/taxonomy_converter.py:105
↓ 2 callersMethod_pipeline_block_reductions
(self, block_id)
segmentation/apex/apex/contrib/optimizers/distributed_fused_adam.py:319
↓ 2 callersMethod_pipeline_block_reductions
(self, block_id)
segmentation/apex/apex/contrib/optimizers/distributed_fused_adam_v2.py:370
↓ 2 callersMethod_pipeline_block_reductions
(self, block_id)
segmentation/apex/apex/contrib/optimizers/distributed_fused_lamb.py:417
↓ 2 callersMethod_test_same_output
(self, batch_size)
segmentation/apex/tests/L0/run_fused_layer_norm/test_fused_layer_norm.py:16
↓ 2 callersMethod_update_scale
(self, skip)
segmentation/apex/apex/contrib/optimizers/fp16_optimizer.py:142
↓ 2 callersFunctionaccuracy
Computes the precision@k for the specified values of k
segmentation/apex/tests/L1/common/main_amp.py:503
↓ 2 callersFunctionaccuracy
Computes the precision@k for the specified values of k
segmentation/apex/examples/imagenet/main_amp.py:519
↓ 2 callersMethodaddFeature
test/DBow3/src/FeatureVector.cpp:31
↓ 2 callersMethodadd_param_group
(self, param_group)
segmentation/apex/apex/parallel/LARC.py:75
↓ 2 callersFunctionadd_wrapper
(mod, fn_name)
segmentation/apex/apex/pyprof/nvtx/nvmarker.py:67
↓ 2 callersMethodaxpby
(self, sizea, sizeb, applier, repeat_tensors, x_type, y_type, out_type, inplace=False, nhwc=Fals
segmentation/apex/tests/L0/run_amp/test_multi_tensor_axpby.py:45
↓ 2 callersMethodbce_common
(self, assertion)
segmentation/apex/tests/L0/run_amp/test_basic_casts.py:82
↓ 2 callersMethodbytesFlops
(self)
segmentation/apex/apex/pyprof/prof/linear.py:148
↓ 2 callersMethodbytes_flops
(self)
segmentation/apex/apex/pyprof/prof/conv.py:190
↓ 2 callersFunctioncall_once
(path_from_test_file, k_means_codebook_object, vlad_descriptors)
test/vlad/query_image_closest_image_generation.py:107
↓ 2 callersFunctioncall_once
(path_from_test_file, k_means_codebook_object, vlad_descriptors)
test/vlad_SP/find_closest.py:112
↓ 2 callersFunctioncheck_optimizers
(optimizers)
segmentation/apex/apex/amp/_initialize.py:119
↓ 2 callersFunctioncollect_results_at_res
segmentation/mseg-semantic/mseg_semantic/scripts/collect_results.py:157
↓ 2 callersMethodcompute_weight
Computes reparameterized weight value to assign value to module attribute with name `name`. See WeightNorm class for example.
segmentation/apex/apex/reparameterization/reparameterization.py:28
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
segmentation/mseg-semantic/mseg_semantic/model/seg_hrnet.py:34
↓ 2 callersFunctionconvertTransToCublasOperation
segmentation/apex/apex/contrib/csrc/multihead_attn/strided_batched_gemm.h:21
↓ 2 callersFunctionconvert_dictionaries
segmentation/mseg-api/mseg/utils/dictionary_utils.py:5
↓ 2 callersFunctionconvert_label_to_pred_taxonomy
Args: - label map in `semseg`-format taxonomy, e.g. `coco-panoptic-133` or `coco-panoptic-133-relabeled` Retur
segmentation/mseg-semantic/mseg_semantic/tool/relabeled_eval_utils.py:25
↓ 2 callersFunctionconvert_network
Converts a network's parameters and buffers to dtype.
segmentation/apex/apex/fp16_utils/fp16util.py:60
↓ 2 callersFunctioncuChanOnlineSum
segmentation/apex/apex/contrib/csrc/multihead_attn/layer_norm.h:22
↓ 2 callersFunctioncv2_write_rgb
Args: - save_fpath Returns: - None
segmentation/mseg-api/mseg/utils/cv2_utils.py:12
↓ 2 callersFunctiondescribe_SP
(img, model)
test/vlad_SP/main.py:53
↓ 2 callersMethoddraw_binary_mask
Args: binary_mask (ndarray): numpy array of shape (H, W), where H is the image height and W is the image width. E
segmentation/mseg-api/mseg/utils/mask_utils_detectron2.py:586
↓ 2 callersMethoddraw_polygon
Args: segment: numpy array of shape Nx2, containing all the points in the polygon. color: color of the polygon. Refer
segmentation/mseg-api/mseg/utils/mask_utils_detectron2.py:646
↓ 2 callersMethoddraw_text
Args: text (str): class label position (tuple): a tuple of the x and y coordinates to place text on image.
segmentation/mseg-api/mseg/utils/mask_utils_detectron2.py:535
↓ 2 callersMethoddump_acc_results_to_file
Save per-class IoUs and mIoU to a .txt file. When evaluating a model trained within the universal taxonomy, on the val split
segmentation/mseg-semantic/mseg_semantic/tool/accuracy_calculator.py:291
← previousnext →201–300 of 2,323, ranked by callers