Skip to content

Commit

Permalink
Adjust the order of package imports in boxam_vis_demo.py (#655)
Browse files Browse the repository at this point in the history
  • Loading branch information
RangeKing committed Mar 13, 2023
1 parent 3118eef commit cbf5a80
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions demo/boxam_vis_demo.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,18 +18,18 @@
from mmengine import Config, DictAction, MessageHub
from mmengine.utils import ProgressBar

from mmyolo.utils.boxam_utils import (BoxAMDetectorVisualizer,
BoxAMDetectorWrapper, DetAblationLayer,
DetBoxScoreTarget, GradCAM,
GradCAMPlusPlus, reshape_transform)
from mmyolo.utils.misc import get_file_list

try:
from pytorch_grad_cam import AblationCAM, EigenCAM
except ImportError:
raise ImportError('Please run `pip install "grad-cam"` to install '
'pytorch_grad_cam package.')

from mmyolo.utils.boxam_utils import (BoxAMDetectorVisualizer,
BoxAMDetectorWrapper, DetAblationLayer,
DetBoxScoreTarget, GradCAM,
GradCAMPlusPlus, reshape_transform)
from mmyolo.utils.misc import get_file_list

GRAD_FREE_METHOD_MAP = {
'ablationcam': AblationCAM,
'eigencam': EigenCAM,
Expand Down

0 comments on commit cbf5a80

Please sign in to comment.