diff --git a/tools/feature_maps_visualization/fm_vis.py b/tools/feature_maps_visualization/fm_vis.py index d3bea2be3a8772bcf409683d598f8b4b3b2d94e1..efb983cadf1b3c192ba835fd521efb0d3fb904e5 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 72d1949aa09e94867654580c99329b8c764942b3..e11fe968a4659310919866599ac4599d5bc3a03d 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)