cam should be initialized to zero matrix. Modify example images

dbcool · Jan 7, 2018 · 5153665 · 5153665
1 parent e701e5c
commit 5153665
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 1 deletion.
diff --git a/examples/cat.jpg b/examples/cat.jpg
diff --git a/examples/dog.jpg b/examples/dog.jpg
diff --git a/grad-cam.py b/grad-cam.py
@@ -109,7 +109,7 @@ def __call__(self, input, index = None):
 		target = target.cpu().data.numpy()[0, :]
 
 		weights = np.mean(grads_val, axis = (2, 3))[0, :]
-		cam = np.ones(target.shape[1 : ], dtype = np.float32)
+		cam = np.zeros(target.shape[1 : ], dtype = np.float32)
 
 		for i, w in enumerate(weights):
 			cam += w * target[i, :, :]