Как заставить keras ImageGenerator правильно загружать 16-битные изображения? - PullRequest
2 голосов
/ 25 сентября 2019

Я искал похожие вопросы, но не нашел решения того, что я пытаюсь сделать.У меня есть 16-битные изображения в градациях серого, и я пытаюсь вписать их в Keras ImageDataGenerator.При использовании таких функций, как: flow_from_dataframe, он выдает изображения с одинаковым значением пикселя (неверно).

Я пытался использовать keras preprocess_input, масштабировать до [0,1], до [-1,1] с пользовательской функцией предварительной обработки, но ничего из этого не сработало.Я также установил color_mode = 'grayscale' в ImageDataGenerator.

Далее я проверил преобразование в 8-битное, и это сработало.Я утроил количество каналов, и это не было проблемой, так как он все еще работал с 8-битным.Я читал, что в этом случае keras использует библиотеку PIL для чтения изображений, и, поскольку он не обрабатывает 16-битный код правильно, он возвращает эту ошибку.Я видел, что мы можем установить другую библиотеку для загрузки изображений, но я не знаю, как это сделать.

Кто-нибудь знает альтернативу для использования 16-битных изображений?В последнем случае я бы попробовал собственный генератор, но мне бы очень хотелось получить выгоду от уже созданных и протестированных для этой цели функций.

Я хочу использовать эти изображения для тонкой настройки предварительно обученной сети., поэтому я хотел бы стандартизировать тип ввода, который я использую.

Спасибо.

...