Could you please provide the feature importance included dataset that has been generated?

utkuozbulak / pytorch-cnn-visualizations

Pytorch implementation of convolutional neural network visualization techniques

MIT License

7.81k stars 1.49k forks source link

Could you please provide the feature importance included dataset that has been generated? #110

Closed bosaiiasliu closed 2 years ago

bosaiiasliu commented 2 years ago

I found that using some methods like CAM to generate feature importance interpretations is slow, taking about 2 seconds for one image. If I want to generate all the image explanations in the Imagenet dataset, the time it takes seems very long. Or, is there a suitable acceleration method?

utkuozbulak commented 2 years ago

This is intended. The code is designed to work in cpu not cuda (so that it runs for everyone, even those who do not have a dedicated gpu). You should convert the parts that work on cpu to work on gpu to accelerate the speed.