Я делаю проект о суперразрешении, поэтому я получаю изображение в качестве ввода и возвращаю большее изображение (x2, x4) в качестве вывода. Я пытаюсь загрузить модель на платформу AI , но у меня возникают проблемы с комбинациями GPU - альтернативы прогнозирования.
Мне удалось сделать пользовательский прогноз Код работает без проблем, вход и выход представляет собой изображение base64 jpg. Но я не могу использовать машины с графическим процессором с такими прогнозами.
Я попытался сгенерировать Оценщик из моей модели keras , а затем создать SavedModel для онлайн-прогнозирования но я могу определить только обслуживающую функцию ввода для декодирования изображения jpg, но я не могу определить эквивалентную функцию для вывода, поэтому платформа онлайн-прогнозирования возвращает json файлов необработанных изображений, которые действительно большие ( для ввода jpg-изображения 13K он возвращает необработанное изображение размером 19 МБ)
Есть ли способ определения шагов постпрогнозирования в SaveModel, чтобы можно было кодировать необработанное изображение в jpg и до ответа на запрос случаи? Я хотел бы использовать машины с графическим процессором, которые в настоящее время недоступны для пользовательских процедур прогнозирования.