深度学习模型可视化1

一、可视化特征层

我是直接加在网络中的代码段，对特征层直接取出来，进行可视化。

二、使用步骤

1.引入库

基本上是这几个库：

import matplotlib.pyplot as plt
import numpy as np
import cv2

2.加入在特征层之后

代码如下：

		#取出当前的feat
		res = []
        feature_map = feat.squeeze(0)
        feature_map = feature_map.cpu().numpy()
        feature_map_num = feature_map.shape[0]
        # row_num = np.ceil(np.sqrt(feature_map_num)