search

arnoweng / CheXNet

A pytorch reimplementation of CheXNet
576 stars 221 forks source link

How to create the heatmap #3

Closed mengquann closed 6 years ago

mengquann commented 6 years ago

Your work is good. I want to know how to create the heatmap. Now the "output" variable in the code is a 10*14 mat, and it seems not the heatmap.

arnoweng commented 6 years ago

@mengquann Get feature maps from the last conv layer. You can refer this paper for more details.

smivv commented 6 years ago

@arnoweng could you please provide the code how you do that?

taewookim commented 6 years ago

ditto @smivv any code that can be shared?

arnoweng commented 6 years ago

@smivv @taewookim You can adapt the author's code as needed.