search

utkuozbulak / pytorch-cnn-visualizations

Pytorch implementation of convolutional neural network visualization techniques
MIT License
7.81k stars 1.49k forks source link

Functionality requests #73

Closed cianeastwood closed 4 years ago

cianeastwood commented 4 years ago

Fantastic repo!

If you had time, would be nice to add the following for completeness:

  1. "Top 9 images" function: returns top 9 images for a given filter (highest activations), showing them in a 3x3 grid
  2. "Selected channel" function: spits out an enlarged version of channel activations for a given filter

These functions are depicted in "Understanding Neural Networks Through Deep Visualization", Figure 1.

They are implemented in Caffe in Deep Visualisation Toolbox, but it would be nice to have them all together in this neat PyTorch repo.

Cheers :)

C

cianeastwood commented 4 years ago

nevermind!

utkuozbulak commented 4 years ago

Hey @cianeastwood, unfortunately I do not have necessary free time to implement those techniques. Feel free to contribute to the repository though.