将预测张量转换为图像

import numpy as np

def tensor_to_image(tensor):
    # 将张量归一化到0-255范围
    tensor = np.clip(tensor, 0, 255)
    # 转换为uint8类型
    tensor = tensor.astype(np.uint8)
    return tensor

def reshape_tensor_to_image(tensor, target_shape):
    # 确保张量形状与目标形状一致
    tensor = tensor.reshape(target_shape)
    return tensor

def convert_channels(tensor, target_channels='RGB'):
    if tensor.shape[-1] == 1:  # 灰度图像转RGB
        tensor = np.repeat(tensor, 3, axis=-1)
    elif tensor.shape[-1] == 4 and target_channels == 'RGB':  # RGBA转RGB
        tensor = tensor[..., :3]
    return tensor