code: def convert_rgb_to_real(prediction, targets):
    
    prediction = tf.squeeze(prediction, axis=-1)
    prediction = common_layers.convert_rgb_to_real(prediction)
    targets = common_layers.convert_rgb_to_real(targets)
    return prediction, targets
docs: Convert prediction and target from rgb to real.