From fee2f98f23d4f6f866ea4e8522866c83dae849f0 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Tue, 10 Oct 2023 22:19:28 +0200
Subject: [PATCH] [engine.saliency] Fix imports
---
src/ptbench/engine/saliency/completeness.py | 6 +++---
src/ptbench/engine/saliency/generator.py | 4 ++--
2 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/src/ptbench/engine/saliency/completeness.py b/src/ptbench/engine/saliency/completeness.py
index 87778d74..a75d57b2 100644
--- a/src/ptbench/engine/saliency/completeness.py
+++ b/src/ptbench/engine/saliency/completeness.py
@@ -182,9 +182,9 @@ def run(
* combined: ROAD combined score by evaluating ``(lerf-morf)/2``.
"""
- from ..models.densenet import Densenet
- from ..models.pasa import Pasa
- from .saliencymap_generator import _create_saliency_map_callable
+ from ...models.densenet import Densenet
+ from ...models.pasa import Pasa
+ from .generator import _create_saliency_map_callable
if isinstance(model, Pasa):
if saliency_map_algorithm == "fullgrad":
diff --git a/src/ptbench/engine/saliency/generator.py b/src/ptbench/engine/saliency/generator.py
index eda78ed4..5d87167a 100644
--- a/src/ptbench/engine/saliency/generator.py
+++ b/src/ptbench/engine/saliency/generator.py
@@ -136,8 +136,8 @@ def run(
"""
from pytorch_grad_cam.utils.model_targets import ClassifierOutputTarget
- from ..models.densenet import Densenet
- from ..models.pasa import Pasa
+ from ...models.densenet import Densenet
+ from ...models.pasa import Pasa
if isinstance(model, Pasa):
if "fullgrad" in saliency_map_algorithms:
--
GitLab