From cfa709aac458e55ac214f7db4cdc75113f94724f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adam=20Wawrzy=C5=84ski?= Date: Wed, 28 Dec 2022 16:10:22 +0100 Subject: [PATCH] Remove incorrect standardization method MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Adam Wawrzyński --- autoxai/explainer/base_explainer.py | 5 ----- autoxai/explainer/model_utils.py | 13 ------------- 2 files changed, 18 deletions(-) diff --git a/autoxai/explainer/base_explainer.py b/autoxai/explainer/base_explainer.py index 493be381..f847b743 100644 --- a/autoxai/explainer/base_explainer.py +++ b/autoxai/explainer/base_explainer.py @@ -6,8 +6,6 @@ import torch from captum.attr import visualization as viz -from autoxai.explainer.model_utils import standardize_matrix - class CVExplainer(ABC): """Abstract explainer class.""" @@ -82,9 +80,6 @@ def visualize( if len(transformed_img.shape) >= 3: transformed_img_np = np.transpose(transformed_img_np, (1, 2, 0)) - attributions_np = standardize_matrix(matrix=attributions_np) - transformed_img_np = standardize_matrix(matrix=transformed_img_np) - figure, _ = viz.visualize_image_attr_multiple( attr=attributions_np, original_image=transformed_img_np, diff --git a/autoxai/explainer/model_utils.py b/autoxai/explainer/model_utils.py index 6c0a43ef..88e47ef7 100644 --- a/autoxai/explainer/model_utils.py +++ b/autoxai/explainer/model_utils.py @@ -1,6 +1,5 @@ """File contains functions to modifiy DNN models.""" -import numpy as np import torch @@ -21,15 +20,3 @@ def modify_modules(model: torch.nn.Module) -> torch.nn.Module: module.inplace = False return model - - -def standardize_matrix(matrix: np.ndarray) -> np.ndarray: - """Standardize matrix to range [0-1]. - - Args: - matrix: Any matrix. - - Returns: - Standardize matrix. - """ - return (matrix - np.min(matrix)) / np.max(matrix)