MCPcopy Create free account

hub / github.com/dddzg/up-detr / functions

Functions193 in github.com/dddzg/up-detr

↓ 29 callersFunctionprint
(*args, **kwargs)
util/misc.py:381
↓ 16 callersMethodto
(self, device)
util/misc.py:300
↓ 9 callersMethodmax
(self)
util/misc.py:78
↓ 9 callersMethodupdate
(self, **kwargs)
util/misc.py:169
↓ 6 callersFunctionnested_tensor_from_tensor_list
(tensor_list: List[Tensor])
util/misc.py:318
↓ 6 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
models/transformer.py:216
↓ 4 callersMethod__init__
(self, d_model=512, nhead=8, num_encoder_layers=6, num_decoder_layers=6, dim_feedforward=2048
models/transformer.py:26
↓ 4 callersMethod__init__
(self, detr, freeze_detr=False)
models/segmentation.py:31
↓ 4 callersFunction_expand
(tensor, length: int)
models/segmentation.py:72
↓ 4 callersMethod_get_src_permutation_idx
(self, indices)
models/detr.py:220
↓ 4 callersMethoddecompose
(self)
util/misc.py:311
↓ 4 callersFunctionis_dist_avail_and_initialized
()
util/misc.py:389
↓ 3 callersMethod__init__
(self, backbone, position_embedding)
models/backbone.py:114
↓ 3 callersFunction_run
(command)
util/misc.py:257
↓ 3 callersMethodadd_meter
(self, name, meter)
util/misc.py:196
↓ 3 callersFunctionbuild_dataset
(image_set, args)
datasets/__init__.py:25
↓ 3 callersFunctioncrop
(image, target, region)
datasets/transforms.py:22
↓ 3 callersFunctionget_world_size
()
util/misc.py:397
↓ 3 callersFunctioninterpolate
Equivalent to nn.functional.interpolate, but with support for empty batch sizes. This will eventually be supported natively by PyTorch, and t
util/misc.py:462
↓ 2 callersMethod__init__
(self, input_dim, hidden_dim, output_dim, num_layers)
models/detr.py:321
↓ 2 callersFunction_get_activation_fn
Return an activation function given a string
models/transformer.py:296
↓ 2 callersFunction_get_clones
(module, N)
models/transformer.py:279
↓ 2 callersMethod_set_aux_loss
(self, outputs_class, outputs_coord, outputs_feature, backbone_output)
models/updetr.py:131
↓ 2 callersMethod_set_aux_loss
(self, outputs_class, outputs_coord)
models/detr.py:81
↓ 2 callersFunctionall_gather
Run all_gather on arbitrary picklable data (not necessarily tensors) Args: data: any picklable object Returns: list[data]
util/misc.py:94
↓ 2 callersFunctionbox_cxcywh_to_xyxy
(x)
util/box_ops.py:15
↓ 2 callersFunctionbuild_model
(args)
models/__init__.py:18
↓ 2 callersFunctionconvert_to_xywh
(boxes)
datasets/coco_eval.py:171
↓ 2 callersFunctionevaluate
(model, criterion, postprocessors, data_loader, base_ds, device, output_dir)
engine.py:75
↓ 2 callersFunctionget_coco_api_from_dataset
(dataset)
datasets/__init__.py:15
↓ 2 callersMethodget_loss
(self, loss, outputs, targets, indices, num_boxes, **kwargs)
models/detr.py:232
↓ 2 callersMethodlog_every
(self, iterable, print_freq, header=None)
util/misc.py:199
↓ 2 callersFunctionmake_coco_transforms
(image_set)
datasets/coco.py:121
↓ 2 callersMethodprepare
(self, predictions, iou_type)
datasets/coco_eval.py:75
↓ 2 callersMethodsynchronize_between_processes
(self)
util/misc.py:192
↓ 2 callersFunctiontrain_one_epoch
(model: torch.nn.Module, criterion: torch.nn.Module, data_loader: Iterable, optimizer: tor
engine.py:23
↓ 2 callersMethodupdate
(self, value, n=1)
util/misc.py:45
↓ 2 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
models/transformer.py:153
↓ 1 callersMethod__init__
(self, num_pos_feats=64, temperature=10000, normalize=False, scale=None)
models/position_encoding.py:23
↓ 1 callersMethod_get_tgt_permutation_idx
(self, indices)
models/detr.py:226
↓ 1 callersFunction_max_by_axis
(the_list)
util/misc.py:286
↓ 1 callersFunction_onnx_nested_tensor_from_tensor_list
(tensor_list: List[Tensor])
util/misc.py:346
↓ 1 callersMethod_reset_parameters
(self)
models/transformer.py:48
↓ 1 callersMethodaccumulate
(self)
datasets/coco_eval.py:66
↓ 1 callersFunctionaccuracy
Computes the precision@k for the specified values of k
util/misc.py:444
↓ 1 callersFunctionbox_iou
(boxes1, boxes2)
util/box_ops.py:30
↓ 1 callersFunctionbox_xyxy_to_cxcywh
(x)
util/box_ops.py:22
↓ 1 callersFunctionbuild_backbone
(args)
models/backbone.py:134
↓ 1 callersFunctionbuild_matcher
(args)
models/matcher.py:91
↓ 1 callersFunctionbuild_position_encoding
(args)
models/position_encoding.py:85
↓ 1 callersFunctionbuild_selfdet
(image_set, args)
datasets/selfdet.py:148
↓ 1 callersFunctionbuild_transformer
(args)
models/transformer.py:283
↓ 1 callersFunctionconvert_coco_poly_to_mask
(segmentations, height, width)
datasets/coco.py:39
↓ 1 callersFunctioncreate_common_coco_eval
(coco_eval, img_ids, eval_imgs)
datasets/coco_eval.py:198
↓ 1 callersFunctiondice_loss
Compute the DICE loss, similar to generalized IOU for masks Args: inputs: A float tensor of arbitrary shape. The pred
models/segmentation.py:180
↓ 1 callersFunctionevaluate
Run per image evaluation on given images and store results (a list of dict) in self.evalImgs :return: None
datasets/coco_eval.py:214
↓ 1 callersMethodforward_post
(self, src, src_mask: Optional[Tensor] = None,
models/transformer.py:156
↓ 1 callersMethodforward_post
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask:
models/transformer.py:219
↓ 1 callersMethodforward_pre
(self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: O
models/transformer.py:171
↓ 1 callersMethodforward_pre
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: Op
models/transformer.py:242
↓ 1 callersFunctiongeneralized_box_iou
Generalized IoU from https://giou.stanford.edu/ The boxes should be in [x0, y0, x1, y1] format Returns a [N, M] pairwise matrix, where
util/box_ops.py:46
↓ 1 callersFunctionget_args_parser
()
detr_main.py:26
↓ 1 callersFunctionget_args_parser
()
main.py:25
↓ 1 callersFunctionget_query_transforms
(image_set)
datasets/selfdet.py:122
↓ 1 callersFunctionget_random_patch_from_img
:param img: original image :param min_pixel: min pixels of the query patch :return: query_patch,x,y,w,h
datasets/selfdet.py:18
↓ 1 callersFunctionget_rank
()
util/misc.py:403
↓ 1 callersFunctionget_size
(image_size, size, max_size=None)
datasets/transforms.py:105
↓ 1 callersFunctionget_size_with_aspect_ratio
(image_size, size, max_size=None)
datasets/transforms.py:85
↓ 1 callersFunctionhflip
(image, target)
datasets/transforms.py:65
↓ 1 callersFunctionis_main_process
()
util/misc.py:409
↓ 1 callersFunctionmain
(args)
detr_main.py:112
↓ 1 callersFunctionmain
(args)
main.py:104
↓ 1 callersFunctionmake_self_det_transforms
(image_set)
datasets/selfdet.py:83
↓ 1 callersFunctionmasks_to_boxes
Compute the bounding boxes around the provided masks The masks should be in format [N, H, W] where N is the number of masks, (H, W) are the spati
util/box_ops.py:70
↓ 1 callersFunctionmerge
(img_ids, eval_imgs)
datasets/coco_eval.py:176
↓ 1 callersFunctionpad
(image, target, padding)
datasets/transforms.py:141
↓ 1 callersMethodprepare_for_coco_detection
(self, predictions)
datasets/coco_eval.py:85
↓ 1 callersMethodprepare_for_coco_keypoint
(self, predictions)
datasets/coco_eval.py:144
↓ 1 callersMethodprepare_for_coco_segmentation
(self, predictions)
datasets/coco_eval.py:109
↓ 1 callersMethodreset_parameters
(self)
models/position_encoding.py:67
↓ 1 callersFunctionresize
(image, target, size, max_size=None)
datasets/transforms.py:82
↓ 1 callersFunctionsetup_for_distributed
This function disables printing when not in master process
util/misc.py:374
↓ 1 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:198
↓ 1 callersMethodsummarize
(self)
datasets/coco_eval.py:70
↓ 1 callersMethodsummarize
(self)
datasets/panoptic_eval.py:43
↓ 1 callersMethodsynchronize_between_processes
(self)
datasets/coco_eval.py:61
↓ 1 callersMethodsynchronize_between_processes
(self)
datasets/panoptic_eval.py:36
↓ 1 callersMethodupdate
(self, predictions)
datasets/coco_eval.py:42
↓ 1 callersMethodupdate
(self, predictions)
datasets/panoptic_eval.py:29
Method__call__
(self, img, target)
datasets/transforms.py:158
Method__call__
(self, img: PIL.Image.Image, target: dict)
datasets/transforms.py:168
Method__call__
(self, img, target)
datasets/transforms.py:179
Method__call__
(self, img, target)
datasets/transforms.py:191
Method__call__
(self, img, target=None)
datasets/transforms.py:203
Method__call__
(self, img, target)
datasets/transforms.py:212
Method__call__
(self, img, target)
datasets/transforms.py:228
Method__call__
(self, img, target)
datasets/transforms.py:235
Method__call__
(self, img, target)
datasets/transforms.py:244
Method__call__
(self, image, target=None)
datasets/transforms.py:253
Method__call__
(self, image, target)
datasets/transforms.py:271
next →1–100 of 193, ranked by callers