vision.transform

List of transforms for data augmentation in CV

Image transforms

fastai provides a complete image transformation library written from scratch in PyTorch. Although the main purpose of the library is data augmentation for use when training computer vision models, you can also use it for more general image transformation purposes. Before we get in to the detail of the full API, we’ll look at a quick overview of the data augmentation pieces that you’ll almost certainly need to use.

Data augmentation

Data augmentation is perhaps the most important regularization technique when training a model for Computer Vision: instead of feeding the model with the same pictures every time, we do small random transformations (a bit of rotation, zoom, translation, etc…) that don’t change what’s inside the image (to the human eye) but do change its pixel values. Models trained with data augmentation will then generalize better.

To get a set of transforms with default values that work pretty well in a wide range of tasks, it’s often easiest to use get_transforms. Depending on the nature of the images in your data, you may want to adjust a few arguments, the most important being:

  • do_flip: if True the image is randomly flipped (default behavior)
  • flip_vert: limit the flips to horizontal flips (when False) or to horizontal and vertical flips as well as 90-degrees rotations (when True)

get_transforms returns a tuple of two lists of transforms: one for the training set and one for the validation set (we don’t want to modify the pictures in the validation set, so the second list of transforms is limited to resizing the pictures). This can be passed directly to define a DataBunch object (see below) which is then associated with a model to begin training.

Note that the defaults for get_transforms are generally pretty good for regular photos - although here we’ll add a bit of extra rotation so it’s easier to see the differences.

  1. tfms = get_transforms(max_rotate=25)
  2. len(tfms)
  1. 2

We first define a function to return a new image, since transformation functions modify their inputs. We also define a little helper function plots_f to let us output a grid of transformed images based on a function - the details of this function aren’t important here.

  1. def get_ex(): return open_image('imgs/cat_example.jpg')
  2. def plots_f(rows, cols, width, height, **kwargs):
  3. [get_ex().apply_tfms(tfms[0], **kwargs).show(ax=ax) for i,ax in enumerate(plt.subplots(
  4. rows,cols,figsize=(width,height))[1].flatten())]

If we want to have a look at what these transforms actually do, we need to use the apply_tfms function. It will be in charge of picking the values of the random parameters and doing the transformations to the Image object. This function has multiple arguments you can customize (see its documentation for details); here we will highlight the most useful. The first one we’ll need to set, especially if our images are of different shapes, is the target size. It will ensure that all the images are cropped or padded to the same size so we can then collate them into batches.

  1. plots_f(2, 4, 12, 6, size=224)

vision.transform - 图1

Note that the target size can be a rectangle if you specify a tuple of int.

Note: In fastai we follow the convention of numpy and pytorch for image dimensions: (height, width). It’s different from PIL or matplolib so don’t get confused.

  1. plots_f(2, 4, 12, 8, size=(300,200))

vision.transform - 图2

The second argument that can be customized is how we treat missing pixels: when applying transforms (like a rotation), some of the pixels inside the square won’t have values from the image. We can set missing pixels to one of the following:

  • black (padding_mode=’zeros’)
  • the value of the pixel at the nearest border (padding_mode=’border’)
  • the value of the pixel symmetric to the nearest border (padding_mode=’reflection’)

padding_mode=’reflection’ is the default. Here is what padding_mode=’zeros’ looks like:

  1. plots_f(2, 4, 12, 6, size=224, padding_mode='zeros')

vision.transform - 图3

And padding_mode=’border’ looks like this:

  1. plots_f(2, 4, 12, 6, size=224, padding_mode='border')

vision.transform - 图4

The third argument that might be useful to change is resize_method. Images are often rectangles of different ratios, so to get them to the target size, we may need to crop, squish, or pad them to get the ratio right.

By default, the library resizes the image while keeping its original ratio so that the smaller size corresponds to the given size, then takes a crop (ResizeMethod.CROP). You can choose to resize the image while keeping its original ratio so that the bigger size corresponds to the given size, then take a pad (ResizeMethod.PAD). Another way is to just squish the image to the given size (ResizeMethod.SQUISH).

  1. _,axs = plt.subplots(1,3,figsize=(9,3))
  2. for rsz,ax in zip([ResizeMethod.CROP, ResizeMethod.PAD, ResizeMethod.SQUISH], axs):
  3. get_ex().apply_tfms([crop_pad()], size=224, resize_method=rsz, padding_mode='zeros').show(ax=ax, title=rsz.name.lower())

