composer.algorithms.blurpool

Modules

composer.algorithms.blurpool.blurpool	composer.algorithms.blurpool.blurpool
composer.algorithms.blurpool.blurpool_layers	composer.algorithms.blurpool.blurpool_layers

BlurPool adds anti-aliasing filters to convolutional layers to increase accuracy and invariance to small shifts in the input.

See BlurPool or the Method Card for details.

Functions

apply_blurpool	Add anti-aliasing filters to the strided torch.nn.Conv2d and/or torch.nn.MaxPool2d modules within model.
blur_2d	Apply a spatial low-pass filter.
blurmax_pool2d	Max-pooling with anti-aliasing.

Classes

BlurConv2d	This module is a drop-in replacement for PyTorch's Conv2d, but with an anti-aliasing filter applied.
BlurMaxPool2d	This module is a (nearly) drop-in replacement for PyTorch's MaxPool2d, but with an anti- aliasing filter applied.
BlurPool	BlurPool adds anti-aliasing filters to convolutional layers to increase accuracy and invariance to small shifts in the input.
BlurPool2d	This module just calls blur_2d() in forward using the provided arguments.