Segmenting images using top activated units

[RaphaelGinoulhac]

Hi,
I've read your extended paper (https://arxiv.org/pdf/1711.05611.pdf) and reproduced your main results.
What I'm looking for is segmenting images using top activated units (section 3.10 and fig.27 in the paper). I don't think it has been implemented in this repo, but I guess it is in the original NetDissect?
If it has been implemented, could you give me the link to where it's uploaded (compatible with PyTorch would be best)? Or else, do you have ideas on how to efficiently code it?