MCPcopy Create free account

hub / github.com/CDTrans/CDTrans / functions

Functions356 in github.com/CDTrans/CDTrans

↓ 33 callersMethodstate_dict
(self)
solver/scheduler.py:51
↓ 25 callersMethodget_imagedata_info
(self, data)
datasets/bases.py:30
↓ 13 callersFunction_cfg
(url='', **kwargs)
model/backbones/vit_pytorch.py:76
↓ 13 callersFunction_cfg
(url='', **kwargs)
model/backbones/vit_pytorch_uda.py:76
↓ 10 callersMethodcompute
(self)
utils/metrics.py:208
↓ 10 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:733
↓ 10 callersMethodupdate
(self, output)
utils/metrics.py:202
↓ 9 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/backbones/vit_pytorch.py:122
↓ 9 callersMethodreset
(self)
utils/meter.py:10
↓ 8 callersMethod__init__
(self, in_features, hidden_features=None, out_features=None, act_layer=nn.GELU, drop=0.)
model/backbones/vit_pytorch_uda.py:121
↓ 8 callersMethodupdate
(self, val, n=1)
utils/meter.py:16
↓ 7 callersFunction_check_args_tf
(kwargs)
datasets/autoaugment.py:53
↓ 6 callersMethod__init__
(self, num_query, max_rank=50, feat_norm=True, reranking=False, reranking_track=False)
utils/metrics.py:311
↓ 6 callersMethodcompute
(self)
utils/metrics.py:286
↓ 6 callersMethodstep
(self, epoch: int, metric: float = None)
solver/scheduler.py:63
↓ 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_uda.py:645
↓ 5 callersFunction_randomly_negate
With 50% prob, negate the value
datasets/autoaugment.py:176
↓ 5 callersMethodupdate
(self, output)
utils/metrics.py:278
↓ 5 callersMethodupdate
(self, output)
utils/metrics.py:326
↓ 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:64
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
model/backbones/resnet_ibn_a.py:107
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
model/backbones/se_resnet_ibn_a.py:136
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
model/backbones/resnet.py:99
↓ 4 callersFunctioncount_target_usage
(logger, idxs, label_memory1, label_memory2, img_num1, img_num2, source_idxs=None)
processor/processor_uda.py:140
↓ 4 callersMethodload_param_finetune
(self, model_path)
model/make_model.py:157
↓ 4 callersFunctionmake_dataloader
(cfg)
datasets/make_dataloader.py:72
↓ 4 callersMethodreset
(self)
utils/metrics.py:197
↓ 4 callersMethodupdate_groups
(self, values)
solver/scheduler.py:77
↓ 3 callersMethod__init__
(self, planes)
model/backbones/se_resnet_ibn_a.py:14
↓ 3 callersMethod_process_dir
(self, dir_path, relabel=False)
datasets/ourapi.py:51
↓ 3 callersMethodload_param
(self, trained_path)
model/make_model.py:137
↓ 3 callersMethodload_state_dict
(self, state_dict: Dict[str, Any])
solver/scheduler.py:54
↓ 3 callersMethodload_un_param
(self, trained_path)
model/make_model.py:147
↓ 3 callersFunctionre_ranking
(probFea, galFea, k1, k2, lambda_value, local_distmat=None, only_local=False)
utils/reranking.py:29
↓ 2 callersMethodEntropy
(self, input_)
utils/metrics.py:146
↓ 2 callersMethod__init__
(self, num_classes, cfg)
model/make_model.py:36
↓ 2 callersMethod__init__
(self, planes)
model/backbones/resnet_ibn_a.py:19
↓ 2 callersMethod__init__
(self, last_stride=2, block=Bottleneck, frozen_stages=-1,layers=[3, 4, 6, 3])
model/backbones/resnet.py:85
↓ 2 callersMethod_add_noise
(self, lrs, t)
solver/scheduler.py:83
↓ 2 callersMethod_process_dir
(self, list_path, dir_path)
datasets/office_home.py:72
↓ 2 callersMethod_process_dir
(self, list_path, dir_path)
datasets/visda.py:78
↓ 2 callersMethod_process_dir
(self, list_path, dir_path)
datasets/domainnet.py:78
↓ 2 callersMethod_process_dir
(self, list_path, dir_path)
datasets/office.py:78
↓ 2 callersFunctionauto_augment_policy
(name="original")
datasets/autoaugment.py:481
↓ 2 callersFunctionconv3x3
(in_planes, out_planes, stride=1)
model/backbones/se_resnet_ibn_a.py:9
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
model/backbones/resnet.py:7
↓ 2 callersFunctioncosine_similarity
(qf, gf)
utils/metrics.py:27
↓ 2 callersFunctioneuclidean_distance
(qf, gf)
utils/metrics.py:11
↓ 2 callersFunctionmake_model
(cfg, num_class, camera_num, view_num)
model/make_model.py:404
↓ 2 callersFunctionnorm_cdf
(x)
model/backbones/vit_pytorch.py:702
↓ 2 callersFunctionnorm_cdf
(x)
model/backbones/vit_pytorch_uda.py:614
↓ 2 callersMethodprint_dataset_statistics
(self, train, test)
datasets/visda.py:64
↓ 2 callersFunctionre_ranking_numpy
(probFea, galFea, k1, k2, lambda_value, local_distmat=None, only_local=False)
utils/reranking.py:102
↓ 2 callersFunctionread_image
Keep reading image until succeed. This can avoid IOError incurred by heavy IO process.
datasets/bases.py:9
↓ 2 callersMethodreset
(self)
utils/metrics.py:271
↓ 2 callersFunctionsetup_logger
(name, save_dir, if_train)
utils/logger.py:6
↓ 1 callersMethod__fetch_current_node_idxs
(self, final_idxs, length)
datasets/sampler_ddp.py:159
↓ 1 callersMethod__init__
(self, in_features, out_features, s=30.0, m=0.50, bias=False)
loss/arcface.py:9
↓ 1 callersMethod_apply_basic
(self, img, mixing_weights, m)
datasets/autoaugment.py:736
↓ 1 callersMethod_apply_blended
(self, img, mixing_weights, m)
datasets/autoaugment.py:720
↓ 1 callersMethod_calc_blended_weights
(self, ws, m)
datasets/autoaugment.py:710
↓ 1 callersMethod_check_before_run
Check if all files are available before going deeper
datasets/ourapi.py:42
↓ 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:12
↓ 1 callersFunction_interpolation
(kwargs)
datasets/autoaugment.py:45
↓ 1 callersMethod_load_parameter
(self, pretrain_choice, model_path)
model/make_model.py:216
↓ 1 callersFunction_no_grad_trunc_normal_
(tensor, mean, std, a, b)
model/backbones/vit_pytorch.py:699
↓ 1 callersFunction_no_grad_trunc_normal_
(tensor, mean, std, a, b)
model/backbones/vit_pytorch_uda.py:611
↓ 1 callersFunction_ntuple
(n)
model/backbones/vit_pytorch.py:34
↓ 1 callersFunction_ntuple
(n)
model/backbones/vit_pytorch_uda.py:34
↓ 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:38
↓ 1 callersFunction_posterize_level_to_arg
(level, _hparams)
datasets/autoaugment.py:223
↓ 1 callersFunction_select_rand_weights
(weight_idx=0, transforms=None)
datasets/autoaugment.py:594
↓ 1 callersFunction_serialize_to_tensor
(data, group)
datasets/sampler_ddp.py:22
↓ 1 callersFunction_solarize_level_to_arg
(level, _hparams)
datasets/autoaugment.py:244
↓ 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:64
↓ 1 callersFunctionaugmix_ops
(magnitude=10, hparams=None, transforms=None)
datasets/autoaugment.py:689
↓ 1 callersFunctionauto_augment_policy_original
(hparams)
datasets/autoaugment.py:415
↓ 1 callersFunctionauto_augment_policy_originalr
(hparams)
datasets/autoaugment.py:448
↓ 1 callersFunctionauto_augment_policy_v0
(hparams)
datasets/autoaugment.py:348
↓ 1 callersFunctionauto_augment_policy_v0r
(hparams)
datasets/autoaugment.py:381
↓ 1 callersFunctioncompute_knn_idx
(logger, model, train_loader1, train_loader2, feat_memory1, feat_memory2, label_memory1, label_memory2, img_nu
processor/processor_uda.py:120
↓ 1 callersFunctioncreate_scheduler
(cfg, optimizer)
solver/schedular_factory.py:7
↓ 1 callersFunctiondistill_loss
(teacher_output, student_out)
processor/processor_uda.py:332
↓ 1 callersFunctiondo_inference
(cfg, model, val_loader, num_query)
processor/processor.py:198
↓ 1 callersFunctiondo_inference_uda
(cfg, model, val_loader, num_query)
processor/processor_uda.py:494
↓ 1 callersFunctiondo_train_pretrain
(cfg, model, center_criterion, train_loader, val_loader,
processor/processor.py:16
↓ 1 callersFunctiondo_train_uda
(cfg, model, center_criterion, train_loader, train_loader1
processor/processor_uda.py:232
↓ 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:45
↓ 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_uda.py:45
↓ 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 callersFunctioneval_func
Evaluation with market1501 metric Key: for each query identity, its gallery images from the same camera view are discarded.
utils/metrics.py:69
↓ 1 callersMethodforward_features
(self, x)
model/backbones/vit_pytorch.py:308
↓ 1 callersMethodforward_features
(self, x)
model/backbones/vit_pytorch.py:429
↓ 1 callersMethodforward_features
(self, x, camera_id, view_id)
model/backbones/vit_pytorch.py:588
↓ 1 callersMethodforward_features
(self, x, x2, camera_id, view_id, domain_norm=False, cls_embed_specific=False,inference_target_only=False)
model/backbones/vit_pytorch_uda.py:466
↓ 1 callersFunctiongenerate_new_dataset
(cfg, logger, label_memory2, s_dataset, t_dataset, knnidx, target_knnidx, target_pseudo_label, label_memory1,
processor/processor_uda.py:168
↓ 1 callersMethodget_epoch_values
(self, epoch: int)
solver/scheduler.py:57
↓ 1 callersMethodget_update_values
(self, num_updates: int)
solver/scheduler.py:60
↓ 1 callersMethodguassian_kernel
(self, source, target, kernel_mul=2.0, kernel_num=5, fix_sigma=None)
loss/mmd_loss.py:12
next →1–100 of 356, ranked by callers