How to get the final Anomaly Prediction image which you present in your paper?

[pswena]

How could I get the image like this which presented in your paper

[pswena]

When I test demo.py, I get the anomaly prediction picture like this .But how to get the picture like what you presented in your paper?

[carlosdavidardon]

I haven't try it myself. But if you go to the Google Colab Demo Notebook ( https://colab.research.google.com/drive/1HQheunEWYHvOJhQQiWbQ9oHXCNi9Frfl?usp=sharing#scrollTo=gC-ViJmm23eM ) in section "Get and Display Perceptual Difference" they used this code:

heatmap_prediction = cv2.applyColorMap((255-diss_pred), cv2.COLORMAP_JET)
heatmap_pred_im = Image.fromarray(heatmap_prediction).resize((2048, 1024))
combined_image = Image.blend(img_og, heatmap_pred_im, alpha=.5)

display(combined_image)