search

DSGT-DLP / Deep-Learning-Playground

Web Application where people new to Deep Learning can input a dataset and toy around with basic Pytorch modules without writing any code
MIT License
25 stars 8 forks source link

Visualizations of significant regions in Image Models #433

Open hassannaveed1997 opened 2 years ago

hassannaveed1997 commented 2 years ago

Since we have image model support, it would be nice to have some visualizations of what areas of the image are important.

I don't know about the front end part, but I was thinking once training has finished there would be an additional section un Deep Learning Results. There, the user can select the image they are interested in, and see visualizations of significant regions.

GradCAM or Guided BackProp can be used for that. Like this:

image

This can either be implemented manually (see https://arxiv.org/abs/1610.02391) or using a library called captum.

karkir0003 commented 1 year ago

How is progress going on this? @hassannaveed1997