0
点赞
收藏
分享

微信扫一扫

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

飞鸟不急 2022-04-01 阅读 85

如果你从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 条评论