vision.transform - 图5

Data augmentation details

If you want to quickly get a set of random transforms that have worked well in a wide range of tasks, you should use the get_transforms function. The most important parameters to adjust are do_flip and flip_vert, depending on the type of images you have.

get_transforms[source][test]

get_transforms(do_flip:bool=True, flip_vert:bool=False, max_rotate:float=10.0, max_zoom:float=1.1, max_lighting:float=0.2, max_warp:float=0.2, p_affine:float=0.75, p_lighting:float=0.75, xtra_tfms:Optional[Collection[Transform]]=None) → Collection[Transform] Tests found for get_transforms:

  • pytest -sv tests/test_vision_data.py::test_image_to_image_different_tfms [source]
  • pytest -sv tests/test_vision_data.py::test_image_to_image_different_y_size [source]

Some other tests where get_transforms is used:

  • pytest -sv tests/test_vision_transform.py::test_crop_without_size [source]

To run tests please refer to this guide.

Utility func to easily create a list of flip, rotate, zoom, warp, lighting transforms.

  • do_flip: if True, a random flip is applied with probability 0.5
  • flip_vert: requires do_flip=True. If True, the image can be flipped vertically or rotated by 90 degrees, otherwise only an horizontal flip is applied
  • max_rotate: if not None, a random rotation between -max_rotate and max_rotate degrees is applied with probability p_affine
  • max_zoom: if not 1. or less, a random zoom between 1. and max_zoom is applied with probability p_affine
  • max_lighting: if not None, a random lightning and contrast change controlled by max_lighting is applied with probability p_lighting
  • max_warp: if not None, a random symmetric warp of magnitude between -max_warp and maw_warp is applied with probability p_affine
  • p_affine: the probability that each affine transform and symmetric warp is applied
  • p_lighting: the probability that each lighting transform is applied
  • xtra_tfms: a list of additional transforms you would like to be applied

This function returns a tuple of two lists of transforms, one for the training set and the other for the validation set (which is limited to a center crop by default).

  1. tfms = get_transforms(max_rotate=25); len(tfms)
  1. 2

Let’s see how get_transforms changes this little kitten now.

  1. plots_f(2, 4, 12, 6, size=224)

vision.transform - 图6

Another useful function that gives basic transforms is zoom_crop:

zoom_crop[source][test]

zoom_crop(scale:float, do_rand:bool=False, p:float=1.0) No tests found for zoom_crop. To contribute a test please refer to this guide and this discussion.

Randomly zoom and/or crop.

  • scale: Decimal or range of decimals to zoom the image
  • do_rand: If true, transform is randomized, otherwise it’s a zoom of scale and a center crop
  • p: Probability to apply the zoom

scale should be a given float if do_rand is False, otherwise it can be a range of floats (and the zoom will have a random value in between). Again, here is a sense of what this can give us:

  1. tfms = zoom_crop(scale=(0.75,2), do_rand=True)
  2. plots_f(2, 4, 12, 6, size=224)

vision.transform - 图7

rand_resize_crop[source][test]

rand_resize_crop(size:int, max_scale:float=2.0, ratios:Point=(0.75, 1.33)) No tests found for rand_resize_crop. To contribute a test please refer to this guide and this discussion.

Randomly resize and crop the image to a ratio in ratios after a zoom of max_scale.

  • size: Final size of the image
  • max_scale: Zooms the image to a random scale up to this
  • ratios: Range of ratios in which a new one will be randomly picked

This transform is an implementation of the main approach used for nearly all winning Imagenet entries since 2013, based on Andrew Howard’s Some Improvements on Deep Convolutional Neural Network Based Image Classification. It determines a new width and height of the image after the random scale and squish to the new ratio are applied. Those are switched with probability 0.5. Then we return the part of the image with the width and height computed, centered in row_pct, col_pct if width and height are both less than the corresponding size of the image. Otherwise we try again with new random parameters.

  1. tfms = [rand_resize_crop(224)]
  2. plots_f(2, 4, 12, 6, size=224)

vision.transform - 图8

Randomness

The functions that define each transform, such as rotateor flip_lr are deterministic. The fastai library will then randomize them in two different ways:

  • each transform can be defined with an argument named p representing the probability for it to be applied
  • each argument that is type-annotated with a random function (like uniform or rand_bool) can be replaced by a tuple of arguments accepted by this function, and on each call of the transform, the argument that is passed inside the function will be picked randomly using that random function.

