demo.py 文件源码

def make_liposomes(lip_class, n, out_dir):
    """

    :param lip_class:
    :param n:
    :param out_dir:
    :return:
    """
    try:
        os.makedirs(out_dir)
    except OSError as exc:
        if exc.errno != errno.EEXIST:
            raise

    for i in xrange(n):
        liposome = lip_class(128, 128, prob_deviation=0.5)
        liposome.make()
        img_name = '{}.png'.format(i + 1)
        img_path = os.path.join(out_dir, img_name)
        print('\rSaving {}'.format(img_path), end='')
        mpimg.imsave(img_path, liposome.data, cmap='Greys_r', vmin=0, vmax=1)