如果你从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))
如果还是不行,则尝试大牛的博客