mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tools/visualization/vis_cam.py
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) 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 """get default target layers from given model, here choose nrom type layer
as default target layer.""" as default target layer."""
norm_layers = [ norm_layers = [
@ -241,7 +241,7 @@ def main():
get_layer(layer, model) for layer in args.target_layers get_layer(layer, model) for layer in args.target_layers
] ]
else: else:
target_layers = get_default_traget_layers(model, args) target_layers = get_default_target_layers(model, args)
# init a cam grad calculator # init a cam grad calculator
use_cuda = ('cuda' in args.device) use_cuda = ('cuda' in args.device)