def main(args): """ Runs the pipeline on a given input/output pair and dataset. @params args.dataset: subfolder of data/ args.name: name of the image to process, input and output are stored under data/args.dataset/args.name """ # Setup params params = Parameters() processor = Processor(params) # I/O paths outputDir = os.path.join(OUTPUT_DIR, args.dataset) inputDir = os.path.join(DATA_DIR, args.dataset) # Result struct r = Result() r.dataset = args.dataset r.name = args.name r.dataPath = os.path.join(inputDir,args.name) r.outputPath = os.path.join(outputDir,args.name) r.error = "" # Run r = processor.process(r) print "---------------------------" print "* Processed image %s/%s" % (args.dataset, args.name) print " - time\t%.2f s." % r.computation_time print " - PSNR:\t %5.2f dB" % r.psnr print " - input:\t %5.2f %%" % (r.compression_up*100) print " - output:\t %5.2f %%" % (r.compression_down*100)
start_time = time.time() def generator(): case = Case() for cr in range(randrange(0, 19)): x = randrange(-1000, 1000) y = randrange(-1000, 1000) r = randrange(-360, 360) route = Route(x, y, r) for cc in range(randrange(1, 24)): command_type = choice(list(CommandType)) if command_type == CommandType.TURN: command = Command(command_type, randrange(-360, 360)) else: command = Command(command_type, randrange(-1000, 1000)) route.add_command(command) case.add_route(route) yield case for i in reversed(range(99)): g = generator() g_case = g.next() Processor.process(g_case) print("average x: %s, average y: %s, average worst: %s" % (round(g_case.sx, 4), round(g_case.sy, 4), round(g_case.aw, ))) print("--- %s seconds ---" % (time.time() - start_time)) print(resource.getrusage(resource.RUSAGE_SELF).ru_maxrss)
import os from processor.processor import Processor if __name__ == "__main__": current_path = os.path.dirname(os.path.abspath(__file__)) data_path = os.path.join(current_path, "data") resource_path = os.path.join(data_path, "resources") input_path = os.path.join(data_path, "inputs", "1984.txt") output_path = os.path.join(data_path, "outputs") processor = Processor(resource_path=resource_path, config={}) processor.process(input_path=input_path, output_path=output_path)