Just to correct a typo of 'target' (#1655)

pull/1717/head
BBC 2023-07-14 04:16:35 -04:00 committed by GitHub
parent db395d35b1
commit 569324b180
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -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)