Caffe2 implementation of preprocess_image, which is called inside each MetaArch's forward. It normalizes the input images, and the fi

Calculate proper im2col step size, which should be divisible by input_size and not larger than prefer_size. Meanwhile the step size s

Returns: bool: whether the arguments contain CfgNode and should be considered forwarded to from_config.

convert prediction from Dict to D2 Instances format

convert prediction from Dict to D2 Instances format

convert prediction from Dict to D2 Instances format

convert prediction from Dict to D2 Instances format

Run `_decode_per_level_predictions` for all feature levels and concat the results.

Transform anchors into proposals by applying the predicted anchor deltas. Returns: proposals (list[Tensor]): A list of L

Derive the desired score numbers from summarized COCOeval. Args: coco_eval (None or COCOEval): None represents no predic

Find proper places to draw text given a binary mask.

Apply EMA update to `self.name` using `value`. This is mainly used for loss normalizer. In Detectron1, loss is normalized by number

Extract arrays of I, U and V values at given points as numpy arrays given the data mode stored in self._dpDataMode

Forward logic of the box prediction branch. If `self.train_on_pred_boxes is True`, the function puts predicted boxes in the `prop

Args: features, targets: the same as in Same as in :meth:`ROIHeads.forward`. proposals (list[Instance

Forward logic of the densepose prediction branch. Args: features (dict[str, Tensor]): input data as a mapping from featu

Use `from_config` to obtain explicit arguments. Returns: dict: arguments to be used for cls.__init__

Produces embeddings and geodesic distance tensors for a given mesh. May subsample the mesh, if it contains too many vertices (control

Return a process group based on gloo backend, containing all the ranks The result is cached.

Group elements of an iterable by chunks of size `n`, e.g. grouper(range(9), 4) -> (0, 1, 2, 3), (4, 5, 6, 7), (8, None, None, None)

A context which ignores CUDA OOM exception from pytorch.

Computes utility values for linear interpolation at points v. The points are given as normalized offsets in the source interval (v0_src,

["abc.zfg", "abc.zef"] -> "abc."

merge dict ``a`` into dict ``b`` (non-inplace). Values in ``a`` will overwrite ``b``. ``b`` is copied first to avoid in-place modific

Prepare ._gts and ._dts for evaluation based on params :return: None

Abstract method to produce a sample of indices to select data To be implemented in descendants Args: values (tor

Based on the matching between N proposals and M groundtruth, sample the proposals and set their classification labels. Args:

Randomly sample a subset of positive and negative examples, and overwrite the label vector to the ignore value (-1) for all elements

Wrap Transform into Augmentation. Private, used internally to implement augmentations.

Open a context where some heads in `model.roi_heads` are temporarily turned off. Args: attr (list[str]): the attribute in

Calculate decomposed Relative Positional Embeddings from :paper:`mvitv2`. https://github.com/facebookresearch/mvit/blob/19786631e330df9f3622e

Add config for DeepLab.

Called after the last iteration.

Match names between the two state-dict, and returns a new chkpt_state_dict with names converted to match model_state_dict with heuristics. Th

An FX-tracing safe version of assert. Avoids erroneous type assertion triggering when types are masked inside an fx.proxy.Proxy object during

Create evaluator(s) for a given dataset. This uses the special metadata "evaluator_type" associated with each builtin dataset. For your o