MCPcopy Create free account

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

Functions193 in github.com/dddzg/up-detr

Method__call__
(self, x)
datasets/selfdet.py:116
Method__call__
(self, image, target)
datasets/coco.py:60
Method__getattr__
(self, attr)
util/misc.py:176
Method__getitem__
(self, idx)
datasets/coco_panoptic.py:40
Method__getitem__
(self, item)
datasets/selfdet.py:55
Method__getitem__
(self, idx)
datasets/coco.py:29
Method__init__
(self, size)
datasets/transforms.py:155
Method__init__
(self, min_size: int, max_size: int)
datasets/transforms.py:164
Method__init__
(self, size)
datasets/transforms.py:176
Method__init__
(self, p=0.5)
datasets/transforms.py:188
Method__init__
(self, sizes, max_size=None)
datasets/transforms.py:198
Method__init__
(self, max_pad)
datasets/transforms.py:209
Method__init__
(self, transforms1, transforms2, p=0.5)
datasets/transforms.py:223
Method__init__
(self, *args, **kwargs)
datasets/transforms.py:241
Method__init__
(self, mean, std)
datasets/transforms.py:249
Method__init__
(self, transforms)
datasets/transforms.py:268
Method__init__
(self, coco_gt, iou_types)
datasets/coco_eval.py:29
Method__init__
(self, ann_file, ann_folder, output_dir="panoptic_eval")
datasets/panoptic_eval.py:20
Method__init__
(self, img_folder, ann_folder, ann_file, transforms=None, return_masks=True)
datasets/coco_panoptic.py:22
Method__init__
(self, root, detection_transform, query_transform, num_patches=10)
datasets/selfdet.py:39
Method__init__
(self, sigma=[.1, 2.])
datasets/selfdet.py:113
Method__init__
(self, img_folder, ann_file, transforms, return_masks)
datasets/coco.py:24
Method__init__
(self, return_masks=False)
datasets/coco.py:57
Method__init__
Creates the matcher Params: cost_class: This is the relative weight of the classification error in the matching cost
models/matcher.py:26
Method__init__
(self, encoder_layer, num_layers, norm=None)
models/transformer.py:71
Method__init__
(self, decoder_layer, num_layers, norm=None, return_intermediate=False)
models/transformer.py:95
Method__init__
(self, d_model, nhead, dim_feedforward=2048, dropout=0.1, activation="relu", normalize_before
models/transformer.py:136
Method__init__
(self, d_model, nhead, dim_feedforward=2048, dropout=0.1, activation="relu", normalize_before
models/transformer.py:196
Method__init__
Initializes the model. Parameters: backbone: torch module of the backbone to be used. See backbone.py transformer: to
models/updetr.py:19
Method__init__
(self, dim, fpn_dims, context_dim)
models/segmentation.py:82
Method__init__
(self, query_dim, hidden_dim, num_heads, dropout=0.0, bias=True)
models/segmentation.py:150
Method__init__
(self, threshold=0.5)
models/segmentation.py:227
Method__init__
Parameters: is_thing_map: This is a whose keys are the class ids, and the values a boolean indicating whether
models/segmentation.py:253
Method__init__
Initializes the model. Parameters: backbone: torch module of the backbone to be used. See backbone.py transformer: to
models/detr.py:26
Method__init__
Create the criterion. Parameters: num_classes: number of object categories, omitting the special no-object category m
models/detr.py:96
Method__init__
(self, n)
models/backbone.py:34
Method__init__
(self, backbone: nn.Module, train_backbone: bool, num_channels: int, return_interm_layers: bool)
models/backbone.py:66
Method__init__
(self, name: str, train_backbone: bool, return_interm_layers: bool,
models/backbone.py:97
Method__init__
(self, num_pos_feats=256)
models/position_encoding.py:61
Method__init__
(self, window_size=20, fmt=None)
util/misc.py:37
Method__init__
(self, delimiter="\t")
util/misc.py:165
Method__init__
(self, tensors, mask: Optional[Tensor])
util/misc.py:296
Method__len__
(self)
datasets/coco_panoptic.py:76
Method__len__
(self)
datasets/selfdet.py:52
Method__repr__
(self)
datasets/transforms.py:276
Method__repr__
(self)
util/misc.py:314
Method__str__
(self)
util/misc.py:85
Method__str__
(self)
util/misc.py:184
Method_load_from_state_dict
(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys
models/backbone.py:41
Methodavg
(self)
util/misc.py:69
Functionbuild
(image_set, args)
datasets/coco_panoptic.py:86
Functionbuild
(image_set, args)
datasets/coco.py:153
Functioncollate_fn
(batch)
util/misc.py:274
Methodforward
Performs the matching Params: outputs: This is a dict that contains at least these entries: "pred_logits": Tens
models/matcher.py:41
Methodforward
(self, src, mask, query_embed, pos_embed, decoder_mask=None)
models/transformer.py:53
Methodforward
(self, src, mask: Optional[Tensor] = None, src_key_padding_mask: Optional[Tens
models/transformer.py:77
Methodforward
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: Optional[T
models/transformer.py:102
Methodforward
(self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: Optional[
models/transformer.py:185
Methodforward
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: Optional[T
models/transformer.py:265
Methodforward
 The forward expects a NestedTensor samples and patches Tensor. samples consists of: - samples.tensor: batched images, of s
models/updetr.py:55
Methodforward
(self, samples: NestedTensor)
models/segmentation.py:43
Methodforward
(self, x: Tensor, bbox_mask: Tensor, fpns: List[Tensor])
models/segmentation.py:109
Methodforward
(self, q, k, mask: Optional[Tensor] = None)
models/segmentation.py:165
Methodforward
(self, results, outputs, orig_target_sizes, max_target_sizes)
models/segmentation.py:232
Methodforward
This function computes the panoptic prediction from the model's predictions. Parameters: outputs: This is a dict coming directly
models/segmentation.py:264
Methodforward
 The forward expects a NestedTensor, which consists of: - samples.tensor: batched images, of shape [batch_size x 3 x H x W]
models/detr.py:47
Methodforward
This performs the loss computation. Parameters: outputs: dict of tensors, see the output specification of the model for the form
models/detr.py:243
Methodforward
Perform the computation Parameters: outputs: raw outputs of the model target_sizes: tensor of dimension [batch_size x
models/detr.py:290
Methodforward
(self, x)
models/detr.py:327
Methodforward
(self, x)
models/backbone.py:51
Methodforward
supports both NestedTensor and torch.Tensor
models/backbone.py:78
Methodforward
supports both NestedTensor and torch.Tensor
models/backbone.py:117
Methodforward
(self, tensor_list: NestedTensor)
models/position_encoding.py:34
Methodforward
(self, tensor_list: NestedTensor)
models/position_encoding.py:71
Methodget_height_and_width
(self, idx)
datasets/coco_panoptic.py:79
Methodget_ids_area
(masks, scores, dedup=False)
models/segmentation.py:309
Functionget_sha
()
util/misc.py:254
Methodglobal_avg
(self)
util/misc.py:74
Functioninit_distributed_mode
(args)
util/misc.py:418
Methodloss_boxes
Compute the losses related to the bounding boxes, the L1 regression loss and the GIoU loss targets dicts must contain the key "boxes" conta
models/detr.py:170
Methodloss_cardinality
Compute the cardinality error, ie the absolute error in the number of predicted non-empty boxes This is not really a loss, it is intended for
models/detr.py:157
Methodloss_feature
Compute the mse loss between normalized features.
models/detr.py:136
Methodloss_labels
Classification loss (NLL) targets dicts must contain the key "labels" containing a tensor of dim [nb_target_boxes]
models/detr.py:115
Methodloss_masks
Compute the losses related to the masks: the focal loss and the dice loss. targets dicts must contain the key "masks" containing a tensor o
models/detr.py:191
Methodmedian
(self)
util/misc.py:64
Functionplot_logs
Function to plot specific fields from training log(s). Plots both training and test results. :: Inputs - logs = list containing Path objects
util/plot_utils.py:20
Functionplot_precision_recall
(files, naming_scheme='iter')
util/plot_utils.py:83
Functionreduce_dict
Args: input_dict (dict): all the values will be reduced average (bool): whether to do average or sum Reduce the values in the
util/misc.py:137
Functionsave_on_master
(*args, **kwargs)
util/misc.py:413
Methodsynchronize_between_processes
Warning: does not synchronize the deque!
util/misc.py:50
Methodto_tuple
(tup)
models/segmentation.py:280
Functionupdetr_collate_fn
(batch)
util/misc.py:279
Methodvalue
(self)
util/misc.py:82
← previous101–193 of 193, ranked by callers