haofanwang / Score-CAM

Official implementation of Score-CAM in PyTorch
MIT License
399 stars 66 forks source link

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

Closed bosaiiasliu closed 2 years ago

bosaiiasliu commented 2 years ago

I found that using ScoreCAM 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?

haofanwang commented 2 years ago

Hi, @bosaiiasliu

I found a speed-up version of Score-CAM, which only selected channels with large variances as mask images. By the way, why you want to generate features over the whole ImageNet? In most of case, we just randomly sample a small portion from it.

bosaiiasliu commented 2 years ago

Because I found through experiments on CIFAR10 that pictures with feature importance are likely to be a useful data augmentation method. I want to try it on ImageNet. Unfortunately, I found that someone has already done something similar. https://arxiv.org/abs/2006.01791