try:
    from setuptools import setup
except:
    from distutils.core import setup

setup(
    name='keras_augm_layer',
    version='1.0.6',
    author='Roman Sol (ZFTurbo)',
    packages=['keras_augm_layer', ],
    url='https://github.com/ZFTurbo/Keras-augmentation-layer',
    description='Keras implementation of layer which performs augmentations of images using GPU.',
    long_description='Keras implementation of layer which performs augmentations of images using GPU.'
                     'More details: https://github.com/ZFTurbo/Keras-augmentation-layer',
    install_requires=[
        "numpy",
        "tensorflow>=2.5.0",
        "tensorflow_addons>=0.14.0",
    ],
)
