pytorch_tutorials package

Submodules

pytorch_tutorials.coco_eval module

pytorch_tutorials.coco_utils module

pytorch_tutorials.engine module

pytorch_tutorials.transforms module

class pytorch_tutorials.transforms.Compose(transforms)

Bases: object

class pytorch_tutorials.transforms.FixedSizeCrop(size, fill=0, padding_mode='constant')

Bases: Module

forward(img, target=None)

Define the computation performed at every call.

Should be overridden by all subclasses.

Note

Although the recipe for forward pass needs to be defined within this function, one should call the Module instance afterwards instead of this since the former takes care of running the registered hooks while the latter silently ignores them.

class pytorch_tutorials.transforms.PILToTensor(*args, **kwargs)

Bases: Module

forward(image, target=None)

Define the computation performed at every call.

Should be overridden by all subclasses.

Note

Although the recipe for forward pass needs to be defined within this function, one should call the Module instance afterwards instead of this since the former takes care of running the registered hooks while the latter silently ignores them.

Return type:

Tuple[Tensor, Optional[Dict[str, Tensor]]]

class pytorch_tutorials.transforms.RandomHorizontalFlip(p=0.5)

Bases: RandomHorizontalFlip

forward(image, target=None)

Parameters:

img (PIL Image or Tensor) – Image to be flipped.

Returns:

Randomly flipped image.

Return type:

PIL Image or Tensor

class pytorch_tutorials.transforms.RandomIoUCrop(min_scale=0.3, max_scale=1.0, min_aspect_ratio=0.5, max_aspect_ratio=2.0, sampler_options=None, trials=40)

Bases: Module

forward(image, target=None)

Define the computation performed at every call.

Should be overridden by all subclasses.

Note

Although the recipe for forward pass needs to be defined within this function, one should call the Module instance afterwards instead of this since the former takes care of running the registered hooks while the latter silently ignores them.

Return type:

Tuple[Tensor, Optional[Dict[str, Tensor]]]

class pytorch_tutorials.transforms.RandomPhotometricDistort(contrast=(0.5, 1.5), saturation=(0.5, 1.5), hue=(-0.05, 0.05), brightness=(0.875, 1.125), p=0.5)

Bases: Module

forward(image, target=None)

Define the computation performed at every call.

Should be overridden by all subclasses.

Note

Although the recipe for forward pass needs to be defined within this function, one should call the Module instance afterwards instead of this since the former takes care of running the registered hooks while the latter silently ignores them.

Return type:

Tuple[Tensor, Optional[Dict[str, Tensor]]]

class pytorch_tutorials.transforms.RandomShortestSize(min_size, max_size, interpolation=InterpolationMode.BILINEAR)

Bases: Module

forward(image, target=None)

Define the computation performed at every call.

Should be overridden by all subclasses.

Note

Although the recipe for forward pass needs to be defined within this function, one should call the Module instance afterwards instead of this since the former takes care of running the registered hooks while the latter silently ignores them.

Return type:

Tuple[Tensor, Optional[Dict[str, Tensor]]]

class pytorch_tutorials.transforms.RandomZoomOut(fill=None, side_range=(1.0, 4.0), p=0.5)

Bases: Module

forward(image, target=None)

Define the computation performed at every call.

Should be overridden by all subclasses.

Note

Although the recipe for forward pass needs to be defined within this function, one should call the Module instance afterwards instead of this since the former takes care of running the registered hooks while the latter silently ignores them.

Return type:

Tuple[Tensor, Optional[Dict[str, Tensor]]]

class pytorch_tutorials.transforms.ScaleJitter(target_size, scale_range=(0.1, 2.0), interpolation=InterpolationMode.BILINEAR, antialias=True)

Bases: Module

Randomly resizes the image and its bounding boxes within the specified scale range. The class implements the Scale Jitter augmentation as described in the paper `”Simple Copy-Paste is a Strong Data Augmentation Method for Instance Segmentation”

<https://arxiv.org/abs/2012.07177>`_.

Parameters:

• target_size (tuple of ints) – The target size for the transform provided in (height, weight) format.

• scale_range (tuple of ints) – scaling factor interval, e.g (a, b), then scale is randomly sampled from the range a <= scale <= b.

• interpolation (InterpolationMode) – Desired interpolation enum defined by torchvision.transforms.InterpolationMode. Default is InterpolationMode.BILINEAR.

forward(image, target=None)

Define the computation performed at every call.

Should be overridden by all subclasses.

Note

Although the recipe for forward pass needs to be defined within this function, one should call the Module instance afterwards instead of this since the former takes care of running the registered hooks while the latter silently ignores them.

Return type:

Tuple[Tensor, Optional[Dict[str, Tensor]]]

class pytorch_tutorials.transforms.SimpleCopyPaste(blending=True, resize_interpolation=InterpolationMode.BILINEAR)

Bases: Module

forward(images, targets)

Define the computation performed at every call.

Should be overridden by all subclasses.

Note

Although the recipe for forward pass needs to be defined within this function, one should call the Module instance afterwards instead of this since the former takes care of running the registered hooks while the latter silently ignores them.

Return type:

Tuple[List[Tensor], List[Dict[str, Tensor]]]

class pytorch_tutorials.transforms.ToDtype(dtype, scale=False)

Bases: Module

forward(image, target=None)

Define the computation performed at every call.

Should be overridden by all subclasses.

Note

Although the recipe for forward pass needs to be defined within this function, one should call the Module instance afterwards instead of this since the former takes care of running the registered hooks while the latter silently ignores them.

Return type:

Tuple[Tensor, Optional[Dict[str, Tensor]]]

pytorch_tutorials.utils module

class pytorch_tutorials.utils.MetricLogger(delimiter='\t')

Bases: object

add_meter(name, meter)

log_every(iterable, print_freq, header=None)

synchronize_between_processes()

update(**kwargs)

class pytorch_tutorials.utils.SmoothedValue(window_size=20, fmt=None)

Bases: object

Track a series of values and provide access to smoothed values over a window or the global series average.

property avg

property global_avg

property max

property median

synchronize_between_processes()

Warning: does not synchronize the deque!

update(value, n=1)

property value

pytorch_tutorials.utils.all_gather(data)

Run all_gather on arbitrary picklable data (not necessarily tensors) :param data: any picklable object

Returns:

list of data gathered from each rank

Return type:

list[data]

pytorch_tutorials.utils.collate_fn(batch)

pytorch_tutorials.utils.get_rank()

pytorch_tutorials.utils.get_world_size()

pytorch_tutorials.utils.init_distributed_mode(args)

pytorch_tutorials.utils.is_dist_avail_and_initialized()

pytorch_tutorials.utils.is_main_process()

pytorch_tutorials.utils.mkdir(path)

pytorch_tutorials.utils.reduce_dict(input_dict, average=True)

Parameters:

• input_dict (dict) – all the values will be reduced

• average (bool) – whether to do average or sum

Reduce the values in the dictionary from all processes so that all processes have the averaged results. Returns a dict with the same fields as input_dict, after reduction.

pytorch_tutorials.utils.save_on_master(*args, **kwargs)

pytorch_tutorials.utils.setup_for_distributed(is_master)

This function disables printing when not in master process