Functions — github.com/aimagelab/VKD

hub / github.com/aimagelab/VKD / functions

Functions164 in github.com/aimagelab/VKD

⨍ Functions 164◇ Types & classes 36

↓ 9 callersMethoddump_metric_tb

(self, value: float, epoch: int, m_type: str, m_desc: str)

utils/saver.py:72

↓ 8 callersMethodeval

(self, saver: Saver, iteration: int, verbose: bool, do_tb: bool = True)

tools/eval.py:148

↓ 5 callersMethod__init__

(self, temp: float, reduction: str)

model/loss.py:59

↓ 5 callersMethodget_device

()

utils/conf.py:86

↓ 4 callersMethod__call__

(self, *args, **kwargs)

model/loss.py:77

↓ 4 callersMethod_make_layer

(self, block, planes, blocks, stride=1)

model/cbam/resnet_bam.py:150

↓ 4 callersMethodadd_default_args

(parser: ArgumentParser)

utils/conf.py:29

↓ 4 callersMethodcompute_distance_matrix

(x: torch.Tensor, y: torch.Tensor, metric='cosine')

tools/eval.py:67

↓ 4 callersFunctionevaluate

(distmat, q_pids, g_pids, q_camids, g_camids, return_wrong_matches: bool = False)

data/eval_metrics.py:40

↓ 4 callersMethodextract_features

Extract features for the entire dataloader. It returns also pids and cams.

tools/eval.py:120

↓ 4 callersFunctionget_dataloaders

(dataset_name: str, root: str, device: torch.device, args: Namespace)

data/datasets.py:111

↓ 4 callersMethodload_net

(path: str, chk_name: str, dataset_name: str)

utils/saver.py:46

↓ 4 callersFunctionsave_img

(img, attn, dest_path)

tools/save_heatmaps.py:74

↓ 4 callersMethodsuppress_random

(seed: int = SEED, set_determinism: bool = False)

utils/conf.py:74

↓ 4 callersMethodtb_cmc

(saver: Saver, cmc_scores, it, method)

tools/eval.py:143

↓ 3 callersMethod__call__

(self, *args, **kwargs)

model/net.py:280

↓ 3 callersMethod__init__

(self, backbone_type: str, num_classes: int, pretrained: bool)

model/net.py:189

↓ 3 callersMethod_extract_1stfeame

(self, names, meta_data, home_dir=None, relabel=False)

data/mars/data_manager.py:150

↓ 3 callersMethod_get_paths

(self, fpath)

data/dukevideoreid/data_manager.py:72

↓ 3 callersMethod_process_data

(self, names, meta_data, home_dir=None, relabel=False, min_seq_len=0)

data/mars/data_manager.py:112

↓ 3 callersMethod_process_video

(self, train_paths, re_label: bool)

data/dukevideoreid/data_manager.py:76

↓ 3 callersMethodget_net

(btype: str, pretrained: bool)

model/net.py:53

↓ 3 callersMethodsave_net

(self, net: torch.nn.Module, name: str = 'weights', overwrite: bool = False)

utils/saver.py:75

↓ 2 callersFunctionResidualNet

(network_type, depth, num_classes, zero_init_residual)

model/cbam/resnet_bam.py:188

↓ 2 callersMethod__init__

(self, gate_channel)

model/cbam/bam.py:57

↓ 2 callersMethod__init__

(self, block, layers, network_type, num_classes, zero_init_residual=False)

model/cbam/resnet_bam.py:97

↓ 2 callersMethod_get_names

(self, fpath)

data/veri/data_manager.py:67

↓ 2 callersMethod_get_names

(self, fpath)

data/mars/data_manager.py:104

↓ 2 callersMethodadd

(self, values: List[float])

utils/misc.py:48

↓ 2 callersMethodcdist

