search

cnulab / RealNet

Offical implementation of "RealNet: A Feature Selection Network with Realistic Synthetic Anomaly for Anomaly Detection (CVPR 2024)"
MIT License
143 stars 5 forks source link

有关异常图像生成的问题 #35

Closed houruixiang-1024 closed 3 weeks ago

houruixiang-1024 commented 3 weeks ago

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

cnulab commented 3 weeks ago

你好！ 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')
houruixiang-1024 commented 3 weeks ago

感谢您的回复!😀