diff --git a/MANIFEST.in b/MANIFEST.in new file mode 100644 index 0000000..2192166 --- /dev/null +++ b/MANIFEST.in @@ -0,0 +1,4 @@ +include MANIFEST.in +include README.md +include LICENSE +include requirements.txt diff --git a/README.md b/README.md index 27ca00c..dc74068 100644 --- a/README.md +++ b/README.md @@ -1,12 +1,21 @@ # CutMixImageDataGenerator (Keras) -![Python-v3.7.3](https://img.shields.io/badge/Python-v3.7.3-blue) -![Keras-v2.2.4](https://img.shields.io/badge/Keras-v2.2.4-brightgreen) +![GitHub release (latest by date)](https://img.shields.io/github/v/release/DevBruce/CutMixImageDataGenerator_For_Keras) -**Version 0.1 (Beta)** +> Paper: [CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Features](https://arxiv.org/abs/1905.04899)
+## Install + +```bash +$ pip install cutmix-keras +``` + +
+ +## How To Use + ```python class CutMixImageDataGenerator(): def __init__(self, generator1, generator2, img_size, batch_size): @@ -33,9 +42,13 @@ So there would no augmentation
-## Using Example +### Using Example ```python +# (some codes) ... +from cutmix_keras import CutMixImageDataGenerator # Import CutMix + + train_datagen = ImageDataGenerator( rescale=1./255, ) diff --git a/CutMixImageDataGenerator.py b/cutmix_keras.py similarity index 100% rename from CutMixImageDataGenerator.py rename to cutmix_keras.py diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..e69de29 diff --git a/setup.py b/setup.py new file mode 100644 index 0000000..bd474de --- /dev/null +++ b/setup.py @@ -0,0 +1,21 @@ +from setuptools import setup +import os + + +ROOT_DIR = os.path.dirname(os.path.abspath(__file__)) +README_PATH = os.path.join(ROOT_DIR, 'README.md') + +setup( + name='cutmix-keras', + author='devbruce', + author_email='bruce93k@gmail.com', + description='CutMuxImageDataGenerator For Keras', + long_description=open(README_PATH, encoding='utf-8').read(), + long_description_content_type='text/markdown', + license='MIT', + version='1.0.0', + url='https://github.com/DevBruce/CutMixImageDataGenerator_For_Keras', + py_modules=['cutmix_keras'], + keywords=['cutmix', 'keras'], + # install_requires=['keras'], +)