Functions — github.com/CDTrans/CDTrans

hub / github.com/CDTrans/CDTrans / functions

Functions356 in github.com/CDTrans/CDTrans

⨍ Functions 356◇ Types & classes 70

Methodload_un_param

(self, trained_path)

model/backbones/vit_pytorch.py:345

Methodload_un_param

(self, trained_path)

model/backbones/vit_pytorch.py:468

Methodload_un_param

(self, trained_path)

model/backbones/vit_pytorch.py:638

Methodload_un_param

(self, trained_path)

model/backbones/vit_pytorch_uda.py:549

Methodload_un_param

(self, trained_path)

model/backbones/resnet.py:150

Functionloss_func

(score, feat, target, target_cam)

loss/make_loss.py:24

Functionmake_dataloader_Pseudo

(cfg)

datasets/make_dataloader.py:172

Methodno_weight_decay

(self)

model/backbones/vit_pytorch.py:298

Methodno_weight_decay

(self)

model/backbones/vit_pytorch.py:419

Methodno_weight_decay

(self)

model/backbones/vit_pytorch.py:578

Methodno_weight_decay

(self)

model/backbones/vit_pytorch_uda.py:456

Functionparse

(x)

model/backbones/vit_pytorch.py:35

Functionparse

(x)

model/backbones/vit_pytorch_uda.py:35

Functionposterize

(img, bits_to_keep, **__)

datasets/autoaugment.py:154

Methodprint_dataset_statistics

(self)

datasets/bases.py:46

Methodprint_dataset_statistics

(self, train, query, gallery)

datasets/bases.py:55

Functionrand_augment_transform

Create a RandAugment transform :param config_str: String defining configuration of random augmentation. Consists of multiple sections separat

datasets/autoaugment.py:625

Methodrandom_init

(self)

model/backbones/resnet.py:167

Methodreset_classifier

(self, num_classes, global_pool='')

model/backbones/vit_pytorch.py:304

Methodreset_classifier

(self, num_classes, global_pool='')

model/backbones/vit_pytorch.py:425

Methodreset_classifier

(self, num_classes, global_pool='')

model/backbones/vit_pytorch.py:584

Methodreset_classifier

(self, num_classes, global_pool='')

model/backbones/vit_pytorch_uda.py:462

Functionresnet152_ibn_a

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

model/backbones/resnet_ibn_a.py:190

Functionrotate

(img, degrees, **kwargs)

datasets/autoaugment.py:91

Functionse_resnet152_ibn_a

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

model/backbones/se_resnet_ibn_a.py:219

Functionse_resnet50_ibn_a

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

model/backbones/se_resnet_ibn_a.py:197

Functionsharpness

(img, factor, **__)

datasets/autoaugment.py:172

Functionshear_x

(img, factor, **kwargs)

datasets/autoaugment.py:59

Functionshear_y

(img, factor, **kwargs)

datasets/autoaugment.py:64

Functionsolarize

(img, thresh, **__)

datasets/autoaugment.py:135

Functionsolarize_add

(img, add, thresh=128, **__)

datasets/autoaugment.py:139

Functionsource_target_train_collate_fn

# collate_fn这个函数的输入就是一个list，list的长度是一个batch size，list中的每个元素都是__getitem__得到的结果

datasets/make_dataloader.py:44

Methodstep_update

(self, num_updates: int, metric: float = None)

solver/scheduler.py:70

Functiontrain_collate_fn

# collate_fn这个函数的输入就是一个list，list的长度是一个batch size，list中的每个元素都是__getitem__得到的结果

datasets/make_dataloader.py:27

Functiontranslate_x_abs

(img, pixels, **kwargs)

datasets/autoaugment.py:81

Functiontranslate_x_rel

(img, pct, **kwargs)

datasets/autoaugment.py:69

Functiontranslate_y_abs

(img, pixels, **kwargs)

datasets/autoaugment.py:86

Functiontranslate_y_rel

(img, pct, **kwargs)

datasets/autoaugment.py:75

Functionuda_vit_base_patch16_224_TransReID

(img_size=(256, 128), stride_size=16, drop_path_rate=0.1, local_feature=False,aie_xishu=1.5,use_cross=False, u

model/backbones/vit_pytorch_uda.py:603

Functionuda_vit_small_patch16_224_TransReID

(img_size=(256, 128), stride_size=16, drop_path_rate=0.1, drop_rate=0.0, attn_drop_rate=0.0, local_feature=Fal

model/backbones/vit_pytorch_uda.py:595

Functionval_collate_fn

(batch)

datasets/make_dataloader.py:38

Functionvit_base_patch16_224_TransReID

(img_size=(256, 128), stride_size=16, drop_path_rate=0.1, local_feature=False,aie_xishu=1.5, **kwargs)

model/backbones/vit_pytorch.py:691

Functionvit_small_patch16_224_TransReID

(img_size=(256, 128), stride_size=16, drop_path_rate=0.1, drop_rate=0.0, attn_drop_rate=0.0, local_feature=Fal

model/backbones/vit_pytorch.py:683

Functionweights_init_classifier

(m)

model/make_model.py:27

Functionweights_init_kaiming

(m)

model/make_model.py:12

Functionwrite_json

(obj, fpath)

utils/iotools.py:36

← previous301–356 of 356, ranked by callers