MCPcopy Create free account

hub / github.com/ShihaoShao-GH/SuperGlobal / functions

Functions103 in github.com/ShihaoShao-GH/SuperGlobal

⨍ Functions 103◇ Types & classes 23
↓ 9 callersMethod__init__
(self, RESNET_DEPTH, REDUCTION_DIM, relup)
model/resnet.py:222
↓ 4 callersMethodforward
(self, x)
model/base/conv4d.py:48
↓ 3 callersFunctioncompute_map
Computes the mAP for a given set of returned results. Usage: map = compute_map (ranks, gnd) computes mean
test/evaluate.py:41
↓ 3 callersFunctionmix_score
(sim_ori, sim_corr, ratio)
test/test_utils.py:74
↓ 2 callersMethodbuild_crossscale_correlation
(cls, query_feats, key_feats, scales, conv2ds)
model/base/correlation.py:50
↓ 2 callersFunctionextract_feature
(model, data_dir, dataset, gnd_fn, split, scale_list_fix, gemp, rgem, sgem, scale_list)
test/test_utils.py:11
↓ 2 callersMethodextract_global_descriptor
(self, im_q, gemp, rgem, sgem, scale_list)
model/CVNet_Rerank_model.py:62
↓ 1 callersMethod_construct
(self)
model/resnet.py:229
↓ 1 callersMethod_construct_db
Constructs the db.
test/dataset.py:32
↓ 1 callersMethod_construct_db
Constructs the db.
test/dataset_1m.py:37
↓ 1 callersFunction_construct_loader
Constructs the data loader for the given dataset.
test/test_loader.py:18
↓ 1 callersMethod_forward_singlescale
(self, x, gemp = True, rgem = True)
model/resnet.py:248
↓ 1 callersMethod_prepare_im
Prepares the image for network input.
test/dataset.py:55
↓ 1 callersMethod_prepare_im
Prepares the image for network input.
test/dataset_1m.py:44
↓ 1 callersMethodapply_gaussian_kernel
(cls, corr, sigma=17)
model/base/geometry.py:56
↓ 1 callersMethodattentive_indexing
r"""kps: normalized keypoints x, y (N, 2) returns attentive index map(N, spatial_side, spatial_side)
model/base/geometry.py:39
↓ 1 callersFunctioncenter_crop
Performs center cropping (HWC format).
core/transforms.py:67
↓ 1 callersFunctioncompute_ap
Computes average precision for given ranked indexes. Arguments --------- ranks : zerro-based ranks of positive images nres
test/evaluate.py:5
↓ 1 callersMethodcompute_crossscale_correlation
Build 6-dimensional correlation tensor
model/base/correlation.py:13
↓ 1 callersFunctionconfig_gnd
(dataset, dir_main)
test/config_gnd.py:8
↓ 1 callersFunctionextract_feature
(model, test_loader, scale_list)
extract_rop1m.py:9
↓ 1 callersFunctionget_trans_fun
Retrieves the transformation function by name.
model/resnet.py:28
↓ 1 callersMethodinterpolate4d
(tensor4d, size)
model/base/geometry.py:115
↓ 1 callersFunctionmain
()
test.py:8
↓ 1 callersFunctionmain
()
extract_rop1m.py:41
↓ 1 callersMethodnormalize_kps
(cls, kps)
model/base/geometry.py:25
↓ 1 callersMethodprune
(self, ct)
model/base/conv4d.py:24
↓ 1 callersMethodprune_out2
(self, ct)
model/base/conv4d.py:36
↓ 1 callersFunctionread_imlist
(imlist_fn)
test/config_gnd.py:52
↓ 1 callersFunctionread_imlist
(imlist_fn)
test/dataset_1m.py:22
↓ 1 callersFunctionscale
Performs scaling (HWC format).
core/transforms.py:53
↓ 1 callersFunctionsetup_model
Sets up a model for training or testing and log the results.
core/CVNet_tester.py:22
↓ 1 callersFunctiontest_model
(model, data_dir, dataset_list, scale_list, is_rerank, gemp, rgem, sgem, onemeval, depth, logger)
test/test_model.py:16
↓ 1 callersFunctiontest_revisitop
(cfg, ks, ranks)
test/test_utils.py:39
↓ 1 callersFunctionzero_pad
Performs zero padding (CHW format).
core/transforms.py:24
Method__getitem__
(self, index)
test/dataset.py:64
Method__getitem__
(self, index)
test/dataset_1m.py:53
Method__init__
(self, M=400, K = 9, beta = 0.15)
modules/reranking/RerankwMDA.py:12
Method__init__
(self, M = 400, K = 9, beta = 0.15)
modules/reranking/MDescAug.py:12
Method__init__
(self, pr=2.5, size = 5)
modules/coarse_retrieval/rgem.py:12
Method__init__
(self, p=4.6, eps = 1e-8)
modules/coarse_retrieval/gemp.py:12
Method__init__
(self, ps=10., infinity = True)
modules/coarse_retrieval/sgem.py:12
Method__init__
(self, alpha=0.014)
modules/coarse_retrieval/relup.py:12
Method__init__
(self, RESNET_DEPTH, REDUCTION_DIM, relup)
model/CVNet_Rerank_model.py:19
Method__init__
(self, inch)
model/CVlearner.py:12
Method__init__
(self, norm, output_size=1, eps=1e-6)
model/resnet.py:51
Method__init__
(self, norm=3, output_size=1, eps=1e-6)
model/resnet.py:72
Method__init__
(self, w_in, nc, pp=3)
model/resnet.py:78
Method__init__
(self, w_in, w_out, stride, w_b=None, num_gs=1)
model/resnet.py:92
Method__init__
(self, w_in, w_out, stride, w_b, num_gs, relup_)
model/resnet.py:114
Method__init__
(self, w_in, w_out, stride, trans_fun, w_b=None, num_gs=1, relup_=True)
model/resnet.py:142
Method__init__
(self, w_in, w_out, stride, d, w_b=None, num_gs=1, relup=True)
model/resnet.py:173
Method__init__
(self, w_in, w_out, stride, d, w_b=None, num_gs=1)
model/resnet.py:190
Method__init__
(self, w_in, w_out)
model/resnet.py:207
Method__init__
(self, in_channels, out_channels, kernel_size, stride, padding, bias=True)
model/base/conv4d.py:9
Method__init__
(self, data_path, dataset, fn, split, scale_list)
test/dataset.py:26
Method__init__
(self, data_path_base)
test/dataset_1m.py:30
Method__len__
(self)
test/dataset.py:96
Method__len__
(self)
test/dataset_1m.py:82
Function__main__
Test the model.
core/CVNet_tester.py:33
Method__repr__
(self)
model/resnet.py:62
Functioncolor_norm
Performs per-channel normalization (CHW format).
core/transforms.py:16
Functionconfig_imname
(cfg, i)
test/config_gnd.py:46
Functionconfig_qimname
(cfg, i)
test/config_gnd.py:49
Functionconstruct_loader
Test loader wrapper.
test/test_loader.py:34
Functiondump_cfg
Dumps the config to the output directory.
config.py:124
Functionextract_feat_res_pycls
r""" Extract intermediate features from ResNet
model/base/feature.py:4
Methodextract_featuremap
(self, img)
model/CVNet_Rerank_model.py:67
Methodextract_score_with_featuremap
(self, query_feats, key_feats)
model/CVNet_Rerank_model.py:72
Methodforward
(self, ranks, rerank_dba_final, res_top1000_dba, ranks_trans_1000_pre, x_dba)
modules/reranking/RerankwMDA.py:17
Methodforward
(self, X, Q, ranks)
modules/reranking/MDescAug.py:17
Methodforward
(self, x)
modules/coarse_retrieval/rgem.py:18
Methodforward
(self, x)
modules/coarse_retrieval/gemp.py:16
Methodforward
(self, x)
modules/coarse_retrieval/sgem.py:16
Methodforward
(self, x)
modules/coarse_retrieval/relup.py:15
Methodforward
(self, query_img, key_img)
model/CVNet_Rerank_model.py:52
Methodforward
(self, corr)
model/CVlearner.py:56
Methodforward
(self, x)
model/resnet.py:58
Methodforward
(self, x)
model/resnet.py:82
Methodforward
(self, x)
model/resnet.py:103
Methodforward
(self, x)
model/resnet.py:134
Methodforward
(self, x)
model/resnet.py:158
Methodforward
(self, x)
model/resnet.py:182
Methodforward
(self, x)
model/resnet.py:199
Methodforward
(self, x)
model/resnet.py:214
Methodforward
(self, x_, scale = 3, gemp = True, rgem = True, sgem = True)
model/resnet.py:267
Methodget_coord1d
(coord4d, ksz)
model/base/geometry.py:102
Methodget_distance
(coord1, coord2)
model/base/geometry.py:108
Functionhorizontal_flip
Performs horizontal flip (CHW or HWC format).
core/transforms.py:30
Methodinit_idx4d
(ksz)
model/base/geometry.py:126
Methodinitialize
(cls, img_size)
model/base/geometry.py:12
Methodinterpolate_query_dims
(self, hypercorr, spatial_size=None)
model/CVlearner.py:48
Methodinterpolate_support_dims
(self, hypercorr, spatial_size=None)
model/CVlearner.py:40
Functionlighting
Performs AlexNet-style PCA jitter (CHW format).
core/transforms.py:98
Functionload_cfg
Loads config from specified output directory.
config.py:131
Functionload_cfg_fom_args
Load config from command line arguments and set any specified options.
config.py:137
Functionload_checkpoint
Loads the checkpoint from the given file.
core/checkpoint.py:15
Methodmake_building_block
(in_channel, out_channels, kernel_sizes, query_strides, key_strides, group=4)
model/CVlearner.py:15
Functionrandom_crop
Performs random crop (CHW format).
core/transforms.py:41
Functionrandom_sized_crop
Performs Inception-style cropping (HWC format).
core/transforms.py:77
next →1–100 of 103, ranked by callers