pagenet/render_quads.py

import sys
import numpy as np
import cv2
import os

if len(sys.argv) < 4:
	print("python %s manifest.txt dataset_dir out_dir" % __file__)
	exit()

manifest_file = sys.argv[1]
dataset_dir = sys.argv[2]
out_dir = sys.argv[3]

try:
	os.makedirs(out_dir)
except:
	pass

file_list = [s.strip() for s in open(manifest_file, 'r').readlines()]
for line in file_list:
	tokens = line.split(',')
	f = tokens[0]
	coords = list(map(float, tokens[1:9]))

	resolved = os.path.join(dataset_dir, f)
	im = cv2.imread(resolved, 0)
	gt = np.zeros(im.shape, dtype=np.uint8)
	cv2.fillPoly(gt, np.array(coords).reshape((4, 2)).astype(np.int32)[np.newaxis,:,:], 255)

	out_fn = os.path.join(out_dir, f.replace('/', '_'))[:-4] + ".png"
	cv2.imwrite(out_fn, gt)