keras ImageDataGenerator.flow с color_mode

Question

Я использую python с keras и хочу использовать предварительную обработку изображений keras, некоторые из моих изображений - rgb, а некоторые - в оттенках серого, мне нужно прочитать все из них и изменить размер изображений в оттенках серого на x, x,3 или избавиться от них, и хотел бы сделать это как часть функции .flow так же, как я мог бы использовать color_mode с .flow_from_directory, который можно установить в rgb и читать все изображения как rgb, даже если они в градациях серого,это возможно?

Answer 1

Вы не можете сделать это как часть .flow, потому что это предполагает, что вы уже загрузили подготовленные изображения в тензор 4D.Вы можете использовать функцию load_img , которую использует .flow_from_directory, которая фактически использует аргумент color_mode:

img = load_img(os.path.join(self.directory, fname),
               color_mode=self.color_mode,
               target_size=self.target_size,
               interpolation=self.interpolation)

Это из кода flow_from_directory .Вы можете использовать эту функцию для загрузки изображений, а затем вызвать .flow.