# Save true bitmap GIF and TXT to directory
true_txt_url = true_image_url.replace(".gif",".txt")
true_image_file = "true_img_{:04d}.gif".format(i)
true_txt_file = "true_img_{:04d}.txt".format(i)
urllib.urlretrieve (true_image_url, "{}/{}".format(args.output_dir,true_image_file))
urllib.urlretrieve (true_txt_url, "{}/{}".format(args.output_dir,true_txt_file))
# Make GIF from draw events
bmpmaker.clear()
bmpmaker.process_commands_str(draw_events)
bitmap = bmpmaker.bitmap
drawn_image_file = "drawn_img_{:04d}.gif".format(i)
drawn_image_save_path = "{}/{}".format(args.output_dir,drawn_image_file)
imgmaker.save_bitmap(bitmap, drawn_image_save_path)
# Make TXT from draw events
drawn_txt_file = "drawn_img_{:04d}.txt".format(i)
drawn_txt_save_path = "{}/{}".format(args.output_dir,drawn_txt_file)
np.savetxt(drawn_txt_save_path, bitmap, fmt='%d')
# Save command text and generated events text too
commands_txt_file = "written_commands_{:04d}.txt".format(i)
events_txt_file = "draw_events_{:04d}.txt".format(i)
save_txt_to_file(commands.replace("<br>","\n"), "{}/{}".format(args.output_dir,commands_txt_file))
save_txt_to_file(draw_events, "{}/{}".format(args.output_dir,events_txt_file))
html_lines.append("<h2>Worker: #{:03d} ({}) (WORKER_{:03d}_NUM_HITS hits) </h2>".format(worker_index,worker_id,worker_index))
html_lines.append("<b>Image: {}</b> <br>".format(true_image_url))
parser.add_argument("-dim", type=int, default=5, help="Dimensionality of grid")
parser.add_argument("-csv", type=str, required=True, help="input CSV file with drawing task results")
parser.add_argument("-field", type=str, default="Answer.WritingTexts", help="Name of CSV field containing draw events")
parser.add_argument("-output_dir", type=str, default="output/make_images_from_drawing_task_results", help="output directory for images")
args = parser.parse_args()
imgmaker = ImageMaker(args.width, args.padding)
bmpmaker = BitmapMaker(args.width, args.width)
np.random.seed(0)
if not os.path.exists(args.output_dir):
os.makedirs(args.output_dir)
with open(args.csv) as fin:
reader = csv.reader(fin)
header = next(reader)
colindex = header.index(args.field)
for i,row in enumerate(reader):
commands_str = row[colindex]
bmpmaker.clear()
event_sequence = AbsoluteEventSequence.from_mturk_string(commands_str)
event_sequence = event_sequence.canonicalize()
canonical_events = event_sequence.events
bmpmaker.process_commands(canonical_events)
bitmap = bmpmaker.bitmap
bitmap_file = "{}/img_{:04d}.gif".format(args.output_dir,i)
imgmaker.save_bitmap(bitmap, bitmap_file)
default="Answer.WritingTexts",
help="Name of CSV field containing draw events")
parser.add_argument("-output_dir",
type=str,
default="output/make_images_from_drawing_task_results",
help="output directory for images")
args = parser.parse_args()
imgmaker = ImageMaker(args.width, args.padding)
bmpmaker = BitmapMaker(args.width, args.width)
np.random.seed(0)
if not os.path.exists(args.output_dir):
os.makedirs(args.output_dir)
with open(args.csv) as fin:
reader = csv.reader(fin)
header = next(reader)
colindex = header.index(args.field)
for i, row in enumerate(reader):
commands_str = row[colindex]
bmpmaker.clear()
event_sequence = AbsoluteEventSequence.from_mturk_string(
commands_str)
event_sequence = event_sequence.canonicalize()
canonical_events = event_sequence.events
bmpmaker.process_commands(canonical_events)
bitmap = bmpmaker.bitmap
bitmap_file = "{}/img_{:04d}.gif".format(args.output_dir, i)
imgmaker.save_bitmap(bitmap, bitmap_file)
true_image_file = "true_img_{:04d}.gif".format(i)
true_txt_file = "true_img_{:04d}.txt".format(i)
urllib.urlretrieve(
true_image_url, "{}/{}".format(args.output_dir,
true_image_file))
urllib.urlretrieve(true_txt_url,
"{}/{}".format(args.output_dir, true_txt_file))
# Make GIF from draw events
bmpmaker.clear()
bmpmaker.process_commands_str(draw_events)
bitmap = bmpmaker.bitmap
drawn_image_file = "drawn_img_{:04d}.gif".format(i)
drawn_image_save_path = "{}/{}".format(args.output_dir,
drawn_image_file)
imgmaker.save_bitmap(bitmap, drawn_image_save_path)
# Make TXT from draw events
drawn_txt_file = "drawn_img_{:04d}.txt".format(i)
drawn_txt_save_path = "{}/{}".format(args.output_dir,
drawn_txt_file)
np.savetxt(drawn_txt_save_path, bitmap, fmt='%d')
# Save command text and generated events text too
commands_txt_file = "written_commands_{:04d}.txt".format(i)
events_txt_file = "draw_events_{:04d}.txt".format(i)
save_txt_to_file(
commands.replace("<br>", "\n"),
"{}/{}".format(args.output_dir, commands_txt_file))
save_txt_to_file(draw_events,
"{}/{}".format(args.output_dir, events_txt_file))
