CNN特征可视化

def tf_freature_images(net,name):
    net_copy=net[0]
    images=[]
    net_copy.swapaxes(0,2)
    print(net_copy.shape[0])
    for k in range(net_copy.shape[2]):
        image=np.zeros((net_copy.shape[0],net_copy.shape[1]),dtype=np.float64) 
        for j in range(net_copy.shape[1]):
            for i in range(net_copy.shape[0]):
                image[i,j]=net_copy[i,j,k]
        image_1=np.zeros((net_copy.shape[0],net_copy.shape[1]),dtype=np.uint8) 
        image_1=image
        cv2.normalize(image_1,image,0,255,cv2.NORM_MINMAX)
        image_2=image.astype(np.uint8)
        image = cv2.applyColorMap(image_2, cv2.COLORMAP_JET)
        cv2.imwrite("./feature_image/"+name+"_"+str(k)+".jpg",image)
        images.append(image)
    return images

使用OpenCV对特征进行可视乎，深度学习框架采用tensorflow。

函数参数输入为“net”为[batch_size,cols,rows,channels]