Categories
Misc

Easy way to create augmented tf.data.Dataset (generator) for image.

This package makes it easy for us to create efficient image Dataset generator.

github link

Features

  • Simple, Easy and Efficient image dataset creator for segmentation and classification models.
  • Avoiding some limitations which cause performance bottleneck of learning, generate the Dataset by one-liner.
  • Augment multiple input images and multiple label images with the same transformations.
  • Adjusting sampling ratios from multiple tfrecord files are supported.

Install

python -m pip install git+https://github.com/piyop/tfaug

Supported Augmentations

  • standardize
  • resize
  • random_rotation
  • random_flip_left_right
  • random_flip_up_down
  • random_shift
  • random_zoom
  • random_shear
  • random_brightness
  • random_saturation
  • random_hue
  • random_contrast
  • random_crop
  • random_noise

submitted by /u/last_peng
[visit reddit] [comments]

Leave a Reply

Your email address will not be published. Required fields are marked *