MCPcopy Create free account

hub / github.com/Kitware/COAT / functions

Functions154 in github.com/Kitware/COAT

↓ 27 callersFunctionprint
(*args, **kwargs)
utils/utils.py:278
↓ 10 callersMethodupdate
(self, **kwargs)
utils/utils.py:93
↓ 7 callersMethod__init__
(self, p=0.1)
models/transformer.py:228
↓ 4 callersFunctionmkdir
(path)
utils/utils.py:350
↓ 4 callersMethodpostprocess_boxes
Similar to RoIHeads.postprocess_detections, but can handle embeddings and implement First Classification Score (FCS).
models/coat.py:483
↓ 4 callersFunctionrun_kuhn_munkres
(x_y_values)
utils/km.py:130
↓ 4 callersFunctionto_device
(images, targets, device)
engine.py:18
↓ 3 callersMethod__init__
(self, cfg)
models/coat.py:24
↓ 3 callersFunction_compute_iou
(a, b)
eval_func.py:14
↓ 3 callersFunctionbuild_dataset
(dataset_name, root, transforms, split, verbose=True)
datasets/build.py:54
↓ 3 callersFunctionconv1x1
1x1 convolution
models/transformer.py:14
↓ 3 callersMethodhstripe
utils/mask.py:94
↓ 3 callersFunctionis_dist_avail_and_initialized
()
utils/utils.py:286
↓ 3 callersMethodsquare
utils/mask.py:120
↓ 3 callersMethodvstripe
utils/mask.py:107
↓ 3 callersMethodxpatch_hstripe
utils/mask.py:176
↓ 3 callersMethodxpatch_square
utils/mask.py:204
↓ 3 callersMethodxpatch_vstripe
utils/mask.py:190
↓ 2 callersMethod_flatten_fc_input
(self, x)
models/coat.py:637
↓ 2 callersMethod_get_cam_id
(self, img_name)
datasets/prw.py:20
↓ 2 callersFunctionbuild_transforms
(cfg, is_train)
utils/transforms.py:134
↓ 2 callersMethodchange_exception
(self, nodes, change)
utils/km.py:101
↓ 2 callersFunctionevaluate_performance
Args: use_gt (bool, optional): Whether to use GT as detection results to verify the upper bound of person
engine.py:85
↓ 2 callersMethodget_boxes
Get boxes from proposals.
models/coat.py:466
↓ 2 callersFunctionget_world_size
()
utils/utils.py:294
↓ 2 callersFunctionresume_from_ckpt
(ckpt_path, model, optimizer=None, lr_scheduler=None)
utils/utils.py:416
↓ 2 callersMethodset_false
(self, nodes)
utils/km.py:97
↓ 2 callersMethodupdate
(self, value, n=1)
utils/utils.py:38
↓ 1 callersMethod__init__
(self, resnet)
models/resnet.py:11
↓ 1 callersMethod_load_annotations
For each image, load its annotation that is a dictionary with the following keys: img_name (str): image name img_path
datasets/base.py:20
↓ 1 callersMethod_load_queries
(self)
datasets/cuhk_sysu.py:16
↓ 1 callersMethod_load_queries
(self)
datasets/prw.py:24
↓ 1 callersMethod_load_split_img_names
Load the image names for the specific split.
datasets/cuhk_sysu.py:35
↓ 1 callersMethod_load_split_img_names
Load the image names for the specific split.
datasets/prw.py:53
↓ 1 callersMethod_split_embedding_dim
(self)
models/coat.py:643
↓ 1 callersMethodadd_meter
(self, name, meter)
utils/utils.py:117
↓ 1 callersMethodbackward
(ctx, grad_outputs)
loss/oim.py:18
↓ 1 callersFunctionbuild_resnet
(name="resnet50", pretrained=True)
models/resnet.py:45
↓ 1 callersFunctionbuild_test_loader
(cfg)
datasets/build.py:84
↓ 1 callersFunctionbuild_train_loader
(cfg)
datasets/build.py:70
↓ 1 callersFunctioncreate_small_table
Create a small table using the keys of small_dict as headers. This is only suitable for small dictionaries. Args: small_dict (di
utils/utils.py:383
↓ 1 callersFunctiondetection_losses
( box_cls_scores_1st, box_regs_1st, box_labels_1st, box_reg_targets_1st, box_cls_scores_2n
models/coat.py:691
↓ 1 callersMethoddfs
(self, i)
utils/km.py:79
↓ 1 callersFunctioneval_detection
gallery_det (list of ndarray): n_det x [x1, y1, x2, y2, score] per image det_thresh (float): filter out gallery detections whose scores below
eval_func.py:24
↓ 1 callersMethodforward
(self, x)
models/resnet.py:27
↓ 1 callersMethodget_connect_result
(self)
utils/km.py:106
↓ 1 callersFunctionget_default_cfg
Get a copy of the default config.
defaults.py:215
↓ 1 callersMethodget_max_value_result
(self)
utils/km.py:122
↓ 1 callersFunctionget_rank
()
utils/utils.py:300
↓ 1 callersMethodinference
(self, images, targets=None, query_img_as_gallery=False)
models/coat.py:122
↓ 1 callersFunctionis_main_process
()
utils/utils.py:306
↓ 1 callersMethodkm
(self)
utils/km.py:66
↓ 1 callersMethodlog_every
(self, iterable, print_freq, header=None)
utils/utils.py:120
↓ 1 callersFunctionmain
(args)
train.py:22
↓ 1 callersFunctionmixup_data
(images, alpha=0.8)
utils/transforms.py:12
↓ 1 callersFunctionoim
(inputs, targets, lut, cq, header, momentum=0.5)
loss/oim.py:37
↓ 1 callersFunctionprint_statistics
Print dataset statistics.
datasets/build.py:11
↓ 1 callersFunctionreduce_dict
Reduce the values in the dictionary from all processes so that all processes have the averaged results. Returns a dict with the same fields a
utils/utils.py:242
↓ 1 callersFunctionsave_on_master
(*args, **kwargs)
utils/utils.py:310
↓ 1 callersMethodset_matrix
(self, x_y_values)
utils/km.py:32
↓ 1 callersFunctionset_random_seed
(seed)
utils/utils.py:428
↓ 1 callersFunctionsetup_for_distributed
This function disables printing when not in master process
utils/utils.py:270
↓ 1 callersFunctiontrain_one_epoch
(cfg, model, optimizer, data_loader, device, epoch, tfboard, softmax_criterion_s2, softmax_criterion_s3)
engine.py:26
↓ 1 callersFunctionwarmup_lr_scheduler
(optimizer, warmup_iters, warmup_factor)
utils/utils.py:406
↓ 1 callersFunctionwrite_json
(obj, fpath)
utils/utils.py:364
Method__call__
(self, image, target)
utils/transforms.py:37
Method__call__
(self, image, target)
utils/transforms.py:47
Method__call__
Args: img (Tensor): Tensor image of size (C, H, W). Returns: Tensor: Image with n_holes of dimension length x
utils/transforms.py:67
Method__call__
(self, img, target)
utils/transforms.py:102
Method__call__
(self, image, target)
utils/transforms.py:128
Method__call__
(self, features, mask_box)
utils/mask.py:12
Method__call__
(self, features)
utils/mask.py:27
Method__call__
(self, features)
utils/mask.py:60
Method__call__
(self, features)
utils/mask.py:141
Method__call__
(self, features)
utils/mask.py:225
Method__call__
(self, features)
utils/mask.py:250
Method__call__
(self, features)
utils/mask.py:276
Method__call__
(self, features)
utils/mask.py:303
Method__del__
(self)
utils/km.py:29
Method__getattr__
(self, attr)
utils/utils.py:100
Method__getitem__
(self, index)
datasets/base.py:31
Method__init__
(self, transforms)
utils/transforms.py:34
Method__init__
(self, prob=0.5)
utils/transforms.py:44
Method__init__
(self, n_holes=2, length=100)
utils/transforms.py:63
Method__init__
(self, EPSILON=0.5, mean=[0.485, 0.456, 0.406])
utils/transforms.py:98
Method__init__
(self, window_size=20, fmt=None)
utils/utils.py:30
Method__init__
(self, delimiter="\t")
utils/utils.py:89
Method__init__
(self)
utils/mask.py:9
Method__init__
(self, shift=5, group=2, begin=1)
utils/mask.py:22
Method__init__
(self, shape='stripe', mask_size=2, mode='random_direct')
utils/mask.py:55
Method__init__
(self, shape='stripe', mask_size=2, mode='random_direct')
utils/mask.py:136
Method__init__
(self, mask_size=2)
utils/mask.py:222
Method__init__
(self, mask_size=2)
utils/mask.py:247
Method__init__
(self, mask_size=2)
utils/mask.py:273
Method__init__
(self, shift=5, group=2)
utils/mask.py:299
Method__init__
(self, id, exception=0, match=None, visit=False)
utils/km.py:11
Method__init__
(self)
utils/km.py:19
Method__init__
(self, root, transforms, split)
datasets/base.py:13
Method__init__
(self, root, transforms, split)
datasets/cuhk_sysu.py:11
Method__init__
(self, root, transforms, split)
datasets/prw.py:15
next →1–100 of 154, ranked by callers