Shortcuts

mmedit.datasets.transforms

BinarizeImage

Binarize image.

Clip

Clip the pixels.

ColorJitter

An interface for torch color jitter so that it can be invoked in mmediting pipeline.

CopyValues

Copy the value of source keys to destination keys.

Crop

Crop data to specific size for training.

CropLike

Crop/pad the image in the target_key according to the size of image in the reference_key .

DegradationsWithShuffle

Apply random degradations to input, with degradations being shuffled.

LoadImageFromFile

Load a single image or image frames from corresponding paths.

LoadMask

Load Mask for multiple types.

Flip

Flip the input data with a probability.

FixedCrop

Crop paired data (at a specific position) to specific size for training.

GenerateCoordinateAndCell

Generate coordinate and cell.

GenerateFacialHeatmap

Generate heatmap from keypoint.

GenerateFrameIndices

Generate frame index for REDS datasets.

GenerateFrameIndiceswithPadding

Generate frame index with padding for REDS dataset and Vid4 dataset during testing.

GenerateSegmentIndices

Generate frame indices for a segment.

GetMaskedImage

Get masked image.

GetSpatialDiscountMask

Get spatial discounting mask constant.

MATLABLikeResize

Resize the input image using MATLAB-like downsampling.

MirrorSequence

Extend short sequences (e.g.

ModCrop

Mod crop images, used during testing.

Normalize

Normalize images with the given mean and std value.

PackEditInputs

Pack the inputs data for SR, VFI, matting and inpainting.

PairedRandomCrop

Paried random crop.

RandomAffine

Apply random affine to input images.

RandomBlur

Apply random blur to the input.

RandomDownSampling

Generate LQ image from GT (and crop), which will randomly pick a scale.

RandomJPEGCompression

Apply random JPEG compression to the input.

RandomMaskDilation

Randomly dilate binary masks.

RandomNoise

Apply random noise to the input.

RandomResize

Randomly resize the input.

RandomResizedCrop

Crop data to random size and aspect ratio.

RandomRotation

Rotate the image by a randomly-chosen angle, measured in degree.

RandomTransposeHW

Randomly transpose images in H and W dimensions with a probability.

RandomVideoCompression

Apply random video compression to the input.

RescaleToZeroOne

Transform the images into a range between 0 and 1.

Resize

Resize data to a specific size for training or resize the images to fit the network input regulation for testing.

SetValues

Set value to destination keys.

TemporalReverse

Reverse frame lists for temporal augmentation.

ToTensor

Convert some values in results dict to torch.Tensor type in data loader pipeline.

UnsharpMasking

Apply unsharp masking to an image or a sequence of images.

CropAroundCenter

Randomly crop the images around unknown area in the center 1/4 images.

CropAroundFg

Crop around the whole foreground in the segmentation mask.

GenerateSeg

Generate segmentation mask from alpha matte.

CropAroundUnknown

Crop around unknown area with a randomly selected scale.

GenerateSoftSeg

Generate soft segmentation mask from input segmentation mask.

FormatTrimap

Convert trimap (tensor) to one-hot representation.

TransformTrimap

Transform trimap into two-channel and six-channel.

GenerateTrimap

Using random erode/dilate to generate trimap from alpha matte.

GenerateTrimapWithDistTransform

Generate trimap with distance transform function.

CompositeFg

Composite foreground with a random foreground.

RandomLoadResizeBg

Randomly load a background image and resize it.

MergeFgAndBg

Composite foreground image and background image with alpha.

PerturbBg

Randomly add gaussian noise or gamma change to background image.

RandomJitter

Randomly jitter the foreground in hsv space.

LoadPairedImageFromFile

Load a pair of images from file.

CenterCropLongEdge

Center crop the given image by the long edge.

RandomCropLongEdge

Random crop the given image by the long edge.

NumpyPad

Numpy Padding.

Read the Docs v: zyh/doc-notfound-extend
Versions
master
latest
stable
zyh-doc-notfound-extend
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.