If we look at the function rotate for instance, we see it has an argument degrees that is type-annotated as uniform.

First level of randomness: We can define a transform using rotate with degrees fixed to a value, but by passing an argument p. The rotation will then be executed with a probability of p but always with the same value of degrees.

  1. tfm = [rotate(degrees=30, p=0.5)]
  2. fig, axs = plt.subplots(1,5,figsize=(12,4))
  3. for ax in axs:
  4. img = get_ex().apply_tfms(tfm)
  5. title = 'Done' if tfm[0].do_run else 'Not done'
  6. img.show(ax=ax, title=title)

vision.transform - 图9

Second level of randomness: We can define a transform using rotate with degrees defined as a range, without an argument p. The rotation will then always be executed with a random value picked uniformly between the two floats we put in degrees.

  1. tfm = [rotate(degrees=(-30,30))]
  2. fig, axs = plt.subplots(1,5,figsize=(12,4))
  3. for ax in axs:
  4. img = get_ex().apply_tfms(tfm)
  5. title = f"deg={tfm[0].resolved['degrees']:.1f}"
  6. img.show(ax=ax, title=title)

vision.transform - 图10

All combined: We can define a transform using rotate with degrees defined as a range, and an argument p. The rotation will then be executed with a probability p and with a random value picked uniformly between the two floats we put in degrees.

  1. tfm = [rotate(degrees=(-30,30), p=0.75)]
  2. fig, axs = plt.subplots(1,5,figsize=(12,4))
  3. for ax in axs:
  4. img = get_ex().apply_tfms(tfm)
  5. title = f"Done, deg={tfm[0].resolved['degrees']:.1f}" if tfm[0].do_run else f'Not done'
  6. img.show(ax=ax, title=title)

vision.transform - 图11

List of transforms

Here is the list of all the deterministic functions on which the transforms are built. As explained before, each of them can have a probability p of being executed, and any time an argument is type-annotated with a random function, it’s possible to randomize it via that function.

brightness[source][test]

brightness(x, change:uniform) → Image :: TfmLighting No tests found for brightness. To contribute a test please refer to this guide and this discussion.

Apply change in brightness of image x.

This transform adjusts the brightness of the image depending on the value of change. A change of 0 will transform the image to black, and a change of 1 will transform the image to white. change=0.5 doesn’t adjust the brightness.

  1. fig, axs = plt.subplots(1,5,figsize=(12,4))
  2. for change, ax in zip(np.linspace(0.1,0.9,5), axs):
  3. brightness(get_ex(), change).show(ax=ax, title=f'change={change:.1f}')

vision.transform - 图12

contrast[source][test]

contrast(x, scale:log_uniform) → Image :: TfmLighting No tests found for contrast. To contribute a test please refer to this guide and this discussion.

Apply scale to contrast of image x.

scale adjusts the contrast. A scale of 0 will transform the image to grey, and a scale over 1 will transform the picture to super-contrast. scale = 1. doesn’t adjust the contrast.

  1. fig, axs = plt.subplots(1,5,figsize=(12,4))
  2. for scale, ax in zip(np.exp(np.linspace(log(0.5),log(2),5)), axs):
  3. contrast(get_ex(), scale).show(ax=ax, title=f'scale={scale:.2f}')

vision.transform - 图13

  1. show_doc(crop)

crop[source][test]

crop(x, size, row_pct:uniform=0.5, col_pct:uniform=0.5) → Image :: TfmPixel Tests found for crop:

  • pytest -sv tests/test_vision_transform.py::test_crop_without_size [source]
  • pytest -sv tests/test_vision_transform.py::test_crops_with_tensor_image_sizes [source]
  • pytest -sv tests/test_vision_transform.py::test_deterministic_transforms [source]

To run tests please refer to this guide.

This transform takes a crop of the image to return one of the given size. The position is given by (col_pct, row_pct), with col_pct and row_pct being normalized between 0. and 1.

  1. fig, axs = plt.subplots(1,5,figsize=(12,4))
  2. for center, ax in zip([[0.,0.], [0.,1.],[0.5,0.5],[1.,0.], [1.,1.]], axs):
  3. crop(get_ex(), 300, *center).show(ax=ax, title=f'center=({center[0]}, {center[1]})')

vision.transform - 图14

crop_pad[source][test]

