MCPcopy Create free account

hub / github.com/Alioth2000/Hoss-ReID / functions

Functions199 in github.com/Alioth2000/Hoss-ReID

↓ 17 callersMethodstate_dict
(self)
solver/scheduler.py:55
↓ 13 callersFunction_cfg
(url='', **kwargs)
model/backbones/vit_pytorch.py:76
↓ 8 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/backbones/vit_transoss.py:57
↓ 7 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/backbones/vit_pytorch.py:122
↓ 7 callersMethodupdate
(self, val, n=1)
utils/meter.py:16
↓ 6 callersMethodget_imagedata_info
(self, data)
datasets/bases.py:41
↓ 6 callersFunctiontrunc_normal_
r"""Fills the input Tensor with values drawn from a truncated normal distribution. The values are effectively drawn from the normal distributi
model/backbones/vit_pytorch.py:518
↓ 5 callersMethodstep
(self, epoch: int, metric: float = None)
solver/scheduler.py:67
↓ 4 callersMethod__init__
(self, in_features, out_features, s=30.0, m=0.30, easy_margin=False, ls_eps=0.0)
loss/metric_learning.py:85
↓ 4 callersMethod_get_lr
(self, t)
solver/cosine_lr.py:67
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
model/backbones/resnet.py:98
↓ 4 callersFunctiontrunc_normal_
r"""Fills the input Tensor with values drawn from a truncated normal distribution. The values are effectively drawn from the normal distributi
model/backbones/vit_transoss.py:458
↓ 4 callersMethodupdate_groups
(self, values)
solver/scheduler.py:81
↓ 3 callersMethodcompute
(self)
utils/metrics.py:110
↓ 3 callersMethodload_param
(self, trained_path)
model/make_model.py:109
↓ 3 callersFunctionmake_model
(cfg, num_class, camera_num)
model/make_model.py:256
↓ 3 callersMethodreset
(self)
utils/meter.py:10
↓ 3 callersFunctionsetup_logger
(name, save_dir, if_train)
utils/logger.py:5
↓ 3 callersMethodupdate
(self, output)
utils/metrics.py:104
↓ 2 callersMethod__init__
(self, last_stride=2, block=Bottleneck,layers=[3, 4, 6, 3])
model/backbones/resnet.py:85
↓ 2 callersMethod_add_noise
(self, lrs, t)
solver/scheduler.py:87
↓ 2 callersMethod_process_dir
(self, dir_path, relabel=False)
datasets/hoss.py:71
↓ 2 callersFunctioncontrastive_loss
(logits: torch.Tensor)
loss/contrastive_loss.py:7
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
model/backbones/resnet.py:7
↓ 2 callersFunctioncreate_scheduler
(cfg, optimizer)
solver/scheduler_factory.py:7
↓ 2 callersMethodget_image
(self, img_path)
datasets/bases.py:92
↓ 2 callersFunctionmake_dataloader
(cfg)
datasets/make_dataloader.py:54
↓ 2 callersFunctionmake_loss
(cfg, num_classes)
loss/make_loss.py:13
↓ 2 callersFunctionmake_optimizer
(cfg, model, center_criterion)
solver/make_optimizer.py:4
↓ 2 callersFunctionnorm_cdf
(x)
model/backbones/vit_transoss.py:426
↓ 2 callersFunctionnorm_cdf
(x)
model/backbones/vit_pytorch.py:486
↓ 2 callersFunctionread_image
Keep reading image until succeed. This can avoid IOError incurred by heavy IO process.
datasets/bases.py:11
↓ 2 callersMethodreset
(self)
utils/metrics.py:99
↓ 2 callersMethodtrain_with_pair
(self,)
model/make_model.py:193
↓ 2 callersMethodtrain_with_single
(self,)
model/make_model.py:196
↓ 1 callersMethod__fetch_current_node_idxs
(self, final_idxs, length)
datasets/sampler_ddp.py:159
↓ 1 callersMethod__init__
(self, num_classes, cfg)
model/make_model.py:53
↓ 1 callersMethod__init__
(self, in_features, out_features, s=30.0, m=0.50, bias=False)
loss/arcface.py:9
↓ 1 callersMethod__init__
(self, num_classes, epsilon=0.1, use_gpu=True)
loss/softmax_loss.py:16
↓ 1 callersMethod_check_before_run
Check if all files are available before going deeper
datasets/hoss.py:60
↓ 1 callersFunction_get_global_gloo_group
Return a process group based on gloo backend, containing all the ranks The result is cached.
datasets/sampler_ddp.py:14
↓ 1 callersMethod_init_weights
(self, m)
model/backbones/vit_transoss.py:300
↓ 1 callersFunction_no_grad_trunc_normal_
(tensor, mean, std, a, b)
model/backbones/vit_transoss.py:423
↓ 1 callersFunction_no_grad_trunc_normal_
(tensor, mean, std, a, b)
model/backbones/vit_pytorch.py:483
↓ 1 callersFunction_ntuple
(n)
model/backbones/vit_transoss.py:13
↓ 1 callersFunction_ntuple
(n)
model/backbones/vit_pytorch.py:35
↓ 1 callersFunction_pad_to_largest_tensor
Returns: list[int]: size of the tensor, on each rank Tensor: padded tensor that has the max size
datasets/sampler_ddp.py:42
↓ 1 callersMethod_process_dir_train
(self, dir_path, relabel=False)
datasets/pretrain.py:46
↓ 1 callersMethod_process_dir_train
(self, dir_path, relabel=False)
datasets/hoss.py:89
↓ 1 callersFunction_serialize_to_tensor
(data, group)
datasets/sampler_ddp.py:25
↓ 1 callersFunctionall_gather
Run all_gather on arbitrary picklable data (not necessarily tensors). Args: data: any picklable object group: a torch process
datasets/sampler_ddp.py:65
↓ 1 callersFunctionclip_loss
(similarity: torch.Tensor)
loss/contrastive_loss.py:11
↓ 1 callersFunctiondo_inference
(cfg, model, val_loader, num_query)
processor/processor.py:215
↓ 1 callersFunctiondo_train
(cfg, model, center_criterion, train_loader, val_loader, optimizer, optimizer_center, scheduler, loss_fn, num_
processor/processor.py:89
↓ 1 callersFunctiondo_train_pair
(cfg, model, train_loader_pair, optimizer, scheduler, local_rank)
processor/processor.py:13
↓ 1 callersFunctiondrop_path
Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks). This is the same as the DropConnect impl I created for E
model/backbones/vit_transoss.py:25
↓ 1 callersFunctiondrop_path
Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks). This is the same as the DropConnect impl I created for E
model/backbones/vit_pytorch.py:46
↓ 1 callersFunctioneuclidean_dist
Args: x: pytorch Variable, with shape [m, d] y: pytorch Variable, with shape [n, d] Returns: dist: pytorch Variable, with s
loss/triplet_loss.py:16
↓ 1 callersFunctioneuclidean_distance
(qf, gf)
utils/metrics.py:7
↓ 1 callersFunctioneval_func
Evaluation with market1501 metric Key: for each query identity, its gallery images from the same camera view are discarded.
utils/metrics.py:28
↓ 1 callersMethodforward_features
(self, x, camera_id, img_wh)
model/backbones/vit_transoss.py:320
↓ 1 callersMethodforward_features
(self, x, camera_id, view_id)
model/backbones/vit_pytorch.py:376
↓ 1 callersMethodget_epoch_values
(self, epoch: int)
solver/scheduler.py:61
↓ 1 callersMethodget_imagedata_info_pair
(self, data)
datasets/pretrain.py:33
↓ 1 callersMethodget_imagedata_info_pair
(self, data)
datasets/hoss.py:43
↓ 1 callersMethodget_update_values
(self, num_updates: int)
solver/scheduler.py:64
↓ 1 callersFunctionhard_example_mining
For each anchor, find the hardest positive and negative sample. Args: dist_mat: pytorch Variable, pair wise distance between samples, shape
loss/triplet_loss.py:51
↓ 1 callersMethodload_param
(self, trained_path)
model/make_model.py:232
↓ 1 callersFunctionmake_dataloader_pair
(cfg)
datasets/make_dataloader.py:128
↓ 1 callersFunctionmkdir_if_missing
(directory)
utils/iotools.py:14
↓ 1 callersFunctionnormalize
Normalizing to unit length along the specified dimension. Args: x: pytorch Variable Returns: x: pytorch Variable, same shape as in
loss/triplet_loss.py:5
↓ 1 callersMethodprint_dataset_statistics
(self)
datasets/bases.py:56
↓ 1 callersFunctionre_ranking
(probFea, galFea, k1, k2, lambda_value, local_distmat=None, only_local=False)
utils/reranking.py:29
↓ 1 callersMethodreset_parameters
(self)
loss/arcface.py:28
↓ 1 callersMethodreset_parameters
(self)
loss/arcface.py:62
↓ 1 callersMethodreset_parameters
(self)
loss/metric_learning.py:55
↓ 1 callersFunctionresize_pos_embed
(posemb, posemb_new, hight, width)
model/backbones/vit_transoss.py:397
↓ 1 callersFunctionresize_pos_embed
(posemb, posemb_new, hight, width)
model/backbones/vit_pytorch.py:440
↓ 1 callersMethodsample_list
(self)
datasets/sampler_ddp.py:173
↓ 1 callersFunctionsar32bit2RGB
(img)
datasets/bases.py:27
↓ 1 callersFunctionset_seed
(seed)
train.py:16
↓ 1 callersFunctionset_seed
(seed)
train_pair.py:16
↓ 1 callersFunctionshared_random_seed
Returns: int: a random number that is the same across all workers. If workers need a shared RNG, they can use this shared see
datasets/sampler_ddp.py:99
Method__call__
(self, img)
datasets/preprocessing.py:24
Method__call__
(self, bn_feat, targets)
loss/arcface.py:65
Method__call__
(self, bn_feat, targets)
loss/metric_learning.py:58
Method__call__
(self, global_feat, labels, normalize_feature=False)
loss/triplet_loss.py:121
Method__getitem__
(self, index)
datasets/bases.py:106
Method__init__
(self)
utils/meter.py:4
Method__init__
(self, num_query, max_rank=50, feat_norm=True, reranking=False)
utils/metrics.py:92
Method__init__
(self, num_classes, camera_num, cfg, factory, logit_scale_init_value=2.6592)
model/make_model.py:126
Method__init__
(self, drop_prob=None)
model/backbones/vit_transoss.py:48
Method__init__
(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0.)
model/backbones/vit_transoss.py:76
Method__init__
(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., dr
model/backbones/vit_transoss.py:105
Method__init__
(self, img_size=224, patch_size=16, in_chans=3, embed_dim=768)
model/backbones/vit_transoss.py:126
Method__init__
(self, backbone, img_size=224, feature_size=None, in_chans=3, embed_dim=768)
model/backbones/vit_transoss.py:150
Method__init__
(self, img_size=224, patch_size=16, stride_size=20, in_chans=3, embed_dim=768)
model/backbones/vit_transoss.py:190
Method__init__
(self, input_dim=2, embed_dim=768)
model/backbones/vit_transoss.py:228
Method__init__
(self, img_size=224, patch_size=16, stride_size=16, in_chans=3, num_classes=1000, embed_dim=768, depth=12,
model/backbones/vit_transoss.py:241
Method__init__
(self, drop_prob=None)
model/backbones/vit_pytorch.py:68
next →1–100 of 199, ranked by callers