Блог

Я новичок в глубоком обучении и в настоящее время пытаюсь обучить классификатору с помощью КТ-сканирования. Я использую Keras ImageDataGenerator для выполнения быстрого увеличения своих изображений и включения более надежного классификатора.

 train_datagen = ImageDataGenerator(shear_range = 0.1,
                                   rotation_range = 15,
                                   zoom_range = 0.15,
                                   width_shift_range = 0.1,
                                   height_shift_range = 0.1,
                                   horizontal_flip = False, 
                                   vertical_flip = False, 
                                   preprocessing_function = preprocess_input)
 

Помимо случайного масштабирования, сдвига и сдвига, как описано в коде выше, я также хочу добавить приложение random CLAHE, как это сделать проще всего?

Проверьте этот метод :

 def AHE(img):
    img_adapteq = exposure.equalize_adapthist(img, clip_limit=0.03)
    return img_adapteq


datagen = ImageDataGenerator(rotation_range=30, horizontal_flip=0.5, preprocessing_function=AHE)
 

Счастливого кодирования!