提交 0644c970 编写于 作者: W wqz960

add feature maps visualization

上级 f9f8d032
...@@ -19,7 +19,6 @@ import numpy as np ...@@ -19,7 +19,6 @@ import numpy as np
import cv2 import cv2
import utils import utils
import argparse import argparse
import matplotlib.pyplot as plt
def parse_args(): def parse_args():
def str2bool(v): def str2bool(v):
......
...@@ -23,6 +23,8 @@ tar -xf ResNet50_pretrained.tar ...@@ -23,6 +23,8 @@ tar -xf ResNet50_pretrained.tar
找到我们所需要的特征图位置,设置self.fm将其fetch出来,本文以resnet50中的stem层之后的特征图为例。 找到我们所需要的特征图位置,设置self.fm将其fetch出来,本文以resnet50中的stem层之后的特征图为例。
在fm_vis.py中修改模型的名字。
在ResNet50的__init__函数中定义self.fm 在ResNet50的__init__函数中定义self.fm
```python ```python
self.fm = None self.fm = None
...@@ -60,6 +62,9 @@ python tools/feature_maps_visualization/fm_vis.py -i the image you want to test ...@@ -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/test.jpg)
输出特征图: 输出特征图:
![](../../tools/feature_maps_visualization/fm.jpg)
\ No newline at end of file ![](../../tools/feature_maps_visualization/fm.jpg)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册