Add implementation for mini-batch computation of Score-CAM

haofanwang / Score-CAM

Official implementation of Score-CAM in PyTorch

MIT License

399 stars 66 forks source link

Add implementation for mini-batch computation of Score-CAM #9

Closed YuhengHuang42 closed 3 years ago

YuhengHuang42 commented 3 years ago

test_batch.py gives the test result for images in /images file

YuhengHuang42 commented 3 years ago

Please feel free to close this pull request. It seems that there is another implementation which does further optimization. However in this PR there is a test_batch.py file which may help you test the code.