kazuto1011 / grad-cam-pytorch

PyTorch re-implementation of Grad-CAM (+ vanilla/guided backpropagation, deconvnet, and occlusion sensitivity maps)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Process multiple image inputs at the same time

oyxhust opened this issue · comments

I find that the current code can only support the single image input. Can I input a batch of images to get the grad-cam outputs?

Updated #9! Using my sample, now you can specify batch images like:

python main.py demo1 -a resnet152 \
                     -t layer4 \
                     -i a.png -i b.jpg -i c.bmp -i ...

Or you can import the wrapper classes into your projects.