From 0644c970a6fd581bfc6818868467b6a5dac25af9 Mon Sep 17 00:00:00 2001
From: wqz960 <362379625@qq.com>
Date: Sat, 4 Jul 2020 05:19:09 +0000
Subject: [PATCH] add feature maps visualization

---
 tools/feature_maps_visualization/fm_vis.py      | 1 -
 tools/feature_maps_visualization/get_started.md | 7 ++++++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/tools/feature_maps_visualization/fm_vis.py b/tools/feature_maps_visualization/fm_vis.py
index d3bea2be..efb983ca 100644
--- a/tools/feature_maps_visualization/fm_vis.py
+++ b/tools/feature_maps_visualization/fm_vis.py
@@ -19,7 +19,6 @@ import numpy as np
 import cv2 
 import utils
 import argparse
-import matplotlib.pyplot as plt 
 
 def parse_args():
     def str2bool(v):
diff --git a/tools/feature_maps_visualization/get_started.md b/tools/feature_maps_visualization/get_started.md
index 72d1949a..e11fe968 100644
--- a/tools/feature_maps_visualization/get_started.md
+++ b/tools/feature_maps_visualization/get_started.md
@@ -23,6 +23,8 @@ tar -xf ResNet50_pretrained.tar
 
 找到我们所需要的特征图位置，设置self.fm将其fetch出来，本文以resnet50中的stem层之后的特征图为例。
 
+在fm_vis.py中修改模型的名字。
+
 在ResNet50的__init__函数中定义self.fm
 ```python
 self.fm = None
@@ -60,6 +62,9 @@ python tools/feature_maps_visualization/fm_vis.py -i the image you want to test
 
 ## 四、结果
 输入图片：  
+
 ![](../../tools/feature_maps_visualization/test.jpg)  
+
 输出特征图：  
-![](../../tools/feature_maps_visualization/fm.jpg)  
\ No newline at end of file
+
+![](../../tools/feature_maps_visualization/fm.jpg)
-- 
GitLab