tensorflow保存特征图代码

将channel进行平铺，按照长宽最接近的方式进行拼接

# -*- coding: UTF-8 -*-
#!/usr/bin/python
from __future__ import absolute_import

import cv2
import os
import numpy as np
from PIL import Image
import matplotlib.pyplot as plt


def save_feature_map(raw_img,feature_map,save_dir,name):
    if not os.path.exists('./utils/feature_map'):
        os.mkdir('./utils/feature_map')
    
    if not os.path.exists('./utils/feature_map/{}'.format(save_dir)):
        os.mkdir('./utils/feature_map/{}'.format(save_dir))    
    
#    print(raw_img.shape)
#    print(feature_map.shape)
    
    feature_map_shape = feature_map.shape
    feature_map_h,feature_map_w = feature_map_shape[0], feature_map_shape[1]
    
    channel = feature_map_shape[-1]
    
    save_h,save_w = crack(channel)
    
    array = np.zeros((feature_map_h * save_h ,feature_map_w * save_w))
#    print('mean:',np.mean(feature_map[:,:,0]))
#    print('min:',np.min(feature_map[:,:,0]))   
#    print('max:',np.max(feature_map[:,:,0]))  
    
    k = 0
    for i in range(save_h):
        for j in range(save_w):
            ###归一化到0-255，便于显示
            max_ = np.max(feature_map[:,:,k])
            min_ = np.min(feature_map[:,:,k])
            if min_ < 0:
                a = (feature_map[:,:,k] - min_ ) / (max_ - min_) * 255
            else:
                a = (feature_map[:,:,k] + min_ ) / (max_ - min_) * 255
            array[feature_map_h * i : feature_map_h * (i + 1), 
                  feature_map_w * j : feature_map_w * (j + 1)] = a
            

            k += 1

    cv2.imwrite('./utils/feature_map/{}/{}.png'.format(save_dir,name),array)
#    save_img('./utils/feature_map/{}/{}.png'.format(save_dir,name),array)

def save_img(path,img_array):

    img = Image.fromarray(np.uint8(img_array))
    img.save(path) #还可以指定图片质量img.save('new1.jpg',quality=95)，默认值为75


###将一个数分解为两个最相近的数的乘积
def crack(integer):
    def is_integer(number):
        if int(number) == number:
            return True
        else:
            return False
    start = int(np.sqrt(integer))
    factor = integer / start
    while not is_integer(factor):
        start += 1
        factor = integer / start
    return int(factor), start