Модель работает без ошибок, но выходное изображение имеет заметный красноватый/фиолетовый оттенок вместо правильных цветов.
Код: Выделить всё
import tensorflow as tf
import matplotlib.pyplot as plt
model = tf.saved_model.load('saved_model')
infer = model.signatures['serving_default']
image_1 = tf.io.decode_image(tf.io.read_file('/content/0001x4m.png'))
image = tf.cast(image_1, tf.float32) / 127.5 - 1.0
image = tf.expand_dims(image, axis=0)
result = infer(image)
result = list(result.values())[0]
print(result.shape) #TensorShape([1, 1404, 2040, 3])
res = tf.squeeze(result, axis=0)
res = tf.clip_by_value(res, -1.0, 1.0)
sr_image = (res + 1.0) * 127.5
sr_image = tf.cast(sr_image, tf.uint8)
plt.imshow(sr_image)
plt.show()
Я пробовал перейти с RGB на BGR, я также пробовал другую нормализацию и другой метод масштабирования, но все не работает.
это входное изображение после предварительной обработки и перед подачей его в модель
введите здесь описание изображения
Подробнее здесь: https://stackoverflow.com/questions/797 ... processing
