cnulab/RealNet

有关异常图像生成的问题

Closed this issue · 2 comments

我在realnet.py中尝试采用下面方面进行异常图片的保存
image
很遗憾的是保存的图片仿佛存在一些问题,可能是经过了某种处理
image
生成的sdas异常图片仿佛也经过了某种处理
image

你好!
train_loader中的image经过了normalize,要想还原为可保存的图像需要以下步骤:

    from PIL import Image
    for data in train_loader:
        image = data['image'][0]
        image = image.permute(1,2,0) * torch.from_numpy(np.array([[0.229, 0.224, 0.225]])) + torch.from_numpy(np.array([[[0.485, 0.456, 0.406]]]))
        image = image * 255
        Image.fromarray( image.cpu().numpy().astype(np.uint8)).save('image_name.jpg')

感谢您的回复!😀