MCPcopy Create free account

hub / github.com/JeremyZhao1998/MRT-release / functions

Functions182 in github.com/JeremyZhao1998/MRT-release

⨍ Functions 182◇ Types & classes 40
↓ 1 callersFunctiontrain_one_epoch_standard
(model: torch.nn.Module, criterion: torch.nn.Module,
engine.py:17
↓ 1 callersFunctiontrain_one_epoch_teaching
(student_model: torch.nn.Module, teacher_model: torch.nn.Module,
engine.py:129
↓ 1 callersFunctiontrain_one_epoch_with_mae
(model: torch.nn.Module, criterion: torch.nn.Module,
engine.py:66
↓ 1 callersMethodupdate
(self, predictions)
datasets/coco_eval.py:80
↓ 1 callersMethodwith_pos_embed
(tensor, pos)
models/deformable_transformer.py:232
FunctionPYBIND11_MODULE
models/ops/src/vision.cpp:13
Method__call__
(self, image, annotation=None)
datasets/transforms.py:43
Method__call__
(self, image, annotation=None)
datasets/transforms.py:68
Method__call__
(self, image, annotation)
datasets/transforms.py:132
Method__call__
(self, image, annotation=None)
datasets/transforms.py:172
Method__call__
(self, image, annotation=None)
datasets/transforms.py:193
Method__call__
(self, image, annotation=None)
datasets/transforms.py:241
Method__getitem__
(self, idx)
datasets/coco_style_dataset.py:140
Method__init__
(self, sizes, max_size=1333)
datasets/transforms.py:38
Method__init__
(self, min_size: int, max_size: int)
datasets/transforms.py:49
Method__init__
(self, p=0.5)
datasets/transforms.py:94
Method__init__
(self, transforms, p=0.5)
datasets/transforms.py:111
Method__init__
(self, transforms1, transforms2, p=0.5)
datasets/transforms.py:127
Method__init__
(self, brightness=0, contrast=0, saturation=0, hue=0)
datasets/transforms.py:142
Method__init__
(self, p=0.1)
datasets/transforms.py:153
Method__init__
(self, sigma=None)
datasets/transforms.py:167
Method__init__
(self, p=0.5, scale=(0.02, 0.33), ratio=(0.3, 3.3), value=0, inplace=False)
datasets/transforms.py:182
Method__init__
(self, mean=None, std=None, inplace=False, norm_image=True)
datasets/transforms.py:210
Method__init__
(self, transforms)
datasets/transforms.py:238
Method__init__
(self, coco_gt=None, coco_dt=None, iou_type='bbox')
datasets/coco_eval.py:15
Method__init__
(self, coco_gt)
datasets/coco_eval.py:73
Method__init__
(self, root_dir: str, dataset_name: str, domain: str,
datasets/coco_style_dataset.py:54
Method__init__
(self, root_dir: str, dataset_name: str, domain: str,
datasets/coco_style_dataset.py:127
Method__init__
(self, input_dim, hidden_dim, output_dim, num_layers)
models/deformable_detr.py:27
Method__init__
(self, input_dim, hidden_dim, output_dim, num_layers)
models/deformable_detr.py:43
Method__init__
(self, backbone, position_encoding, transformer,
models/deformable_detr.py:74
Method__init__
(self, d_model=256, d_ffn=1024, dropout=0.1, activation="relu",
models/deformable_transformer.py:214
Method__init__
(self, encoder_layer, num_layers, hda=None)
models/deformable_transformer.py:254
Method__init__
(self, d_model=256, d_ffn=1024, dropout=0.1, activation="relu", n_levels=4
models/deformable_transformer.py:289
Method__init__
(self, decoder_layer, num_layers, return_intermediate=False, hda=None)
models/deformable_transformer.py:339
Method__init__
(self, hidden_dim=256, feedforward_dim=1024, num_heads=8
models/deformable_transformer.py:389
Method__init__
(self)
models/backbones.py:33
Method__init__
(self)
models/backbones.py:46
Method__init__
(self)
models/backbones.py:59
Method__init__
(self, num_pos_feats=256)
models/positional_encoding.py:44
Method__init__
(self, coef_class: float = 2, coef_bbox: float = 5, coef
models/criterion.py:16
Method__init__
(self, d_model=256, n_levels=4, n_heads=8, n_points=4)
models/ops/modules/ms_deform_attn.py:32
Method_discard_empty_labels
(out, annotations)
models/criterion.py:196
Method_summarize
(iou_thr=None, area_rng='all', max_dets=100)
datasets/coco_eval.py:44
Methodapply_dis
(memory, discriminator)
models/deformable_detr.py:233
Methodbackward
(ctx, grad_output)
models/ops/functions/ms_deform_attn_func.py:32
Methodcollate_fn
Function used in dataloader. batch: [sample_{i} for i in range(batch_size)] sample_{i}: (image, annotation)
datasets/coco_style_dataset.py:113
Methodcollate_fn_teaching
Function used in dataloader. batch: [sample_{i} for i in range(batch_size)] sample_{i}: (teacher_image, student_image, an
datasets/coco_style_dataset.py:149
Methodforward
(self, image, annotation=None)
datasets/transforms.py:97
Methodforward
(self, image, annotation=None)
datasets/transforms.py:114
Methodforward
(self, image, annotation=None)
datasets/transforms.py:145
Methodforward
(self, image, annotation=None)
datasets/transforms.py:156
Methodforward
(self, image, annotation=None)
datasets/transforms.py:185
Methodforward
(self, image, annotation=None)
datasets/transforms.py:220
Methodforward
(self, x)
models/deformable_detr.py:19
Methodforward
(self, x)
models/deformable_detr.py:35
Methodforward
(self, x)
models/deformable_detr.py:51
Methodforward
(ctx, x, eta=1.0)
models/deformable_detr.py:59
Methodforward
(self, images, masks, enable_mae=False, mask_ratio=0.8)
models/deformable_detr.py:164
Methodforward
(self, src, pos, reference_points, spatial_shapes, level_start_index, padding_mask=None)
models/deformable_transformer.py:241
Methodforward
(self, src, spatial_shapes, level_start_index, valid_ratios, pos=None, padding_mask=None)
models/deformable_transformer.py:275
Methodforward
(self, tgt, query_pos, reference_points, src, src_spatial_shapes, level_start_index, src_padd
models/deformable_transformer.py:319
Methodforward
(self, tgt, reference_points, src, src_spatial_shapes, src_level_start_index, src_valid_ratios,
models/deformable_transformer.py:349
Methodforward
(self, tgt, reference_points, src, src_spatial_shapes, src_level_start_index, src_valid_ratios,
models/deformable_transformer.py:425
Methodforward
(self, tensor)
models/backbones.py:26
Methodforward
(self, tensor, mask)
models/positional_encoding.py:22
Methodforward
(self, tensor)
models/positional_encoding.py:54
Methodforward
(self, pred_logits, pred_boxes, annotations)
models/criterion.py:26
Methodforward
(self, out, annotations=None, domain_label=None, enable_mae=False)
models/criterion.py:211
Methodforward
(self, query, reference_points, input_flatten, input_spatial_shapes, input_level_start_index,
models/ops/modules/ms_deform_attn.py:68
Methodforward
(ctx, value, value_spatial_shapes, value_level_start_index, sampling_locations, attention_weights, im2col_step
models/ops/functions/ms_deform_attn_func.py:23
Methodget_backbone
(self)
models/backbones.py:36
Methodget_backbone
(self)
models/backbones.py:49
Methodget_backbone
(self)
models/backbones.py:62
Functionget_local_rank
()
utils/distributed_utils.py:47
Functionget_local_size
()
utils/distributed_utils.py:41
Functionms_deform_attn_backward
models/ops/src/ms_deform_attn.h:41
Functionms_deform_attn_cpu_backward
models/ops/src/cpu/ms_deform_attn_cpu.cpp:29
Functionms_deform_attn_cpu_forward
models/ops/src/cpu/ms_deform_attn_cpu.cpp:17
Functionms_deform_attn_forward
models/ops/src/ms_deform_attn.h:20
Functionnew_print
(*args, **kwargs)
utils/distributed_utils.py:17
Functionsave_on_master
(*args, **kwargs)
utils/distributed_utils.py:57
← previous101–182 of 182, ranked by callers