Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/aimagelab/VKD
/ functions
Functions
164 in github.com/aimagelab/VKD
⨍
Functions
164
◇
Types & classes
36
Method
__init__
(self, path: str, uuid: str)
utils/saver.py:16
Method
__init__
(self, label: str)
utils/misc.py:25
Method
__init__
(self, labels: List[str])
utils/misc.py:41
Method
__init__
Constructor class.
utils/conf.py:21
Method
__init__
(self)
model/loss.py:12
Method
__init__
(self, num_classes: int, eps: float = 0.1, reduction: str='sum')
model/loss.py:25
Method
__init__
(self, reduction: str)
model/loss.py:83
Method
__init__
(self, metric: str)
model/loss.py:96
Method
__init__
(self, margin='soft', batch_hard=True, reduction='mean')
model/loss.py:120
Method
__init__
(self, btype: str, pretrained: bool = True, last_stride: int = 1)
model/net.py:30
Method
__init__
(self, num_classes, feat_in: int)
model/net.py:144
Method
__init__
(self)
model/net.py:291
Method
__init__
(self, gate_channel, reduction_ratio=16, num_layers=1)
model/cbam/bam.py:12
Method
__init__
(self, gate_channel, reduction_ratio=16, dilation_conv_num=2, dilation_val=4)
model/cbam/bam.py:33
Method
__init__
(self, inplanes, planes, stride=1, downsample=None)
model/cbam/resnet_bam.py:26
Method
__init__
(self, inplanes, planes, stride=1, downsample=None)
model/cbam/resnet_bam.py:59
Method
__init__
(self, width, height, interpolation=3)
data/misc.py:121
Method
__init__
(self, data_source, p: int, k: int)
data/samplers.py:21
Method
__init__
(self, size: int = 4)
data/temporal_transforms.py:7
Method
__init__
(self, size: int = 4)
data/temporal_transforms.py:30
Method
__init__
(self, size: int = 4)
data/temporal_transforms.py:56
Method
__init__
(self, num_images=4)
data/temporal_transforms.py:75
Method
__init__
(self, dataset, spatial_transform=None, temporal_transform=
data/datasets.py:31
Method
__init__
(self, perform_x2i, perform_x2v, augment_gallery)
data/datasets.py:97
Method
__init__
(self, root='/data/datasets/', min_seq_len=0)
data/veri/data_manager.py:13
Method
__init__
(self, root='/data/datasets/', min_seq_len=0)
data/mars/data_manager.py:14
Method
__init__
(self, root='/data/datasets/', min_seq_len=0)
data/dukevideoreid/data_manager.py:16
Method
__iter__
(self)
data/samplers.py:33
Method
__len__
(self)
data/samplers.py:54
Method
__len__
(self)
data/datasets.py:42
Method
add
(self, value: float)
utils/misc.py:29
Method
backbone_features
(self, x: torch.Tensor)
model/net.py:207
Method
block_parameters
(self, reinitl4: bool, reinitl3: bool)
model/net.py:267
Method
dump_hparams
(self, hparams: dict, metrics: dict)
utils/saver.py:81
Method
forward
target are long in [0, num_classes)!
model/loss.py:41
Method
forward
(self, teacher_logits: torch.Tensor, student_logits: torch.Tensor)
model/loss.py:66
Method
forward
(self, teacher_logits: torch.Tensor, student_logits: torch.Tensor)
model/loss.py:87
Method
forward
(self, teacher_embs: torch.Tensor, student_embs: torch.Tensor)
model/loss.py:102
Method
forward
(self, feat, id=None, pos_mask=None, neg_mask=None, mode='id', dis_func='eu', n_dis=0)
model/loss.py:130
Method
forward
(self, x: torch.Tensor)
model/net.py:126
Method
forward
(self, feats)
model/net.py:157
Method
forward
(self, x: torch.Tensor, return_logits: bool = False)
model/net.py:214
Method
forward
(self, x: torch.Tensor)
model/net.py:294
Method
forward
(self, x)
model/cbam/bam.py:7
Method
forward
(self, in_tensor)
model/cbam/bam.py:26
Method
forward
(self, in_tensor)
model/cbam/bam.py:52
Method
forward
(self, in_tensor)
model/cbam/bam.py:62
Method
forward
(self, x)
model/cbam/resnet_bam.py:37
Method
forward
(self, x)
model/cbam/resnet_bam.py:73
Method
forward
(self, x)
model/cbam/resnet_bam.py:167
Function
get_default_video_loader
()
data/misc.py:54
Method
get_num_cams
(self)
data/datasets.py:48
Method
get_tmp_path
()
utils/conf.py:50
Function
init_worker
(worker_id)
utils/misc.py:8
Function
l2_norm
(x: torch.Tensor)
model/loss.py:6
Method
reset
(self)
utils/misc.py:35
Method
reset
(self)
utils/misc.py:55
Method
sample
(set, n)
data/samplers.py:35
Method
set_teacher_mode
(self, is_teacher: bool)
data/datasets.py:51
Function
str2bool
(v)
utils/misc.py:12
Function
video_loader
(img_paths, image_loader)
data/misc.py:43
Method
weights_init_classifier
(m)
model/net.py:176
Method
weights_init_kaiming
(m)
model/net.py:161
Method
write_image
(self, image: np.ndarray, epoch: int, name: str)
utils/saver.py:65
← previous
101–164 of 164, ranked by callers