gen_histo.py 文件源码

def create_histo(folder_path, min_val, max_val, num_bins=None):

    if num_bins is None:
        num_bins = int(max_val - min_val)    
        print num_bins

    histogram = {}

    for f in os.listdir(folder_path):

        im = nib.load(os.path.join(folder_path, f))
        data = im.get_data()

        hist, bins = np.histogram(data, num_bins, range=(min_val, max_val))

        count = 0

        for el in bins[0:-1]:
            try:
                histogram[el] += hist[count]
            except:
                histogram[el] = hist[count]
            count += 1

    return histogram.items()