keras channel_shift_range example

channel_shift_range_example.py

import keras
import numpy as np
from keras.preprocessing.image import ImageDataGenerator

#data
data = np.array([[0,0,0],[10,10,10],[100,100,100]])
print(data)

#ImageDataGenerator
expand_data = np.expand_dims(data, axis = 0)
augmented_data = []
datagen = ImageDataGenerator(channel_shift_range = 10) 

for data in datagen.flow(np.expand_dims(expand_data, axis = 0)):
    augmented_data.append(data) 
    break;
    
print(augmented_data[0])
print(augmented_data[0][0] - expand_data)