def parseConfig(config_file): config_parser = ConfigParser.RawConfigParser() config_parser.read(config_file) data = { 'K' : 2, 'L' : 0, 'gt' : None, 'alpha' : None, 'beta' : None, 'gamma' : None } if config_parser.sections() == []: raise IOError("Invalid configuration file.") if config_parser.has_section('Ground Truth'): config.readVisualization(config_parser, 'Ground Truth', os.path.dirname(config_file), options) config.readFilter(config_parser, 'Ground Truth', os.path.dirname(config_file), options) config.readAlgorithm(config_parser, 'Ground Truth', os.path.dirname(config_file), options) config.readMgsSamples(config_parser, 'Ground Truth', os.path.dirname(config_file), options) data['gt'] = config.readVector(config_parser, 'Ground Truth', 'gt', float) data['L'] = len(data['gt']) data['alpha'], data['beta'], data['gamma'] = \ config.getParameters(config_parser, 'Ground Truth', os.path.dirname(config_file), data['K'], data['L']) config.readStrategy(config_parser, 'Ground Truth', options) result = loadResult() if not result['states']: result['states'] = config.readSeeds(config_parser, 'Ground Truth', 'seeds') result = sample(result, data) if config_parser.has_section('Experiment'): config.readVisualization(config_parser, 'Experiment', os.path.dirname(config_file), options) config.readFilter(config_parser, 'Experiment', os.path.dirname(config_file), options) config.readAlgorithm(config_parser, 'Experiment', os.path.dirname(config_file), options) config.readMgsSamples(config_parser, 'Experiment', os.path.dirname(config_file), options) data['L'] = int(config_parser.get('Experiment', 'bins')) data['alpha'], data['beta'], data['gamma'] = \ config.getParameters(config_parser, 'Experiment', os.path.dirname(config_file), data['K'], data['L']) config.readStrategy(config_parser, 'Experiment', options) result = loadResult() result = sample(result, data) if options['video']: save_video() if options['save']: saveResult(result) else: if options['savefig']: importMatplotlib('Agg') from matplotlib.pyplot import savefig vis.plotSampling(result, options, data) # savefig(options['savefig'], bbox_inches='tight', pad_inches=0) savefig(options['savefig']) else: importMatplotlib() from matplotlib.pyplot import show vis.plotSampling(result, options, data) show()
def save_frame(result, data, utility, i): importMatplotlib('Agg') from matplotlib.pyplot import savefig pyplot.clf() bin_result = call_posterior(result['counts'], data, options) bin_result['counts'] = result['counts'] bin_result['samples'] = result['samples'] bin_result['states'] = result['states'] bin_result['utility'] = utility vis.plotSampling(bin_result, options, data) savefig('%s_%03d.png' % (options['video'], i), bbox_inches='tight', pad_inches=0)