crop_pad(x, size, padding_mode='reflection', row_pct:uniform=0.5, col_pct:uniform=0.5) → Image :: TfmCrop No tests found for crop_pad. To contribute a test please refer to this guide and this discussion.

  • x: Image to transform
  • size: Size of the crop, if it’s an int, the crop will be square
  • padding_mode: How to pad the output image (‘zeros’, ‘border’ or ‘reflection’)
  • row_pct: Between 0. and 1., position of the center on the y axis (0. is top, 1. is bottom, 0.5 is center)
  • col_pct: Between 0. and 1., position of the center on the x axis (0. is left, 1. is right, 0.5 is center)

This works like crop but if the target size is bigger than the size of the image (on either dimension), padding is applied according to padding_mode (see pad for an example of all the options) and the position of center is ignored on that dimension.

  1. fig, axs = plt.subplots(1,5,figsize=(12,4))
  2. for size, ax in zip(np.linspace(200,600,5), axs):
  3. crop_pad(get_ex(), int(size), 'zeros', 0.,0.).show(ax=ax, title=f'size = {int(size)}')

vision.transform - 图15

dihedral[source][test]

dihedral(x, k:partial(uniform_int, 0, 7``)) → Image :: TfmPixel Tests found for dihedral:

  • pytest -sv tests/test_vision_transform.py::test_all_dihedral [source]

To run tests please refer to this guide.

Randomly flip x image based on k.

This transform applies combines a flip (horizontal or vertical) and a rotation of a multiple of 90 degrees.

  1. fig, axs = plt.subplots(2,4,figsize=(12,8))
  2. for k, ax in enumerate(axs.flatten()):
  3. dihedral(get_ex(), k).show(ax=ax, title=f'k={k}')
  4. plt.tight_layout()

vision.transform - 图16

dihedral_affine[source][test]

dihedral_affine(k:partial(uniform_int, 0, 7``)) → Image :: TfmAffine No tests found for dihedral_affine. To contribute a test please refer to this guide and this discussion.

Randomly flip x image based on k.

This is an affine implementation of dihedral that should be used if the target is an ImagePoints or an ImageBBox.

flip_lr[source][test]

flip_lr(x) → Image :: TfmPixel Tests found for flip_lr:

  • pytest -sv tests/test_vision_transform.py::test_deterministic_transforms [source]

To run tests please refer to this guide.

Flip x horizontally.

This transform horizontally flips the image. flip_lr mirrors the image.

  1. fig, axs = plt.subplots(1,2,figsize=(6,4))
  2. get_ex().show(ax=axs[0], title=f'no flip')
  3. flip_lr(get_ex()).show(ax=axs[1], title=f'flip')

vision.transform - 图17

flip_affine[source][test]

flip_affine() → Image :: TfmAffine Tests found for flip_affine:

  • pytest -sv tests/test_vision_transform.py::test_deterministic_transforms [source]

To run tests please refer to this guide.

Flip x horizontally.

This is an affine implementation of flip_lr that should be used if the target is an ImagePoints or an ImageBBox.

jitter[source][test]

jitter(c, magnitude:uniform) → Image :: TfmCoord No tests found for jitter. To contribute a test please refer to this guide and this discussion.

This transform changes the pixels of the image by randomly replacing them with pixels from the neighborhood (how far the neighborhood extends is controlled by the value of magnitude).

  1. fig, axs = plt.subplots(1,5,figsize=(12,4))
  2. for magnitude, ax in zip(np.linspace(-0.05,0.05,5), axs):
  3. tfm = jitter(magnitude=magnitude)
  4. get_ex().jitter(magnitude).show(ax=ax, title=f'magnitude={magnitude:.2f}')

vision.transform - 图18

pad[source][test]

pad(x, padding:int, mode='reflection') → Image :: TfmPixel Tests found for pad:

  • pytest -sv tests/test_vision_transform.py::test_deterministic_transforms [source]

To run tests please refer to this guide.

Pad the image by adding padding pixel on each side of the picture according to mode:

  • mode='zeros': pads with zeros,
  • mode='border': repeats the pixels at the border.
  • mode='reflection': pads by taking the pixels symmetric to the border.
  1. fig, axs = plt.subplots(1,3,figsize=(12,4))
  2. for mode, ax in zip(['zeros', 'border', 'reflection'], axs):
  3. pad(get_ex(), 50, mode).show(ax=ax, title=f'mode={mode}')

vision.transform - 图19

