MCPcopy Create free account

hub / github.com/Sense-GVT/DeCLIP / functions

Functions606 in github.com/Sense-GVT/DeCLIP

⨍ Functions 606◇ Types & classes 121
Method__call__
(self, img)
prototype/data/transforms.py:102
Method__call__
(self, img)
prototype/data/transforms.py:129
Method__call__
(self, img)
prototype/data/transforms.py:142
Method__call__
Args: img (Tensor): Tensor image of size (C, H, W). Returns: Tensor: Image with n_holes of dimension length x
prototype/data/auto_augmentation.py:19
Method__call__
(self, img)
prototype/data/auto_augmentation.py:95
Method__call__
(self, img)
prototype/data/auto_augmentation.py:150
Method__call__
(self, img)
prototype/data/auto_augmentation.py:205
Method__call__
(self, img)
prototype/data/auto_augmentation.py:279
Method__del__
(self)
prototype/data/nvidia_dali_dataloader.py:60
Method__getitem__
Get a single image data: from dataset Arguments: - idx (:obj:`int`): index of image, 0 <= idx < len(self)
prototype/data/datasets/base_dataset.py:44
Method__getitem__
(self, idx)
prototype/data/datasets/clip_dataset.py:180
Method__getitem__
(self, idx)
prototype/data/datasets/imagenet_dataset.py:51
Method__getstate__
(self)
prototype/solver/base_solver.py:14
Method__init__
(self, *args, **kwargs)
linklink/fp16.py:3
Method__init__
(self, config)
prototype/tools/inference.py:25
Method__init__
(self, model)
prototype/tools/convert.py:20
Method__init__
(self, config_file, recover='')
prototype/tools/convert.py:33
Method__init__
(self, optimizer, base_lr, warmup_lr, warmup_steps, last_iter=0)
prototype/lr_scheduler/scheduler.py:68
Method__init__
(self, optimizer, lr_steps, lr_mults, base_lr, warmup_lr, warmup_steps, max_iter, last_iter=0)
prototype/lr_scheduler/scheduler.py:118
Method__init__
(self, optimizer, step_size, decay, base_lr, warmup_lr, warmup_steps, max_iter, last_iter=0)
prototype/lr_scheduler/scheduler.py:180
Method__init__
(self, optimizer, max_iter, min_lr, base_lr, warmup_lr, warmup_steps, last_iter=0)
prototype/lr_scheduler/scheduler.py:229
Method__init__
(self, optimizer, power, max_iter, base_lr, warmup_lr, warmup_steps, last_iter=0)
prototype/lr_scheduler/scheduler.py:283
Method__init__
K: queue size; number of negative keys (default: 65536) m: moco momentum of updating key encoder (default: 0.999) T: softmax
prototype/utils/moco_builder.py:8
Method__init__
(self, encoder)
prototype/utils/simclr_builder.py:7
Method__init__
(self, length=0)
prototype/utils/misc.py:25
Method__init__
(self, model, decay, copy_init=False, use_double=False, inner_T=1, warmup=1)
prototype/utils/ema.py:7
Method__init__
K: queue size; number of negative keys (default: 65536) m: moco momentum of updating key encoder (default: 0.999) T: softmax
prototype/utils/clsa_builder.py:7
Method__init__
(self, module, sync=False)
prototype/utils/dist.py:50
Method__init__
(self, in_dim, hidden_dim=1024, out_dim=1024, num_layers=3, out_bn=True)
prototype/model/slip.py:51
Method__init__
(self,image_encode, text_encode, use_allgather, EDA=True, feature_dim=1024, sim_dim=256, forw
prototype/model/slip.py:210
Method__init__
(self,image_encode, text_encode, use_allgather)
prototype/model/clip.py:54
Method__init__
(self,image_encode, text_encode, use_allgather, nn_size=2**16, nn_topk=1, \ return_dense=Fals
prototype/model/filip.py:28
Method__init__
(self, in_dim, hidden_dim=1024, out_dim=1024, num_layers=3)
prototype/model/defilip.py:51
Method__init__
(self, in_dim, hidden_dim=512, out_dim=1024)
prototype/model/defilip.py:110
Method__init__
(self, in_dim, hidden_dim=1024, out_dim=1024, num_layers=3)
prototype/model/declip.py:34
Method__init__
(self, in_dim, hidden_dim=512, out_dim=1024)
prototype/model/declip.py:93
Method__init__
(self, embed_dim: int, context_length: int, transformer_width:
prototype/model/text_encoder/text_transformer.py:11
Method__init__
(self, d_model: int, n_head: int, attn_mask: torch.Tensor = None, dropout: float = 0.)
prototype/model/text_encoder/base_transformer.py:30
Method__init__
(self, size: int = 2 ** 16)
prototype/model/utils/nnclr_modules/memory_bank.py:40
Method__init__
(self, size: int = 2 ** 16, topk: int = 1)
prototype/model/utils/nnclr_modules/nn_memory_bank.py:38
Method__init__
(self, size: int = 2 ** 16)
prototype/model/utils/nnclr_modules/memory_bank_cuda.py:40
Method__init__
(self, bpe_path: str = default_bpe())
prototype/model/utils/text_utils/simple_tokenizer.py:63
Method__init__
(self, input_resolution: int, patch_size: int, width: int, layers: int, heads: int, embed_dim: int, checkpoint
prototype/model/image_encoder/visual_transformer.py:7
Method__init__
(self, spacial_dim: int, embed_dim: int, num_heads: int, output_dim: int = None)
prototype/model/image_encoder/modified_resnet_modified.py:61
Method__init__
(self, layers, embed_dim, heads, input_resolution=224, width=64, bn_group_size=1,
prototype/model/image_encoder/modified_resnet_modified.py:115
Method__init__
(self, spacial_dim: int, embed_dim: int, num_heads: int, output_dim: int = None)
prototype/model/image_encoder/modified_resnet.py:61
Method__init__
(self, layers, embed_dim, heads, input_resolution=224, width=64, bn_group_size=1,
prototype/model/image_encoder/modified_resnet.py:115
Method__init__
(self, d_model: int, n_head: int, attn_mask: torch.Tensor = None, dropout: float = 0.)
prototype/model/image_encoder/base_transformer.py:30
Method__init__
(self, config_file)
prototype/solver/base_solver.py:7
Method__init__
(self, loader)
prototype/solver/slip_solver.py:30
Method__init__
(self, param, threshold)
prototype/solver/slip_solver.py:64
Method__init__
(self, config_file)
prototype/solver/slip_solver.py:90
Method__init__
(self, loader)
prototype/solver/clip_solver.py:31
Method__init__
(self, param, threshold)
prototype/solver/clip_solver.py:65
Method__init__
(self, config_file)
prototype/solver/clip_solver.py:91
Method__init__
(self, loader)
prototype/solver/defilip_solver.py:31
Method__init__
(self, param, threshold)
prototype/solver/defilip_solver.py:65
Method__init__
(self, config_file)
prototype/solver/defilip_solver.py:91
Method__init__
(self, loader)
prototype/solver/declip_solver.py:31
Method__init__
(self, param, threshold)
prototype/solver/declip_solver.py:65
Method__init__
(self, config_file)
prototype/solver/declip_solver.py:91
Method__init__
(self, loader)
prototype/solver/filip_solver.py:30
Method__init__
(self, param, threshold)
prototype/solver/filip_solver.py:64
Method__init__
(self, config_file)
prototype/solver/filip_solver.py:90
Method__init__
(self, pipeline, batch_size, epoch_size, n
prototype/data/nvidia_dali_dataloader.py:30
Method__init__
(self, loader)
prototype/data/nvidia_dali_dataloader.py:74
Method__init__
params: num_of_times: How many times the augment is repeated
prototype/data/clsa_augmentation.py:189
Method__init__
(self, num_output_channels=1)
prototype/data/transforms.py:14
Method__init__
(self, gamma, gain=1)
prototype/data/transforms.py:24
Method__init__
(self, base_transform)
prototype/data/transforms.py:35
Method__init__
(self, base_transform, augment)
prototype/data/transforms.py:46
Method__init__
Note: RandomResizedCrop should be includeed in stronger_transfrom
prototype/data/transforms.py:57
Method__init__
(self, sigma=[.1, 2.])
prototype/data/transforms.py:85
Method__init__
(self, n_holes=2, length=32, prob=0.5)
prototype/data/transforms.py:97
Method__init__
(self, angles)
prototype/data/transforms.py:126
Method__init__
(self, size)
prototype/data/transforms.py:139
Method__init__
(self, n_holes, length)
prototype/data/auto_augmentation.py:15
Method__init__
(self, fillcolor=(128, 128, 128))
prototype/data/auto_augmentation.py:63
Method__init__
(self, fillcolor=(128, 128, 128))
prototype/data/auto_augmentation.py:117
Method__init__
(self, fillcolor=(128, 128, 128))
prototype/data/auto_augmentation.py:172
Method__init__
(self, p1, operation1, magnitude_idx1, p2, operation2, magnitude_idx2, fillcolor=(128, 128, 128))
prototype/data/auto_augmentation.py:214
Method__init__
(self, dataset, world_size=None, rank=None, round_up=True)
prototype/data/sampler.py:9
Method__init__
(self, dataset, total_iter, batch_size, world_size=None, rank=None, last_iter=0)
prototype/data/sampler.py:58
Method__init__
(self, dataset, total_iter, batch_size, world_size=None, rank=None, last_iter=0)
prototype/data/sampler.py:110
Method__init__
(self, dataset, total_iter, batch_size, last_iter=0, shuffle=True)
prototype/data/sampler.py:172
Method__init__
(self, key_metric, defect_classes, recall_thres,
prototype/data/metrics/custom_evaluator.py:27
Method__init__
(self, topk=[1, 5])
prototype/data/metrics/imagenet_evaluator.py:25
Method__init__
(self, topk=(1,))
prototype/data/metrics/multiclass_evaluator.py:20
Method__init__
(self, metric={}, cmp_key='')
prototype/data/metrics/base_evaluator.py:8
Method__init__
(self)
prototype/data/metrics/base_evaluator.py:21
Method__init__
(self, batch_size, num_threads, device_id, crop, colorjitter=None, dali_cpu=False)
prototype/data/pipelines/imagenet_pipeline.py:7
Method__init__
(self, **kwargs)
prototype/data/pipelines/imagenet_pipeline_v2.py:16
Method__init__
(self, data_root, data_list, sampler, crop, colorjitter=None)
prototype/data/pipelines/imagenet_pipeline_v2.py:38
Method__init__
(self, data_root, data_list, sampler, crop, size)
prototype/data/pipelines/imagenet_pipeline_v2.py:87
Method__init__
(self, root_dir, meta_file, transform=None,
prototype/data/datasets/base_dataset.py:15
Method__init__
(self, root_dir, meta_file, transform=None, read_from='mc', evaluator=None, image_reader_type
prototype/data/datasets/clip_dataset.py:315
Method__init__
(self, root_dir, meta_file, transform=None, read_from='mc', evaluator=None, image_reader_typ
prototype/data/datasets/imagenet_dataset.py:22
Method__init__
(self, smooth_ratio, num_classes)
prototype/loss_functions/loss.py:8
Method__init__
(self)
prototype/loss_functions/loss.py:26
Method__init__
(self, batch_size, temperature=0.1, use_cosine_similarity=True, alpha_weight=0.75)
prototype/loss_functions/nt_xent_ConVIRT.py:6
← previousnext →301–400 of 606, ranked by callers