MCPcopy Create free account

hub / github.com/TencentYoutuResearch/PersonReID-YouReID / functions

Functions293 in github.com/TencentYoutuResearch/PersonReID-YouReID

↓ 96 callersMethodget
(self, attr, default=None)
core/config.py:31
↓ 33 callersMethodwrite
(self, msg)
utils/logger.py:53
↓ 13 callersFunctionnormalize
Normalizing to unit length along the specified dimension. Args: x: pytorch Variable Returns: x: pytorch Variable, same shape as in
core/loss.py:7
↓ 9 callersMethod__init__
(self, num_classes, in_feat, scale=64, margin=0.35)
core/loss.py:353
↓ 9 callersMethod__init__
(self, num_channels, num_groups)
core/layers.py:315
↓ 7 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
models/nafs.py:369
↓ 6 callersFunctioninitialize_pretrained_model
(model, num_classes, settings)
models/backbones/senet.py:379
↓ 6 callersFunctiontopk
(sim, target_gallery, target_query, k=[1,5,10], dim=1, print_index=False, reid_sim = None)
models/nafs.py:232
↓ 5 callersFunctionl2norm
L2-normalize columns of X
models/nafs.py:11
↓ 5 callersMethodstep
(self, epoch=None)
utils/epoch_lr.py:23
↓ 4 callersMethod__init__
Parameters ---------- block (nn.Module): Bottleneck class. - For SENet154: SEBottleneck - For SE-ResN
models/backbones/senet.py:209
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1, use_non_local=False)
models/backbones/resnet_ibn_a.py:96
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, groups, reduction, stride=1, downsample_kernel_size=1, downs
models/backbones/senet.py:329
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1, use_non_local=False, use_last_relu=True)
models/backbones/resnet.py:106
↓ 4 callersMethod_postprocess
(self, imgs, least_image_per_class=4)
dataset/formatdata.py:209
↓ 4 callersFunction_resnet
(pretrained, last_stride, block, layers, model_name, model_path='', **kwargs)
models/backbones/resnet.py:183
↓ 4 callersFunctionget_im_names
Get the image names in a dir. Optional to return numpy array, paths.
tools/transform_format.py:52
↓ 4 callersFunctionparse_original_im_name
Get the person id or cam from an image name.
tools/transform_format.py:164
↓ 4 callersMethodpre_process
(self, texts)
models/nafs.py:489
↓ 3 callersMethod__init__
(self, optimizer, milestones, lrs, last_epoch=-1, )
utils/epoch_lr.py:8
↓ 3 callersMethod__init__
(self)
models/nafs.py:481
↓ 3 callersMethodcompute_loss
(self, dist, targets, **kwargs)
core/loss.py:49
↓ 3 callersFunctionconv1x1
1x1 convolution
models/nafs.py:280
↓ 3 callersMethodhead
(self, x)
models/cacenet.py:89
↓ 3 callersMethodheight_shuffle
Shuffle the feature map according to height dimension.
models/nafs.py:441
↓ 3 callersFunctionmay_make_dir
Args: path: a dir, or result of `osp.dirname(osp.abspath(file_path))` Note: `osp.exists('')` returns `False`, while `osp.exists('.')` ret
tools/transform_format.py:20
↓ 2 callersMethod__getitem__
Args: index (int): Index Returns: tuple: (image, target) where target is class_index of the target class.
dataset/formatdata.py:238
↓ 2 callersMethod__init__
(self, root='/data1/home/fufuyu/dataset/', dataname='Occluded_Duke', part='train', loader=rea
dataset/OccludedData.py:16
↓ 2 callersMethod_init_fc
(fc)
models/pcb.py:66
↓ 2 callersMethod_init_fc
(fc)
models/cacenet.py:70
↓ 2 callersFunctionaddBorder
(im, color)
tools/inference_onnx.py:142
↓ 2 callersFunctionaddBorder
tools/inference_onnx.cpp:143
↓ 2 callersMethodbasic_branch
(self, num_conv_out_channels, input_size, pyramid_conv_list,
models/pyramid.py:130
↓ 2 callersMethodclose
(self)
utils/logger.py:69
↓ 2 callersMethodcompute_loss
(self, output, target)
models/mgn.py:125
↓ 2 callersFunctioncompute_similarity
Returns cosine similarity between x1 and x2, computed along dim.
models/nafs.py:22
↓ 2 callersFunctioncompute_weiTexts
Compute weighted text embeddings :param image_embeddings: Tensor with dtype torch.float32, [n_img, n_region, d] :param text_embeddings: T
models/nafs.py:134
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
models/backbones/resnet_ibn_a.py:22
↓ 2 callersMethodevalution
(self, model, test_loader, unique_image)
train/Nafs_Trainer.py:87
↓ 2 callersFunctionextractFeatures
tools/inference_onnx.cpp:73
↓ 2 callersMethodextract_and_eval
(self, test_loader, model)
train/BaseTrainer.py:153
↓ 2 callersFunctionextract_feature
Extract features from images in a target directory :param img_dir: the input image directory :param model_path: path to ReID model :p
tools/inference_onnx.py:40
↓ 2 callersMethodforward
(self, inp)
core/layers.py:323
↓ 2 callersFunctionfunc_attention_MxN
query: (batch, queryL, d) context: (batch, sourceL, d)
models/nafs.py:33
↓ 2 callersFunctiongetNames
tools/inference_onnx.cpp:98
↓ 2 callersMethodget_batch_idxs_dict
(self)
utils/sampler.py:46
↓ 2 callersMethodget_imgs_from_dir
(self, for_query=False)
dataset/OccludedData.py:255
↓ 2 callersMethodget_index_pair_list
Split feature map according to height dimension. :param x: Tensor with dtype torch.float32, [batchsize, num_channels, height, width]
models/nafs.py:424
↓ 2 callersMethodinit_seed
(self)
train/BaseTrainer.py:39
↓ 2 callersFunctionis_image_file
(filename)
utils/iotools.py:59
↓ 2 callersFunctionnormalize
Normalize a N-D numpy array along the specified axis. :param nparry: the array of vectors to be normalized :param order: order of the nor
tools/inference_onnx.py:101
↓ 2 callersMethodparse_im_name
Get the person id or cam from an image name.
dataset/formatdata.py:234
↓ 2 callersFunctionparse_new_im_name
Get the person id or cam from an image name.
tools/transform_format.py:141
↓ 2 callersMethodrecover_shuffle
Recover the feature map to the original order.
models/nafs.py:464
↓ 2 callersFunctionsave_checkpoint
(state, root='../../snapshot/', flag='checkpoint.pth', logger=None)
utils/iotools.py:41
↓ 2 callersFunctionsave_pickle
Create dir and save file.
tools/transform_format.py:35
↓ 1 callersMethod__init__
(self, data_source, batch_size, num_instances, use_tf_sample=False, use_all_sample=False)
utils/sampler.py:19
↓ 1 callersMethod__init__
(self, root='/data1/home/fufuyu/dataset/', dataname='market1501', part='train',
dataset/formatdata.py:80
↓ 1 callersMethod__init__
(self, block, layers, num_classes=1000, last_stride=2, for_test=True, **kwargs)
models/backbones/resnet_ibn_a.py:64
↓ 1 callersMethod__init__
(self, last_stride, block, layers, model_name, use_non_local=False, groups=1
models/backbones/resnet.py:84
↓ 1 callersMethod_init_bn
(bn)
models/baseline.py:83
↓ 1 callersMethod_init_bn
(bn)
models/pcb.py:61
↓ 1 callersMethod_init_bn
(bn)
models/cacenet.py:65
↓ 1 callersMethod_init_fc
(fc)
models/mgn.py:85
↓ 1 callersMethod_init_fc
(fc)
models/baseline.py:88
↓ 1 callersMethod_init_reduction
(reduction)
models/mgn.py:76
↓ 1 callersMethodbuild_dataset
(self)
train/Nafs_Trainer.py:28
↓ 1 callersMethodbuild_dataset
(self, target_w_train=False)
train/BaseTrainer.py:51
↓ 1 callersMethodbuild_imgs
(self)
dataset/formatdata.py:159
↓ 1 callersMethodbuild_joint_embeddings
(self, images_features, text_features)
models/nafs.py:636
↓ 1 callersMethodbuild_opt_and_lr
(self, model)
train/Nafs_Trainer.py:181
↓ 1 callersMethodbuild_opt_and_lr
(self, model)
train/BaseTrainer.py:158
↓ 1 callersMethodbulid_model
(self)
train/Nafs_Trainer.py:55
↓ 1 callersMethodbulid_model
(self, class_num)
train/BaseTrainer.py:109
↓ 1 callersMethodcompute_cmpc_loss
Cross-Modal Projection Classfication loss(CMPC) :param image_embeddings: Tensor with dtype torch.float32 :param text_embeddin
models/nafs.py:671
↓ 1 callersMethodcompute_cmpm_loss
Cross-Modal Projection Matching Loss(CMPM) :param image_embeddings: Tensor with dtype torch.float32 :param text_embeddings: T
models/nafs.py:710
↓ 1 callersMethodcompute_distance
(self, inputs, **kwargs)
core/loss.py:39
↓ 1 callersMethodcompute_distance
(self, inputs, **kwargs)
core/loss.py:95
↓ 1 callersFunctioncompute_topk
(query_global, query, value_bank, gallery_global, gallery_key, gallery_value, gallery_l
models/nafs.py:191
↓ 1 callersMethodcontrastive_loss
(self, i2t_similarites, t2i_similarities, labels)
models/nafs.py:647
↓ 1 callersFunctionconv3x3
3x3 convolution with padding
models/nafs.py:273
↓ 1 callersFunctionconvert_dsbnConstBatch
(model, batch_size=64, constant_batch=32)
core/layers.py:294
↓ 1 callersMethodconvert_to_onnx
(self, model, test_loader)
train/BaseTrainer.py:373
↓ 1 callersMethoddefault_config
(self)
core/config.py:25
↓ 1 callersFunctiondrawRankList
Draw the rank list :param query_name: path of the query image :param gallery_name: path of the gallery image "param output_size: the
tools/inference_onnx.py:135
↓ 1 callersFunctiondrawRankList
tools/inference_onnx.cpp:151
↓ 1 callersFunctioneval_market1501
Evaluation with market1501 metric Key: for each query identity, its gallery images from the same camera view are discarded.
utils/measure.py:4
↓ 1 callersMethodeval_result
(self, **kwargs)
train/BaseTrainer.py:398
↓ 1 callersMethodeval_status
(self, epoch)
train/BaseTrainer.py:386
↓ 1 callersFunctionevaluate_py
(distmat, q_pids, g_pids, q_camids, g_camids, max_rank)
utils/measure.py:65
↓ 1 callersMethodevalution
(self, model, test_loader)
train/BaseTrainer.py:139
↓ 1 callersMethodextract
(self, test_data, model)
train/BaseTrainer.py:316
↓ 1 callersMethodfeatures
(self, x)
models/backbones/senet.py:356
↓ 1 callersMethodflush
(self)
utils/logger.py:63
↓ 1 callersFunctionfocal_equal
consider the confidence g(x) for each fragment as equal sigma_{j} (xi - xj) = sigma_{j} xi - sigma_{j} xj attn: (batch, queryL, sourceL)
models/nafs.py:91
↓ 1 callersFunctionfocal_prob
consider the confidence g(x) for each fragment as the sqrt of their similarity probability to the query fragment sigma_{j} (xi - xj)gj =
models/nafs.py:104
↓ 1 callersFunctiongetTopK
tools/inference_onnx.cpp:120
↓ 1 callersMethodget_imgs
(self)
dataset/OccludedData.py:80
↓ 1 callersMethodget_lr
(self)
utils/epoch_lr.py:20
↓ 1 callersMethodget_lr
(self)
utils/epoch_lr.py:117
next →1–100 of 293, ranked by callers