diff --git a/saliency_nodes.py b/saliency_nodes.py index b4d98e7..63f849f 100644 --- a/saliency_nodes.py +++ b/saliency_nodes.py @@ -132,7 +132,7 @@ def _make_call_model_fn(model, class_idx): images_np = np.array(images, dtype=np.float32) images_tensor = torch.from_numpy(images_np).permute(0, 3, 1, 2) - mmean = torch.tensor(IMAGENET_MEAN).view(1, 3, 1, 1).to(DEVICE) + mean = torch.tensor(IMAGENET_MEAN).view(1, 3, 1, 1).to(DEVICE) std = torch.tensor(IMAGENET_STD).view(1, 3, 1, 1).to(DEVICE) images_tensor = images_tensor.to(DEVICE) images_tensor = (images_tensor - mean) / std