From 569324b180b73dde3775c1e83130a4a3ed1a75aa Mon Sep 17 00:00:00 2001
From: BBC <bryan.bo.cao@outlook.com>
Date: Fri, 14 Jul 2023 04:16:35 -0400
Subject: [PATCH] Just to correct a typo of 'target' (#1655)
---
tools/visualization/vis_cam.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tools/visualization/vis_cam.py b/tools/visualization/vis_cam.py
index dd9de942..21223902 100644
--- a/tools/visualization/vis_cam.py
+++ b/tools/visualization/vis_cam.py
@@ -183,7 +183,7 @@ def show_cam_grad(grayscale_cam, src_img, title, out_path=None):
mmcv.imshow(visualization_img, win_name=title)
-def get_default_traget_layers(model, args):
+def get_default_target_layers(model, args):
"""get default target layers from given model, here choose nrom type layer
as default target layer."""
norm_layers = [
@@ -241,7 +241,7 @@ def main():
get_layer(layer, model) for layer in args.target_layers
]
else:
- target_layers = get_default_traget_layers(model, args)
+ target_layers = get_default_target_layers(model, args)
# init a cam grad calculator
use_cuda = ('cuda' in args.device)