Visualize Areas of Interest

Hello Everyone,

I have recently created a PyTorch model, which is able to detect brain tumours from CT Images. However, I want to highlight the tumor or what the model perceives as a tumor in a color or perhaps just a frame around it. Is this possible using torch vision?

Possibly looking like this:


Thank you :slight_smile:

You can use the generated binary map, overlay it using alpha blending in opencv and get the closest highlight possible. Let me know if you need the code for it.