From bf79ba96bec87e3eb30c57636ef3d60ff4afe278 Mon Sep 17 00:00:00 2001 From: KaiyangZhou Date: Wed, 27 Nov 2019 17:00:18 +0000 Subject: [PATCH] bug fix in tools/visactmap --- tools/visualize_actmap.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/visualize_actmap.py b/tools/visualize_actmap.py index d88ae8c..278d25e 100644 --- a/tools/visualize_actmap.py +++ b/tools/visualize_actmap.py @@ -23,7 +23,7 @@ GRID_SPACING = 10 @torch.no_grad() -def visactmap(model, test_loader, save_dir, width, height, print_freq, use_gpu, img_mean=None, img_std=None): +def visactmap(model, test_loader, save_dir, width, height, use_gpu, img_mean=None, img_std=None): if img_mean is None or img_std is None: # use imagenet mean and std img_mean = IMAGENET_MEAN