Skip to content

Augmentation

Pre-release
Pre-release

Choose a tag to compare

@james-nvidia james-nvidia released this 12 Mar 23:24
2db6b07

This release adds image augmentation (brightness, contrast, hue, saturation) and four degree rotational augmentation.

Added parameters:

  • --augment-rotate: Randomly rotates the training images by 0°, 90°, 180° or 270°.
  • --augment-brightness (float): Randomly adjusts brightness of image. The value sets the standard deviation of a Gaussian distribution. The degree of augmentation is selected from this distribution. Default: 0.05
  • --augment-contrast (float): Randomly adjusts contrast of image. The value sets the standard deviation of a Gaussian distribution. The degree of augmentation is selected from this distribution. Default: 0.05
  • --augment-hue (float): Randomly adjusts hue of image. The value sets the standard deviation of a Gaussian distribution. The degree of augmentation is selected from this distribution. Default: 0.01
  • --augment-saturation (float): Randomly adjusts saturation of image. The value sets the standard deviation of a Gaussian distribution. The degree of augmentation is selected from this distribution. Default: 0.05
  • --regularization-l2 (float): Sets the L2 regularization of the optimizer. Default: 0.0001