perspective_warp[source][test]

perspective_warp(c, magnitude:partial(uniform, size=8``)=0, invert=False) → Image :: TfmCoord Tests found for perspective_warp:

  • pytest -sv tests/test_vision_transform.py::test_all_warps [source]

To run tests please refer to this guide.

Apply warp of magnitude to c.

Perspective warping is a deformation of the image as seen in a different plane of the 3D-plane. The new plane is determined by telling where we want each of the four corners of the image (from -1 to 1, -1 being left/top, 1 being right/bottom).

  1. fig, axs = plt.subplots(2,4,figsize=(12,8))
  2. for i, ax in enumerate(axs.flatten()):
  3. magnitudes = torch.tensor(np.zeros(8))
  4. magnitudes[i] = 0.5
  5. perspective_warp(get_ex(), magnitudes).show(ax=ax, title=f'coord {i}')

vision.transform - 图20

resize

Pytorch’s transforms.Resize(size) equivalent is implemented without an explicit transform function in fastai. It’s done via the arguments size and resize_method.

The size argument can be either a single int 224, or a tuple of ints (224,400). The default behavior is to crop the image to a square when a single int is passed and to squish it in the case of a tuple, so that:

  • if size=224 is passed, it will resize and then crop to (224,224)
  • if size=(224,400) is passed, it will squish it to (224,400)
  • if size=(224,224) is passed, it will squish (not crop!) it to (224,224)

You can override the default resize_method.

Note:

If you receive an error similar to the one below:

  1. RuntimeError: Argument #4: Padding size should be less than the corresponding input dimension, but got: padding (46, 46) at dimension 3 of input [1, 3, 128, 36]

this is caused by an issue with PyTorch’s reflection padding, which the library uses by default. Adding an extra keyword argument padding_mode = 'zeros' should be able to serve as a workaround for now.

The resize is performed slightly differently depending on how ImageDataBunch is created:

  1. When the shortcut ImageDataBunch from_* methods are used, the size and resize_method arguments are passed with the rest of the arguments. For example, to resize images on the fly to 224x224 with from_name_re method, do:

    1. data = ImageDataBunch.from_name_re(path_img, fnames, pat, size=224, bs=bs)

    and to override the resize_method:

    1. data = ImageDataBunch.from_name_re(path_img, fnames, pat, size=224, resize_method=ResizeMethod.SQUISH, bs=bs)
  2. When data block API is used, the size and resize_method are passed via the transform method. For example:

    1. src = ImageList.from_folder(path).split_none().label_from_folder()
    2. tfms = get_transforms() # or tfms=None if none are needed
    3. size=224 # size=(224,224) or (400,224)
    4. data = src.transform(tfms=tfms, size=size, resize_method=ResizeMethod.SQUISH).databunch(bs=bs, num_workers=4).normalize()

Resizing before training

Do note that if you just want to resize the input images, doing it on the fly via transform is inefficient, since it will have to be done on every notebook re-run. Chances are that you will want to resize the images on the filesystem and use the resized datasets as needed.

For example, you could use fastai code to do that, to create low-resolution images under ‘small-96’, and mid-resolution images under small-256:

  1. from fastai.vision import *
  2. path = untar_data(URLs.PETS)
  3. path_hr = path/'images'
  4. path_lr = path/'small-96'
  5. path_mr = path/'small-256'
  6. il = ImageList.from_folder(path_hr)
  7. def resize_one(fn, i, path, size):
  8. dest = path/fn.relative_to(path_hr)
  9. dest.parent.mkdir(parents=True, exist_ok=True)
  10. img = PIL.Image.open(fn)
  11. targ_sz = resize_to(img, size, use_min=True)
  12. img = img.resize(targ_sz, resample=PIL.Image.BILINEAR).convert('RGB')
  13. img.save(dest, quality=75)
  14. # create smaller image sets the first time this nb is run
  15. sets = [(path_lr, 96), (path_mr, 256)]
  16. for p,size in sets:
  17. if not p.exists():
  18. print(f"resizing to {size} into {p}")
  19. parallel(partial(resize_one, path=p, size=size), il.items)

Of course, adjust quality, resample, and other arguments to suit your needs. You will also need to tweak it for custom directories (train, test, etc.)

imagemagick‘s mogrify and convert are commonly-used tools to resize images via your shell. For example, if you’re in a data directory, containing a test directory:

  1. ls -1 *
  2. test

and you want to create a new directory 300x224/test with images resized to 300x224:

  1. SRC=train; DEST=300x224; mkdir -p $DEST/$SRC; find $SRC -name "*.jpg" -exec convert -resize 300x224 -gravity center -extent 300x224 {} $DEST/{} ;

Check the imagemagick documentation for the many various options.

If you already have a directory which is a copy of original images, mogrify is usually applied directly to the files with the same result:

  1. mkdir 300x224
  2. cp -r train 300x224
  3. cd 300x224/train
  4. mogrify -resize 300x224 -gravity center -extent 300x224 *jpg

Note: In fastai we follow the convention of numpy and pytorch for image dimensions: (height, width). It’s different from PIL or matplotlib, so don’t get confused. Passing size=(300,200) for instance will give you a height of 300 and a width of 200.

rotate[source][test]

rotate(degrees:uniform) → Image :: TfmAffine Tests found for rotate:

  • pytest -sv tests/test_vision_transform.py::test_deterministic_transforms [source]

To run tests please refer to this guide.

Rotate image by degrees.

  1. fig, axs = plt.subplots(1,5,figsize=(12,4))
  2. for deg, ax in zip(np.linspace(-60,60,5), axs):
  3. get_ex().rotate(degrees=deg).show(ax=ax, title=f'degrees={deg}')

vision.transform - 图21

rgb_randomize[source][test]

rgb_randomize(x, channel:int=None, thresh:float=0.3) → Image :: TfmPixel No tests found for rgb_randomize. To contribute a test please refer to this guide and this discussion.

Randomize one of the channels of the input image

  • channel: Which channel (RGB) to randomise
  • thresh: After randomising, scale the values to not exceed the thresh value

By randomizing one of the three channels, the learner essentially sees the same image, but with different colors. Usually, every RGB image has one channel that is dominant, and randomizing this channel is the riskiest; thus, a low thresh (threshold) value must be applied. In this example, the Green channel is the dominant one.

  1. fig, axs = plt.subplots(3,3,figsize=(12,12))
  2. channels = ['Red', 'Green', 'Blue']
  3. for i in np.arange(0, 3):
  4. for thresh, ax in zip(np.linspace(0.2, 0.99, 3), axs[:, i]):
  5. get_ex().rgb_randomize(channel = i, thresh = thresh).show(
  6. ax=ax, title = f'{channels[i]}, thresh={thresh}')

vision.transform - 图22

skew[source][test]

skew(c, direction:uniform_int, magnitude:uniform=0, invert=False) → Image :: TfmCoord Tests found for skew:

  • pytest -sv tests/test_vision_transform.py::test_all_warps [source]

To run tests please refer to this guide.

Skew c field with random direction and magnitude.

  1. fig, axs = plt.subplots(2,4,figsize=(12,8))
  2. for i, ax in enumerate(axs.flatten()):
  3. get_ex().skew(i, 0.2).show(ax=ax, title=f'direction={i}')

vision.transform - 图23

squish[source][test]

squish(scale:uniform=1.0, row_pct:uniform=0.5, col_pct:uniform=0.5) → Image :: TfmAffine Tests found for squish:

  • pytest -sv tests/test_vision_transform.py::test_deterministic_transforms [source]

To run tests please refer to this guide.

Squish image by scale. row_pct,col_pct select focal point of zoom.

  1. fig, axs = plt.subplots(1,5,figsize=(12,4))
  2. for scale, ax in zip(np.linspace(0.66,1.33,5), axs):
  3. get_ex().squish(scale=scale).show(ax=ax, title=f'scale={scale:.2f}')

vision.transform - 图24

symmetric_warp[source][test]

symmetric_warp(c, magnitude:partial(uniform, size=4``)=0, invert=False) → Image :: TfmCoord No tests found for symmetric_warp. To contribute a test please refer to this guide and this discussion.

Apply symmetric warp of magnitude to c.

Apply the four tilts at the same time, each with a strength given in the vector magnitude. See tilt just below for the effect of each individual tilt.

  1. tfm = symmetric_warp(magnitude=(-0.2,0.2))
  2. _, axs = plt.subplots(2,4,figsize=(12,6))
  3. for ax in axs.flatten():
  4. img = get_ex().apply_tfms(tfm, padding_mode='zeros')
  5. img.show(ax=ax)

vision.transform - 图25

tilt[source][test]

