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)
