diff --git a/explainer/explainer.py b/explainer/explainer.py index 5a6140633bfb..4d2eff1ce48e 100644 --- a/explainer/explainer.py +++ b/explainer/explainer.py @@ -11,11 +11,11 @@ import sys from pathlib import Path +import cv2 import numpy as np import torch import torch.nn.functional as F import torchvision -import cv2 from pytorch_grad_cam import (AblationCAM, EigenCAM, EigenGradCAM, FullGrad, GradCAM, GradCAMElementWise, GradCAMPlusPlus, HiResCAM, LayerCAM, RandomCAM, ScoreCAM, XGradCAM) from pytorch_grad_cam.utils.image import scale_cam_image, show_cam_on_image