tilt(c, direction:uniform_int, magnitude:uniform=0, invert=False) → Image :: TfmCoord Tests found for tilt:

  • pytest -sv tests/test_vision_transform.py::test_all_warps [source]

To run tests please refer to this guide.

Tilt c field with random direction and magnitude.

direction is a number (0: left, 1: right, 2: top, 3: bottom). A positive magnitude is a tilt forward (toward the person looking at the picture), a negative magnitude a tilt backward.

  1. fig, axs = plt.subplots(2,4,figsize=(12,8))
  2. for i in range(4):
  3. get_ex().tilt(i, 0.4).show(ax=axs[0,i], title=f'direction={i}, fwd')
  4. get_ex().tilt(i, -0.4).show(ax=axs[1,i], title=f'direction={i}, bwd')

vision.transform - 图26

zoom[source][test]

zoom(scale:uniform=1.0, row_pct:uniform=0.5, col_pct:uniform=0.5) → Image :: TfmAffine Tests found for zoom:

  • pytest -sv tests/test_vision_transform.py::test_deterministic_transforms [source]

To run tests please refer to this guide.

Zoom image by scale. row_pct,col_pct select focal point of zoom.

  1. fig, axs = plt.subplots(1,5,figsize=(12,4))
  2. for scale, ax in zip(np.linspace(1., 1.5,5), axs):
  3. get_ex().zoom(scale=scale).show(ax=ax, title=f'scale={scale:.2f}')

vision.transform - 图27

cutout[source][test]

cutout(x, n_holes:uniform_int=1, length:uniform_int=40) → Image :: TfmPixel No tests found for cutout. To contribute a test please refer to this guide and this discussion.

Cut out n_holes number of square holes of size length in image at random locations.

The normalization technique described in this paper: Improved Regularization of Convolutional Neural Networks with Cutout

By default, it will apply a single cutout (n_holes=1) of length=40) with probability p=1. The cutout position is always random. If you choose to do multiple cutouts, they may overlap.

The paper above used cutouts of size 16x16 for CIFAR-10 (10 categiries classification) and cutouts of size 8x8 for CIFAR-100 (100 categories). Generally, the more categories, the less cutout you want.

  1. tfms = [cutout()]
  2. fig, axs = plt.subplots(1,5,figsize=(12,4))
  3. for ax in axs:
  4. get_ex().apply_tfms(tfms).show(ax=ax)

vision.transform - 图28

You can add some randomness to the cutouts like this:

  1. tfms = [cutout(n_holes=(1,4), length=(10, 160), p=.5)]
  2. fig, axs = plt.subplots(1,5,figsize=(12,4))
  3. for ax in axs:
  4. get_ex().apply_tfms(tfms).show(ax=ax)

vision.transform - 图29

Convenience functions

These functions simplify creating random versions of crop_pad and zoom.

rand_crop[source][test]

rand_crop(*args, padding_mode='reflection', p:float=1.0) No tests found for rand_crop. To contribute a test please refer to this guide and this discussion.

Randomized version of crop_pad.

The args are for internal purposes and shouldn’t be touched.

  1. tfm = rand_crop()
  2. _, axs = plt.subplots(2,4,figsize=(12,6))
  3. for ax in axs.flatten():
  4. img = get_ex().apply_tfms(tfm, size=224)
  5. img.show(ax=ax)

vision.transform - 图30

rand_pad[source][test]

rand_pad(padding:int, size:int, mode:str='reflection') No tests found for rand_pad. To contribute a test please refer to this guide and this discussion.

Fixed mode padding and random crop of size

  1. tfm = rand_pad(4, 224)
  2. _, axs = plt.subplots(2,4,figsize=(12,6))
  3. for ax in axs.flatten():
  4. img = get_ex().apply_tfms(tfm, size=224)
  5. img.show(ax=ax)

vision.transform - 图31

rand_zoom[source][test]

rand_zoom(scale:uniform=1.0, p:float=1.0) No tests found for rand_zoom. To contribute a test please refer to this guide and this discussion.

Randomized version of zoom.

  1. tfm = rand_zoom(scale=(1.,1.5))
  2. _, axs = plt.subplots(2,4,figsize=(12,6))
  3. for ax in axs.flatten():
  4. img = get_ex().apply_tfms(tfm)
  5. img.show(ax=ax)

vision.transform - 图32


Company logo

©2021 fast.ai. All rights reserved.
Site last generated: Jan 5, 2021