search

utkuozbulak / pytorch-cnn-visualizations

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

Can this tool be used on non-classification tasks #119

Closed sdu-huanhq closed 1 year ago

sdu-huanhq commented 1 year ago

Hello, First of all, thank you for your excellent work. But I have a question: Can I use this tool to simply observe intermediate features (i.e. which part of the network pays more attention to), and neural networks are not used for tasks related to classification (such as image classification, semantic segmentation, etc.)? Looking forward to your replay. Best wishes!

utkuozbulak commented 1 year ago

You could make some of the techniques work for types of problems out of classification but I'm not sure if what you observe will be meaningful.