caffe模型参数可视化过程中的显示函数的代码解读

下面是模型参数可视化的函数代码,# force the number of filters to be square和# tile the filters into an image这段都表示什么意思阿,为什么会这样写阿,求大神赐教。
def show_feature(data, padsize=1, padval=0):
data -= data.min()
data /= data.max()

# force the number of filters to be square
n = int(np.ceil(np.sqrt(data.shape[0])))
padding = ((0, n ** 2 - data.shape[0]), (0, padsize), (0, padsize)) + ((0, 0),) * (data.ndim - 3)
data = np.pad(data, padding, mode='constant', constant_values=(padval, padval))

# tile the filters into an image
data = data.reshape((n, n) + data.shape[1:]).transpose((0, 2, 1, 3) + tuple(range(4, data.ndim + 1)))
data = data.reshape((n * data.shape[1], n * data.shape[3]) + data.shape[4:])
plt.imshow(data)
plt.axis('off')
已邀请:

alex68 - 一般不扯淡~

赞同来自:

为了在一个正方形的grid中显示 不是什么要紧的内容

要回复问题请先登录注册