Я хочу изменить значение данных, загружаемых keras ImageDataGenerator - PullRequest
0 голосов
/ 06 мая 2020

Я хочу, чтобы мой набор данных изображения имел значения в диапазоне от -1 до 1. Мой код загрузки набора данных выглядит следующим образом:

import keras_preprocessing.image as ki

ig_i1 = ki.ImageDataGenerator(rescale=1/255.)
ig_i2 = ki.ImageDataGenerator(rescale=1/255.)
ig_o1 = ki.ImageDataGenerator(rescale=1/255.)
ig_o2 = ki.ImageDataGenerator(rescale=1/255.)
ig_o3 = ki.ImageDataGenerator(rescale=1/255.)

ig1 = ig_i1.flow_from_directory(route+'image/',shuffle=False,class_mode=None,batch_size=bs,target_size=[128,128])
ig2 = ig_i2.flow_from_directory(route+'lv_tile/',shuffle=False,class_mode=None,batch_size=bs,target_size=[128,128])
og1 = ig_o1.flow_from_directory(route+'diffuse/',shuffle=False,class_mode=None,batch_size=bs,target_size=[128,128])
og2 = ig_o2.flow_from_directory(route+'specular/',shuffle=False,class_mode=None,batch_size=bs,target_size=[128,128])
og3 = ig_o3.flow_from_directory(route+'lambertian_shadow/',shuffle=False,class_mode=None,batch_size=bs,target_size=[128,128],color_mode='grayscale')

Здесь я хочу, чтобы ig2 имел значение в диапазоне от * От 1004 * -1 до 1 , но я не смог найти никакого способа сделать это.

Пожалуйста, научите меня, и большое спасибо.

1 Ответ

1 голос
/ 06 мая 2020

вы должны сделать стандартизацию вместо нормализации

ig_i2 = ki.ImageDataGenerator(featurewise_center=True, featurewise_std_normalization=True)

источник

...