FCN中单张图片可视化的代码

仅参考

import numpy as np
from PIL import Image
import sys
import matplotlib.pyplot as plt
#import scipy.io as sio
import caffe

#import caffe
import glob
#fileList=glob.glob('/home/yangwb/JPEGImages/*.jpg')
#f=open("/home/yangwb/MYwork/FCN/fcn.berkeleyvision.org-master/data/VOC2012/ImageSets/Segmentation/seg11valid1.txt")
#lines=f.readlines()
def vis_square(data):

    # 输入的数据为一个ndarray，尺寸可以为(n, height, width)或者是 (n, height, width, 3)
    # 前者即为n个灰度图像的数据，后者为n个rgb图像的数据
    # 在一个sqrt(n) by sqrt(n)的格子中，显示每一幅图像

    # 对输入的图像进行normlization
    data = (data - data.min()) / (