tensor 转 rgb图像发现特别暗? 尝试反normalize

阅读 85

2022-04-01

如果你从tensor生成的rgb图像非常暗,那么可能是由于你对图片进行了normalize, 可以尝试在生成rgb图像之前这么操作

def inverse_normalize(tensor, mean, std):
    for t, m, s in zip(tensor, mean, std):
        t.mul_(s).add_(m)
    return tensor

input = inverse_normalize(tensor=input, mean=(0.485, 0.456, 0.406), std=(0.229, 0.224, 0.225))

精彩评论(0)

0 0 举报