From 3e0610884bb23f7b7ffe5170e4ac12bf8097efb3 Mon Sep 17 00:00:00 2001 From: Tobias Huber Date: Mon, 26 Apr 2021 14:29:52 +0200 Subject: [PATCH] fixed the size of the score array which will later be interpolated to create the final saliency map. This array was too big for d with 80 % d = 0. For example for d=1, the socres array had size 81x81 instead of the intended 80x80(one value for each pixel) --- saliency.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/saliency.py b/saliency.py index 70b28e78..2aba3b30 100755 --- a/saliency.py +++ b/saliency.py @@ -41,7 +41,7 @@ def score_frame(model, history, ix, r, d, interp_func, mode='actor'): # if d==2 then every other, which is 25% of total pixels for a 2D image) assert mode in ['actor', 'critic'], 'mode must be either "actor" or "critic"' L = run_through_model(model, history, ix, interp_func, mask=None, mode=mode) - scores = np.zeros((int(80/d)+1,int(80/d)+1)) # saliency scores S(t,i,j) + scores = np.zeros((int(79/d)+1,int(79/d)+1)) # saliency scores S(t,i,j) for i in range(0,80,d): for j in range(0,80,d): mask = get_mask(center=[i,j], size=[80,80], r=r)