search

haofanwang / Score-CAM

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

scorecam of batch size more than 1 #1

Closed umairjavaid closed 4 years ago

umairjavaid commented 4 years ago

How are you generating cams when batch_size is more than one? In your implementation there is only one loop which iterates over number of activations but no loop for when batch_size is more than one. Your implementation only returns one activation, I am assuming it does not return multiple cams, when multiple images are given as input

haofanwang commented 4 years ago

Yes, @umairjavaid batch generation is not supported now. I will add it later. If you are interested in, it would be appreciated to PR.

umairjavaid commented 4 years ago

Actually, yes, I have implemented scorecam with batch generation. I am new to github, can u guide me how to contribute. Thank you

haofanwang commented 4 years ago

That's great @umairjavaid. If your implementation is based on mine, then you can directly click "New pull request" and I can see it.