vis_fasterRCNN.py 文件源码

def visualization(net, layer_name, save_dir):
    save_path = os.path.join(cfg.ROOT_DIR, 'visualization', save_dir, layer_name) 
    if not os.path.exists(save_path):
        os.makedirs(save_path)
    feat = net.blobs[layer_name].data[0]
    print feat.shape
    feat -= feat.min()
    feat /= feat.max()
    feat *=255
    i = 0
    for im in feat:
        #iFColor = FColor(im)
        cv2.imwrite(os.path.join(save_path, '{:d}.png'.format(i)), im)
        i = i + 1
    #vis_square(feat, padval=1)