RandomIoUCrop — Torchvision 0.22 documentation (original) (raw)

class torchvision.transforms.v2.RandomIoUCrop(min_scale: float = 0.3, max_scale: float = 1.0, min_aspect_ratio: float = 0.5, max_aspect_ratio: float = 2.0, sampler_options: Optional[List[float]] = None, trials: int = 40)[source]¶

Random IoU crop transformation from“SSD: Single Shot MultiBox Detector”.

This transformation requires an image or video data and tv_tensors.BoundingBoxes in the input.

Warning

In order to properly remove the bounding boxes below the IoU threshold, RandomIoUCropmust be followed by SanitizeBoundingBoxes, either immediately after or later in the transforms pipeline.

If the input is a torch.Tensor or a TVTensor (e.g. Image,Video, BoundingBoxes etc.) it can have arbitrary number of leading batch dimensions. For example, the image can have [..., C, H, W] shape. A bounding box can have [..., 4] shape.

Parameters:

• min_scale (float, optional) – Minimum factors to scale the input size.
• max_scale (float, optional) – Maximum factors to scale the input size.
• min_aspect_ratio (float, optional) – Minimum aspect ratio for the cropped image or video.
• max_aspect_ratio (float, optional) – Maximum aspect ratio for the cropped image or video.
• sampler_options (list of python:float , optional) – List of minimal IoU (Jaccard) overlap between all the boxes and a cropped image or video. Default, None which corresponds to [0.0, 0.1, 0.3, 0.5, 0.7, 0.9, 1.0]
• trials (int, optional) – Number of trials to find a crop for a given value of minimal IoU (Jaccard) overlap. Default, 40.

Examples using RandomIoUCrop:

make_params(flat_inputs: List[Any]) → Dict[str, Any][source]¶

Method to override for custom transforms.

See How to write your own v2 transforms

transform(inpt: Any, params: Dict[str, Any]) → Any[source]¶

Method to override for custom transforms.

See How to write your own v2 transforms