CutMuxImageDataGenerator For Keras
Project description
CutMixImageDataGenerator (Keras)
Paper: CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Features
Install
$ pip install cutmix-keras
How To Use
class CutMixImageDataGenerator(): def __init__(self, generator1, generator2, img_size, batch_size): self.batch_index = 0 self.samples = generator1.samples self.class_indices = generator1.class_indices self.generator1 = generator1 self.generator2 = generator2 . . .
generator1
,generator2
need same generator appliedflow
method
generator1
,generator2
needshuffle=True
Ifshuffle=False
, This generator cutmix with same images.
So there would no augmentation
- Why are there two same generators? (
generator1
,generator2
)
--> To Solve Reference Problem
Using Example
# (some codes) ... from cutmix_keras import CutMixImageDataGenerator # Import CutMix train_datagen = ImageDataGenerator( rescale=1./255, ) train_generator1 = train_datagen.flow_from_dataframe( dataframe=X_train, directory=IMG_DIR, target_size=(IMG_SIZE, IMG_SIZE), x_col='X_Column', y_col='Y_Column', color_mode='rgb', class_mode='categorical', batch_size=BATCH_SIZE, shuffle=True, # Required ) train_generator2 = train_datagen.flow_from_dataframe( dataframe=X_train, directory=IMG_DIR, target_size=(IMG_SIZE, IMG_SIZE), x_col='X_Column', y_col='Y_Column', color_mode='rgb', class_mode='categorical', batch_size=BATCH_SIZE, shuffle=True, # Required ) # CutMixImageDataGenerator train_generator = CutMixImageDataGenerator( generator1=train_generator1, generator2=train_generator2, img_size=IMG_SIZE, batch_size=BATCH_SIZE, )
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Filename, size | File type | Python version | Upload date | Hashes |
---|---|---|---|---|
Filename, size cutmix_keras-1.0.0-py3-none-any.whl (3.6 kB) | File type Wheel | Python version py3 | Upload date | Hashes View |
Close
Hashes for cutmix_keras-1.0.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | ab31f7169394a94f13f1b22caec81f364b97000601d65f8d54b90d371e9b085d |
|
MD5 | 676401ae1f0f4574bb065ae2b3780eac |
|
BLAKE2-256 | 980f6678894736e65537ab42cd4f32dace6748cfd013a4bff2c8ac1f25a29d43 |