Visualize Areas of Interest

vision
1

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:

download

Thank you :slight_smile:

2

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.