Skip to content
Snippets Groups Projects
Commit fee2f98f authored by André Anjos's avatar André Anjos :speech_balloon: Committed by Daniel CARRON
Browse files

[engine.saliency] Fix imports

parent f0eaa87f
Branches
Tags
1 merge request!12Adds grad-cam support on classifiers
......@@ -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":
......
......@@ -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:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment