From 42090edfba7e90180a316fcc7d63173976117329 Mon Sep 17 00:00:00 2001 From: Konstantin Baierer Date: Fri, 21 Aug 2020 15:37:13 +0200 Subject: [PATCH] heatmap.py --help: grad_cam -> heatmap --- heatmap.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/heatmap.py b/heatmap.py index ebbaea6..29b31e9 100644 --- a/heatmap.py +++ b/heatmap.py @@ -210,7 +210,7 @@ def get_args(): return args if __name__ == '__main__': - """ python grad_cam.py + """ python heatmap.py 1. Loads an image with opencv. 2. Preprocesses it for VGG19 and converts to a pytorch variable. 3. Makes a forward pass to find the category index with the highest score,