vis_fasterRCNN.py 文件源码

def visualization_plus(net, layer_name, save_dir):
    save_path = os.path.join(cfg.ROOT_DIR, 'visualization_plus', save_dir) 
    if not os.path.exists(save_path):
        os.makedirs(save_path)
    feat = net.blobs[layer_name].data[0]
    fm = feat[0]
    print type(fm)

    print fm.shape
    for f in feat:
        fm += f
    fm = fm - feat[0]

    print 'fm  max = {}, min = {}'.format(fm.max(), fm.min())
    #fm -= fm.min()
    #fm /= fm.max()

    fm *=255
    i = 0
    cv2.imwrite(os.path.join(save_path, '{:s}.png'.format(layer_name)), fm)