MCPcopy Create free account

hub / github.com/SJTU-LuHe/TransVOD / functions

Functions415 in github.com/SJTU-LuHe/TransVOD

⨍ Functions 415◇ Types & classes 92
↓ 43 callersFunctionprint
(*args, **kwargs)
util/misc.py:374
↓ 24 callersMethodto
(self, device, non_blocking=False)
util/misc.py:344
↓ 24 callersMethodupdate
(self, **kwargs)
util/misc.py:199
↓ 12 callersMethodmax
(self)
util/misc.py:108
↓ 8 callersMethod__init__
(self, encoder_layer, num_layers)
models/deformable_transformer_multi.py:326
↓ 8 callersMethodadd_meter
(self, name, meter)
util/misc.py:226
↓ 8 callersMethodexpand
(tensor, length)
models/segmentation.py:106
↓ 7 callersFunctioninverse_sigmoid
(x, eps=1e-5)
util/misc.py:513
↓ 6 callersFunctionis_dist_avail_and_initialized
()
util/misc.py:382
↓ 6 callersFunctionis_dist_avail_and_initialized
()
util/misc_multi.py:403
↓ 5 callersMethod__init__
(self, d_model=256, nhead=8, num_encoder_layers=6, num_decoder_layers=6, dim_feedforward=102
models/deformable_transformer_single.py:24
↓ 5 callersMethoddecompose
(self)
util/misc.py:360
↓ 5 callersFunctionnested_tensor_from_tensor_list
(tensor_list: List[Tensor])
util/misc.py:319
↓ 5 callersFunctionprint
(*args, **kwargs)
util/misc_multi.py:395
↓ 4 callersMethod__init__
(self, detr, freeze_detr=False)
models/segmentation.py:31
↓ 4 callersFunction_get_activation_fn
Return an activation function given a string
models/deformable_transformer_multi.py:626
↓ 4 callersFunction_get_clones
(module, N)
models/deformable_transformer_multi.py:622
↓ 4 callersMethodget_image
(self, path)
datasets/torchvision_datasets/coco.py:55
↓ 4 callersFunctioninterpolate
Equivalent to nn.functional.interpolate, but with support for empty batch sizes. This will eventually be supported natively by PyTorch, and
util/misc.py:483
↓ 4 callersMethodlog_every
(self, iterable, print_freq, header=None)
util/misc.py:229
↓ 4 callersFunctionmatch_name_keywords
(n, name_keywords)
main.py:195
↓ 4 callersMethodprepare
(self, predictions, iou_type)
datasets/coco_eval.py:77
↓ 4 callersMethodrecord_stream
(self, *args, **kwargs)
util/misc.py:355
↓ 4 callersFunctionsigmoid_focal_loss
Loss used in RetinaNet for dense detection: https://arxiv.org/abs/1708.02002. Args: inputs: A float tensor of arbitrary shape.
models/segmentation.py:196
↓ 4 callersMethodsynchronize_between_processes
(self)
util/misc.py:222
↓ 3 callersMethod__init__
(self, backbone, position_embedding)
models/backbone.py:114
↓ 3 callersFunction_get_activation_fn
Return an activation function given a string
models/deformable_transformer_single.py:421
↓ 3 callersMethod_get_src_permutation_idx
(self, indices)
models/deformable_detr_single.py:310
↓ 3 callersMethod_get_src_permutation_idx
(self, indices)
models/deformable_detr_multi.py:316
↓ 3 callersFunction_run
(command)
util/misc.py:287
↓ 3 callersFunction_run
(command)
util/misc_multi.py:287
↓ 3 callersFunctionbuild_dataset
(image_set, args)
datasets/__init__.py:28
↓ 3 callersFunctioncrop
(clip, target, region)
datasets/transforms_multi.py:51
↓ 3 callersFunctioncrop
(image, target, region)
datasets/transforms_single.py:24
↓ 3 callersFunctionget_local_rank
()
util/misc.py:408
↓ 3 callersFunctionget_local_size
()
util/misc.py:402
↓ 3 callersMethodget_loss
(self, loss, outputs, targets, indices, num_boxes, **kwargs)
models/deformable_detr_single.py:322
↓ 3 callersMethodget_loss
(self, loss, outputs, targets, indices, num_boxes, **kwargs)
models/deformable_detr_multi.py:328
↓ 3 callersFunctionget_world_size
()
util/misc.py:390
↓ 3 callersFunctionget_world_size
()
util/misc_multi.py:411
↓ 3 callersFunctionnested_tensor_from_tensor_list
(tensor_list: List[Tensor], split=True)
util/misc_multi.py:318
↓ 3 callersMethodwith_pos_embed
(tensor, pos)
models/deformable_transformer_multi.py:295
↓ 2 callersMethod__init__
(self, input_dim, hidden_dim, output_dim, num_layers)
models/deformable_detr_single.py:432
↓ 2 callersMethod__init__
(self, input_dim, hidden_dim, output_dim, num_layers)
models/deformable_detr_multi.py:438
↓ 2 callersFunction_get_clones
(module, N)
models/deformable_detr_single.py:31
↓ 2 callersFunction_get_clones
(module, N)
models/deformable_transformer_single.py:417
↓ 2 callersFunction_get_clones
(module, N)
models/deformable_detr_multi.py:31
↓ 2 callersMethodaccumulate
(self)
datasets/coco_eval.py:68
↓ 2 callersFunctionall_gather
Run all_gather on arbitrary picklable data (not necessarily tensors) Args: data: any picklable object Returns: list[
util/misc.py:124
↓ 2 callersMethodbackward
(ctx, grad_output)
models/ops/functions/ms_deform_attn_func.py:32
↓ 2 callersFunctionbox_cxcywh_to_xyxy
(x)
util/box_ops.py:17
↓ 2 callersFunctionbox_xyxy_to_cxcywh
(x)
util/box_ops.py:24
↓ 2 callersFunctionbuild_backbone
(args)
models/backbone.py:133
↓ 2 callersFunctionbuild_deforamble_transformer
(args)
models/deformable_transformer_multi.py:637
↓ 2 callersFunctionbuild_matcher
(args)
models/matcher.py:103
↓ 2 callersFunctionbuild_model
(args)
models/__init__.py:15
↓ 2 callersFunctionconvert_to_xywh
(boxes)
datasets/coco_eval.py:173
↓ 2 callersFunctiondice_loss
Compute the DICE loss, similar to generalized IOU for masks Args: inputs: A float tensor of arbitrary shape. The
models/segmentation.py:178
↓ 2 callersFunctionget_coco_api_from_dataset
(dataset)
datasets/__init__.py:18
↓ 2 callersFunctionis_main_process
()
util/misc.py:414
↓ 2 callersFunctionmake_coco_transforms
(image_set)
datasets/coco.py:125
↓ 2 callersFunctionms_deform_attn_core_pytorch
(value, value_spatial_shapes, sampling_locations, attention_weights)
models/ops/functions/ms_deform_attn_func.py:41
↓ 2 callersMethodnext
(self)
datasets/data_prefetcher_multi.py:58
↓ 2 callersMethodpreload
(self)
datasets/data_prefetcher_multi.py:29
↓ 2 callersMethodpreload
(self)
datasets/data_prefetcher_single.py:23
↓ 2 callersMethodsummarize
(self)
datasets/coco_eval.py:72
↓ 2 callersMethodsummarize
(self)
datasets/panoptic_eval.py:45
↓ 2 callersMethodsynchronize_between_processes
(self)
datasets/coco_eval.py:63
↓ 2 callersMethodsynchronize_between_processes
(self)
datasets/panoptic_eval.py:38
↓ 2 callersMethodto
(self, device, non_blocking=False)
util/misc_multi.py:365
↓ 2 callersFunctionto_cuda
(samples, ref_samples, targets, device)
datasets/data_prefetcher_multi.py:14
↓ 2 callersFunctionto_cuda
(samples, targets, device)
datasets/data_prefetcher_single.py:9
↓ 2 callersMethodupdate
(self, predictions)
datasets/coco_eval.py:44
↓ 2 callersMethodupdate
(self, predictions)
datasets/panoptic_eval.py:31
↓ 2 callersMethodupdate
(self, value, n=1)
util/misc.py:75
↓ 2 callersMethodupdate
(self, value, n=1)
util/misc_multi.py:75
↓ 2 callersMethodwith_pos_embed
(tensor, pos)
models/deformable_transformer_multi.py:361
↓ 2 callersMethodwith_pos_embed
(tensor, pos)
models/deformable_transformer_multi.py:486
↓ 2 callersMethodwith_pos_embed
(tensor, pos)
models/deformable_transformer_single.py:287
↓ 2 callersMethodwith_pos_embed
(tensor, pos)
models/deformable_transformer_single.py:340
↓ 1 callersMethod__getitem__
(self, idx)
datasets/coco.py:33
↓ 1 callersMethod__init__
(self, num_pos_feats=64, temperature=10000, normalize=False, scale=None)
models/position_encoding.py:25
↓ 1 callersFunction_check_size_scale_factor
(dim, size, scale_factor)
util/misc.py:33
↓ 1 callersFunction_check_size_scale_factor
(dim, size, scale_factor)
util/misc_multi.py:33
↓ 1 callersMethod_get_tgt_permutation_idx
(self, indices)
models/deformable_detr_single.py:316
↓ 1 callersMethod_get_tgt_permutation_idx
(self, indices)
models/deformable_detr_multi.py:322
↓ 1 callersFunction_is_power_of_2
(n)
models/ops/modules/ms_deform_attn.py:24
↓ 1 callersFunction_max_by_axis
(the_list)
util/misc.py:310
↓ 1 callersFunction_max_by_axis
(the_list)
util/misc_multi.py:310
↓ 1 callersFunction_output_size
(dim, input, size, scale_factor)
util/misc.py:44
↓ 1 callersFunction_output_size
(dim, input, size, scale_factor)
util/misc_multi.py:44
↓ 1 callersMethod_reset_parameters
(self)
models/deformable_transformer_multi.py:73
↓ 1 callersMethod_reset_parameters
(self)
models/deformable_transformer_single.py:58
↓ 1 callersMethod_reset_parameters
(self)
models/ops/modules/ms_deform_attn.py:62
↓ 1 callersMethod_set_aux_loss
(self, outputs_class, outputs_coord)
models/deformable_detr_single.py:190
↓ 1 callersFunctionaccuracy
Computes the precision@k for the specified values of k
util/misc.py:465
↓ 1 callersFunctionaccuracy
Computes the precision@k for the specified values of k
util/misc_multi.py:487
↓ 1 callersFunctionbbox_overlaps
(bboxes1, bboxes2, mode='iou', eps=1e-6)
datasets/transforms_multi.py:18
↓ 1 callersFunctionbenchmark
()
benchmark.py:46
↓ 1 callersFunctionbox_iou
(boxes1, boxes2)
util/box_ops.py:32
next →1–100 of 415, ranked by callers