Returns euclidean distance between a and b Args: a (2D Tensor): A batch of vectors shaped (B1, D) b (2D Te

model/loss.py:193

↓ 2 callersFunctionconv3x3

3x3 convolution with padding

model/cbam/resnet_bam.py:17

↓ 2 callersMethodevaluate

(self, net: nn.Module)

tools/train_distill.py:108

↓ 2 callersFunctionextract_grad_cam

(net: TriNet, inputs: torch.Tensor, device: torch.device, hook: Hook)

tools/save_heatmaps.py:53

↓ 2 callersMethodget_hostname_config

()

utils/conf.py:54

↓ 2 callersFunctionget_model

(args: Namespace, num_pids: int)

model/net.py:284

↓ 2 callersMethodget_num_pids

(self)

data/datasets.py:45

↓ 2 callersMethodget_padding

(padding)

data/misc.py:127

↓ 2 callersFunctionget_transforms

(train_mode: bool, args: Namespace)

data/misc.py:59

↓ 2 callersFunctionpil_loader

(path)

data/misc.py:11

↓ 2 callersMethodreinit_layers

(self, reinitl4: bool, reinitl3: bool)

model/net.py:245

↓ 2 callersMethodreset

(self)

tools/save_heatmaps.py:29

↓ 2 callersMethodteacher_mode

(self)

model/net.py:233

↓ 2 callersMethodwrite_logs

(self, model: torch.nn.Module, params: dict)

utils/saver.py:59

↓ 1 callersMethod_check_before_run

Check if all files are available before going deeper

data/mars/data_manager.py:89

↓ 1 callersMethod_get_log_path

(self)

utils/conf.py:63

↓ 1 callersMethod_get_nas_path

(self)

utils/conf.py:68

↓ 1 callersMethod_process_gallery

(self)

data/veri/data_manager.py:75

↓ 1 callersMethod_process_query

(self, query_names, gallery)

data/veri/data_manager.py:118

↓ 1 callersMethod_process_query_image

(self, query_names)

data/veri/data_manager.py:136

↓ 1 callersMethod_process_train

(self, train_names)

data/veri/data_manager.py:92

↓ 1 callersFunctionaccimage_loader

(path)

data/misc.py:18

↓ 1 callersMethodbackbone_features

(self, x: torch.Tensor)

model/net.py:121

↓ 1 callersFunctioncompute_ap_cmc

(index, good_index, junk_index, return_is_correct: bool = False)

data/eval_metrics.py:5

↓ 1 callersFunctioncompute_pids_and_pids_dict

(data_source)

data/samplers.py:10

↓ 1 callersMethodevaluate_i2i

(self, verbose: bool = True)

tools/eval.py:108

↓ 1 callersMethodevaluate_i2v

(self, verbose: bool = True)

tools/eval.py:85

↓ 1 callersMethodevaluate_v2i

(self, verbose: bool = True)

tools/eval.py:96

↓ 1 callersMethodevaluate_v2v

(self, verbose: bool = True)

tools/eval.py:74

↓ 1 callersFunctionget_default_image_loader

()

data/misc.py:27

↓ 1 callersMethodget_denset_backbone_layers

(pretrained: bool)

model/net.py:113

↓ 1 callersMethodget_hparams

(self)

model/net.py:204

↓ 1 callersMethodget_mobilenet_backbone_layers

(pretrained: bool, last_stride: int)

model/net.py:75

↓ 1 callersMethodget_output_shape

(self)

model/net.py:118

↓ 1 callersMethodget_resnet_backbone_layers

(btype: str, pretrained: bool, last_stride: int)

model/net.py:83

↓ 1 callersFunctionimage_loader

(path)

data/misc.py:35

↓ 1 callersMethodlabels_to_one_hot

(self, labels)

model/loss.py:36

↓ 1 callersMethodload_logs

(self)

utils/saver.py:38

↓ 1 callersFunctionmain

()

tools/eval.py:189

↓ 1 callersFunctionmain

()

tools/save_heatmaps.py:90

↓ 1 callersFunctionmain

()

tools/train_v2v.py:48

↓ 1 callersFunctionparse

(conf: Conf)

tools/eval.py:173

↓ 1 callersFunctionparse

(conf: Conf)

tools/save_heatmaps.py:33

↓ 1 callersFunctionparse

(conf: Conf)

tools/train_distill.py:34

↓ 1 callersFunctionparse

(conf: Conf)

tools/train_v2v.py:19

↓ 1 callersFunctionresnet101_bam

Constructs a ResNet-50 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet

model/cbam/resnet_bam.py:224

↓ 1 callersFunctionresnet50_bam

Constructs a ResNet-50 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet

model/cbam/resnet_bam.py:210

↓ 1 callersMethodstudent_mode

(self)

model/net.py:239

Method__call__

(self, module, _, ten_out)

tools/save_heatmaps.py:26

Method__call__

(self, epoch: int)

tools/train_distill.py:30

Method__call__

(self, teacher_net: TriNet, student_net: TriNet)

tools/train_distill.py:113

(self, x: torch.Tensor, y: torch.Tensor = None)

model/loss.py:15

Method__call__

(self, *args, **kwargs)

model/loss.py:53

Method__call__

(self, *args, **kwargs)

model/loss.py:90

Method__call__

(self, *args, **kwargs)

model/loss.py:114

Method__call__

(self, *args, **kwargs)

model/loss.py:207

Method__call__

(self, *args, **kwargs)

model/net.py:133

Method__call__

(self, *args, **kwargs)

model/net.py:183

Method__call__

(self, *args, **kwargs)

model/net.py:297

Method__call__

(self, img: Image.Image)

data/misc.py:136

Method__call__

(self, frame_indices, tracklet_index)

data/temporal_transforms.py:10

Method__call__

(self, frame_indices, tracklet_index)

data/temporal_transforms.py:33

Method__call__

(self, candidate, tracklet_index)

data/temporal_transforms.py:59

Method__call__

(self, frame_indices, tracklet_index)

data/temporal_transforms.py:78

Method__getitem__

Args: index (int): Index Returns: tuple: (clip, pid, camid) where pid is identity of the clip.

data/datasets.py:54

Method__init__

(self, model: torch.nn.Module, query_loader: DataLoader, gallery_loader: DataLoader, queryimg

tools/eval.py:19

Method__init__

(self)

tools/save_heatmaps.py:23

Method__init__

(self, initial_lr: float, decay: float, min: float = 1e-5)

tools/train_distill.py:26

Method__init__

(self, train_loader: DataLoader, query_loader: DataLoader, gallery_loader: DataLoader, queryi

tools/train_distill.py:81

next →1–100 of 164, ranked by callers