Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ai4ce/NYU-VPR
/ functions
Functions
2,323 in github.com/ai4ce/NYU-VPR
⨍
Functions
2,323
◇
Types & classes
300
↳
Endpoints
1
↓ 370 callers
Method
size
test/DBow3/tests/nanoflann.hpp:99
↓ 99 callers
Method
backward
(ctx, grad_o)
segmentation/apex/apex/mlp/mlp.py:19
↓ 97 callers
Method
begin
iterator support
test/DBow3/tests/nanoflann.hpp:622
↓ 89 callers
Method
end
test/DBow3/src/timers.h:99
↓ 59 callers
Method
empty
test/DBow3/tests/nanoflann.hpp:660
↓ 56 callers
Method
step
Performs a single optimization step. Arguments: closure (callable, optional): A closure that reevaluates the model
segmentation/apex/tests/L0/run_optimizers/test_lamb.py:49
↓ 48 callers
Function
filter
(image)
process/whiteFilter.py:29
↓ 48 callers
Method
resize
This method has no effects in this class, but raises an exception if the expected size does not match */
test/DBow3/tests/nanoflann.hpp:664
↓ 46 callers
Function
zero_array
segmentation/apex/apex/contrib/csrc/groupbn/nhwc_batch_norm_kernel.h:290
↓ 45 callers
Method
typeToBytes
(t)
segmentation/apex/apex/pyprof/prof/utility.py:11
↓ 43 callers
Method
zero_grad
(self, models, optimizer, how_to_zero)
segmentation/apex/tests/L0/run_amp/test_add_param_group.py:42
↓ 42 callers
Method
close
(self)
segmentation/apex/apex/pyprof/parse/db.py:59
↓ 38 callers
Method
load
test/DBow3/src/Database.cpp:904
↓ 34 callers
Function
info
(name, param, val)
segmentation/apex/tests/distributed/DDP/ddp_race_condition_test.py:58
↓ 34 callers
Method
scale_loss
(self, loss)
segmentation/apex/apex/amp/opt.py:19
↓ 33 callers
Method
numElems
(shape)
segmentation/apex/apex/pyprof/prof/utility.py:6
↓ 32 callers
Function
to_float
segmentation/apex/apex/contrib/csrc/groupbn/nhwc_batch_norm_kernel.h:80
↓ 30 callers
Function
load_class_names
Args: - dataset_name: str Returns: - list of strings, representing class names
segmentation/mseg-api/mseg/utils/names_utils.py:24
↓ 29 callers
Method
backward
(ctx, grad_y)
segmentation/apex/apex/contrib/groupbn/batch_norm.py:32
↓ 29 callers
Function
compare
(desc, inp1, inp2, error)
segmentation/apex/tests/distributed/synced_batchnorm/single_gpu_unit_test.py:12
↓ 29 callers
Function
read_from_gmem
segmentation/apex/apex/contrib/csrc/groupbn/nhwc_batch_norm_kernel.h:197
↓ 29 callers
Function
read_from_smem
segmentation/apex/apex/contrib/csrc/groupbn/nhwc_batch_norm_kernel.h:215
↓ 27 callers
Function
div_up
segmentation/apex/apex/contrib/csrc/groupbn/nhwc_batch_norm_kernel.h:661
↓ 27 callers
Method
remove
removes reparameterization and backward hook (does not remove forward hook)
segmentation/apex/apex/reparameterization/reparameterization.py:127
↓ 24 callers
Function
ldg_stream
segmentation/apex/apex/contrib/csrc/groupbn/nhwc_batch_norm_kernel.h:108
↓ 23 callers
Method
init
test/DBow3/tests/nanoflann.hpp:90
↓ 23 callers
Method
load_state_dict
(self, state_dict)
segmentation/apex/apex/parallel/LARC.py:69
↓ 20 callers
Method
forward
(ctx, x, s, b, rm, riv, mini_m, mini_riv, ret_cta, mom, epsilon, fuse_relu, is_train, bn_group, my_data, pair_
segmentation/apex/apex/contrib/groupbn/batch_norm.py:9
↓ 20 callers
Function
get_dataloader_id_to_classname_map
Get the mapping stored in our `names.txt` file that maps a class name to a data loader class index. Args: - dataset_name Returns: - data
segmentation/mseg-api/mseg/utils/names_utils.py:79
↓ 20 callers
Method
state_dict
(self)
segmentation/apex/apex/parallel/LARC.py:66
↓ 19 callers
Method
step
Performs a single optimization step. Arguments: closure (callable, optional): A closure that reevaluates the model
segmentation/apex/apex/optimizers/fused_sgd.py:138
↓ 18 callers
Function
write_to_gmem
segmentation/apex/apex/contrib/csrc/groupbn/nhwc_batch_norm_kernel.h:247
↓ 17 callers
Method
insert
(self, cmd, data)
segmentation/apex/apex/pyprof/parse/db.py:36
↓ 16 callers
Method
_deactivate
(self)
segmentation/apex/apex/amp/handle.py:233
↓ 16 callers
Function
add
segmentation/apex/apex/contrib/csrc/groupbn/nhwc_batch_norm_kernel.h:310
↓ 16 callers
Function
compare
(desc, inp1, inp2, error)
segmentation/apex/tests/distributed/synced_batchnorm/two_gpu_unit_test.py:9
↓ 16 callers
Function
compare
(desc, inp1, inp2, error)
segmentation/apex/tests/distributed/synced_batchnorm/test_groups.py:9
↓ 16 callers
Function
resize_img_by_short_side
Resize an RGB image and label. Args: - img_rgb: Array of (M,N) Returns: - img_rgb_resized
segmentation/mseg-api/mseg/utils/resize_util.py:17
↓ 16 callers
Method
update
(self, val, n=1)
segmentation/apex/tests/L1/common/main_amp.py:476
↓ 16 callers
Method
zero_grad
(self)
segmentation/apex/apex/optimizers/fused_sgd.py:113
↓ 15 callers
Method
back
test/DBow3/tests/nanoflann.hpp:656
↓ 15 callers
Method
start
test/DBow3/src/timers.h:98
↓ 15 callers
Function
write_to_smem
segmentation/apex/apex/contrib/csrc/groupbn/nhwc_batch_norm_kernel.h:265
↓ 14 callers
Method
clear
* @brief Vocabulary::clear */
test/DBow3/src/Vocabulary.cpp:1503
↓ 14 callers
Function
common_init
(test_case)
segmentation/apex/tests/L0/run_amp/utils.py:15
↓ 14 callers
Function
fast_read
test/DBow3/src/quicklz.c:99
↓ 14 callers
Function
fast_write
test/DBow3/src/quicklz.c:131
↓ 14 callers
Function
run_layer_test
(test_case, fns, expected, input_shape, test_backward=True)
segmentation/apex/tests/L0/run_amp/test_basic_casts.py:14
↓ 13 callers
Function
compare
(desc, inp1, inp2, error)
segmentation/apex/tests/distributed/synced_batchnorm/python_single_gpu_unit_test.py:5
↓ 13 callers
Function
maybe_print
(msg, rank0=False)
segmentation/apex/apex/amp/_amp_state.py:38
↓ 12 callers
Function
get_unique_mask_identifiers
Will be used to obtain the unique: - filesystem parent (i.e. sequence ID), - image ID - segment ID to overwrite this mask/segment later, by m
segmentation/mseg-api/mseg/label_preparation/mseg_write_relabeled_segments.py:56
↓ 12 callers
Function
get_universal_class_names
Args: - None Returns: - u_names: universal class names
segmentation/mseg-api/mseg/utils/names_utils.py:35
↓ 12 callers
Method
reset_parameters
(self)
segmentation/apex/apex/contrib/multihead_attn/self_multihead_attn.py:97
↓ 12 callers
Method
seed
(self)
segmentation/apex/tests/L0/run_amp/test_checkpointing.py:33
↓ 12 callers
Method
step
(self)
segmentation/apex/apex/parallel/LARC.py:78
↓ 12 callers
Function
to_python_float
(t)
segmentation/apex/apex/fp16_utils/fp16util.py:176
↓ 11 callers
Method
apply
Applies reparametrization to module's `name` parameter and modifies instance attributes as appropriate. `hook_child` adds reparameter
segmentation/apex/apex/reparameterization/reparameterization.py:57
↓ 11 callers
Function
empty
* Returns whether the vocabulary is empty (i.e. it has not been trained) * @return true iff the vocabulary is empty */
test/DBow3/src/Vocabulary.h:128
↓ 11 callers
Function
generate_all_img_label_pair_relative_fpaths
Args: - dname: - split: e.g. 'train', 'val', 'trainval', etc. Returns: -
segmentation/mseg-api/mseg/utils/txt_utils.py:105
↓ 11 callers
Method
kdtree_get_pt
Returns the dim'th component of the idx'th point in the class:
test/DBow3/tests/nanoflann.hpp:1374
↓ 11 callers
Method
loss_scale
(self)
segmentation/apex/apex/amp/scaler.py:73
↓ 11 callers
Function
multiply
segmentation/apex/apex/contrib/csrc/groupbn/nhwc_batch_norm_kernel.h:320
↓ 11 callers
Method
normalize
test/DBow3/src/BowVector.cpp:62
↓ 11 callers
Method
save
Args: filepath (str): a string that contains the absolute path, including the file name, where the visualized ima
segmentation/mseg-api/mseg/utils/mask_utils_detectron2.py:256
↓ 11 callers
Method
zero_grad
(self)
segmentation/apex/apex/parallel/LARC.py:72
↓ 10 callers
Method
add
test/DBow3/src/Database.cpp:82
↓ 10 callers
Function
applier
(value, fn)
segmentation/apex/apex/amp/_initialize.py:39
↓ 10 callers
Function
convert_instance_img_to_mask_img
Given a grayscale image where intensities denote instance IDs (same intensity denotes belonging to same instance), convert this to an RGB image whe
segmentation/mseg-api/mseg/utils/mask_utils.py:823
↓ 10 callers
Method
data
(self, a)
segmentation/apex/apex/pyprof/prof/output.py:107
↓ 10 callers
Function
get_classname_to_dataloaderid_map
Args: - dataset_name Returns: - classname_to_dataloaderid_map:
segmentation/mseg-api/mseg/utils/names_utils.py:59
↓ 10 callers
Function
ldg
segmentation/apex/apex/contrib/csrc/groupbn/nhwc_batch_norm_kernel.h:102
↓ 10 callers
Function
stg_stream
segmentation/apex/apex/contrib/csrc/groupbn/nhwc_batch_norm_kernel.h:158
↓ 10 callers
Method
transform
test/DBow3/src/Vocabulary.cpp:613
↓ 9 callers
Method
create
test/DBow3/src/Vocabulary.cpp:142
↓ 9 callers
Method
gen_single_type_test
(self, param_type=torch.float, device='cuda')
segmentation/apex/tests/L0/run_optimizers/test_fused_optimizer.py:54
↓ 9 callers
Function
get_mask_from_polygon
60x faster than the Matplotlib rasterizer... well done pillow! Args: - polygon: iterable e.g. [(x1,y1),(x2,y2),...] - img_h: integer represe
segmentation/mseg-api/mseg/utils/mask_utils.py:1007
↓ 9 callers
Method
isLeaf
* Returns whether the node is a leaf node * @return true iff the node is a leaf */
test/DBow3/src/Vocabulary.h:342
↓ 9 callers
Method
is_active
(self)
segmentation/apex/apex/amp/handle.py:179
↓ 9 callers
Method
train
(self)
test/posenet/solver.py:84
↓ 9 callers
Method
update
(self, val, n: int = 1)
segmentation/mseg-semantic/mseg_semantic/utils/avg_meter.py:22
↓ 8 callers
Function
check_mkdir
segmentation/mseg-api/mseg/utils/dir_utils.py:7
↓ 8 callers
Function
form_hstacked_imgs
Concatenate images along a horizontal axis and save them. Accept RGB images, and convert to BGR for OpenCV to save them. Args: - img_list: lis
segmentation/mseg-api/mseg/utils/cv2_utils.py:55
↓ 8 callers
Function
form_label_mapping_array
Args: - label_mapping_dict: dictionary from int to int, from original class ID to a new class ID. This is NOT the id_to_class_name dictionary
segmentation/mseg-api/mseg/utils/mask_utils.py:327
↓ 8 callers
Method
gen_param_optim
(self, tensors, options)
segmentation/apex/tests/L0/run_optimizers/test_fused_optimizer.py:19
↓ 8 callers
Function
generate_all_img_label_pair_fpaths
Args: - Returns: -
segmentation/mseg-api/mseg/utils/txt_utils.py:88
↓ 8 callers
Method
get_max_diff
(self, ref_param, tst_param)
segmentation/apex/tests/L0/run_optimizers/test_fused_optimizer.py:43
↓ 8 callers
Function
get_polygons_from_binary_img
cv2.RETR_CCOMP flag retrieves all the contours and arranges them to a 2-level hierarchy. External contours (boundary) of the object are placed in h
segmentation/mseg-api/mseg/utils/mask_utils.py:1083
↓ 8 callers
Function
get_unique_stem_from_last_k_strs
For datasets like ScanNet where image filename stem is not unique. Will not return the suffix, e.g. 'aiport_terminal/ADE_train_00000001_seg.pn
segmentation/mseg-semantic/mseg_semantic/utils/img_path_utils.py:37
↓ 8 callers
Method
loss_scale
(self)
segmentation/apex/apex/fp16_utils/loss_scaler.py:37
↓ 8 callers
Method
rbegin
test/DBow3/tests/nanoflann.hpp:643
↓ 8 callers
Method
update
(self, val, n=1)
segmentation/apex/examples/imagenet/main_amp.py:492
↓ 8 callers
Function
warn_or_err
(msg)
segmentation/apex/apex/amp/_amp_state.py:28
↓ 7 callers
Function
create_leading_fpath_dirs
segmentation/mseg-api/mseg/utils/dir_utils.py:12
↓ 7 callers
Method
execute
(self, cmd)
segmentation/apex/apex/pyprof/parse/db.py:46
↓ 7 callers
Method
find_matches
segmentation/mseg-api/tests/verify_all_dataset_paths_exist.py:109
↓ 7 callers
Method
overlay_instances
Args: boxes (Boxes, RotatedBoxes or ndarray): either a :class:`Boxes`, or an Nx4 numpy array of XYXY_ABS format f
segmentation/mseg-api/mseg/utils/mask_utils_detectron2.py:426
↓ 7 callers
Method
select
(self, cmd)
segmentation/apex/apex/pyprof/parse/db.py:21
↓ 6 callers
Function
CublasStridedBatchedGemm
segmentation/apex/apex/contrib/csrc/multihead_attn/strided_batched_gemm.h:31
↓ 6 callers
Function
Result
* Empty constructors */
test/DBow3/src/QueryResults.h:48
↓ 6 callers
Function
bwd_dx
segmentation/apex/apex/contrib/csrc/groupbn/nhwc_batch_norm_kernel.h:1502
next →
1–100 of 2,323, ranked by callers